我们项目用的JPA实现与数据库的交互,这次遇到了一个需求要对订单进行一定规则的排序。因为之前项目都是用的pageable对返回数据进行封装,pageable本身就有对排序的处理,但是jpa还是刚还是用,所以查了下资料,发现也可以对数据进行排序,特此记录。
把Predicate应用到CriteriaQuery中去,因为还可以给CriteriaQuery添加其他的功能,比如排序、分组啥的。如下图使我们的源代码:
首先对已删除数据和locationInfoId对数据进行过滤,然后根据deliveryType,orderPaytime字段对数据进行排序,前三种deliveryType的订单排在前面,并按orderPaytime进行升序排列,后两种类型的订单排在后面,也按orderPaytime进行升序排列。
Specification specification =
(Specification) (root, criteriaQuery, criteriaBuilder) -> {
List predicates = new ArrayList<>();
// 排除逻辑删除数据
predicates.add(criteriaBuilder.equal(root.get("dataStatus"), AuditModel.DATA_STATUS_ACTVIE));
if (null != locationInfoId) {
predicates.add(criteriaBuilder.equal(root.get("locationInfoId"), locationInfoId));
}
if (!StringUtils.isEmpty(shipStatus)) {
CriteriaBuilder.In
Specification spec = new Specification() {
public Predicate toPredicate(Root root,
CriteriaQuery> query, CriteriaBuilder cb) {
Predicate p1 = cb.like(root.get("name").as(String.class), "%"+um.getName()+"%");
Predicate p2 = cb.equal(root.get("uuid").as(Integer.class), um.getUuid());
Predicate p3 = cb.gt(root.get("age").as(Integer.class), um.getAge());
//把Predicate应用到CriteriaQuery中去,因为还可以给CriteriaQuery添加其他的功能,比如排序、分组啥的
query.where(cb.and(p3,cb.or(p1,p2)));
//添加排序的功能
query.orderBy(cb.desc(root.get("uuid").as(Integer.class)));
return query.getRestriction();
}
};
Specification specification =
(Specification) (root, criteriaQuery, criteriaBuilder) -> {
List predicates = new ArrayList<>();
// 排除逻辑删除数据
predicates.add(criteriaBuilder.equal(root.get("dataStatus"), AuditModel.DATA_STATUS_ACTVIE));
if (null != locationInfoId) {
predicates.add(criteriaBuilder.equal(root.get("locationInfoId"), locationInfoId));
}
if (!StringUtils.isEmpty(shipStatus)) {
CriteriaBuilder.In