为某些类型的行实现分页

时间:2019-06-19 15:42:40

标签: java spring-boot jpa spring-data-jpa jpa-2.0

我只想对某些类型的用户实施分页。表结构为:

@Entity
@Table(name = "users")
public class Users implements Serializable {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    @Column(name = "id", unique = true, updatable = false, nullable = false, length = 3)
    private int id;

    @Column(length = 255)
    private String login;

    @Column(length = 255)
    private String email;

    @Column(length = 64)
    private String salt;

    @Column(length = 255)
    private String type;
    .....
}

存储库:

@Repository
public interface UserRepository extends JpaRepository<Users, Integer> {

}

查询:

@Override
public Page<Users> findAll(int page, int size) {
    return dao.findAll(PageRequest.of(page, size));
}

如何仅选择type admin,super_admin等来限制查询。

我想我需要添加诸如IN(adminsuper_admin)之类的限制吗?

1 个答案:

答案 0 :(得分:2)

最简单的方法可能是使用命名约定并使用findAllByTypeIn

interface UserRepository extends JpaRepository<User, Integer> {
  Page<User> findAllByTypeIn(Pageable page,String... types));
}

但是如果您有更复杂的搜索条件,我会考虑使用规格

Spring Data JPA 支持规范,您可以使用JpaSpecificationExecutor接口扩展存储库接口,如下所示:

interface UserRepository extends JpaRepository<User, Integer>, JpaSpecificationExecutor<User> {

}

写您的规范

class UserSpecs {

    public static Specification<User> isOfType(String type) {

        return new Specification<User>() {

            @Override
            public Predicate toPredicate(Root<User> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
                return builder.equal(root.get("type"), type);
            }
        };
    }


    public static Specification<User> isInTypes(String... types) {

        return new Specification<User>() {

            @Override
            public Predicate toPredicate(Root<User> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
               In<String> inClause = builder.in(root.get("type"));
               for (String type: types) {
                 inClause.value(type);      
               }
               return inClause;
            }
        };
    }

}

并通过

获取结果
dao.findAll(UserSpecs.isOfType("admin").or(UserSpecs.isOfType("super_admin")) ,PageRequest.of(page, size));

,或者如果您想在

中使用
dao.findAll(UserSpecs.isInTypes("admin","super_admin")) ,PageRequest.of(page, size));