SpringData JPA 利用Case实现非规则排序

最新推荐文章于 2023-11-07 16:39:17 发布

原创最新推荐文章于 2023-11-07 16:39:17 发布 · 508 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#java #开发语言

java 专栏收录该内容

1 篇文章

订阅专栏

本文介绍如何使用编程技巧实现数据库中巡检任务的排序，针对非标准状态值（如审核不通过-5到完成-9），通过自定义排序逻辑，首先按状态5, 0, 1, 3, 9排列，再依据开始时间升序。

开发时候客户需要将巡检单按照状态和开始时间排序

客户需求：
在这里插入图片描述
但是在数据库中state的值不是正常排序的：
审核不通过-5 待处理-0 处理中-1 待审核-3 完成-9
所以总结以下就是需要对象非规则排序，排序是先按照state属性以 5 → （0，1） → 3 → 9的顺序然后再按照开始时间startTime次排序。
代码实现如下：

 List<BusTaskEntity> busTaskEntities = busTaskReposity.findAll(new Specification<BusTaskEntity>() {
            @Override
            public Predicate toPredicate(Root<BusTaskEntity> root, CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) {
                ArrayList<Predicate> predicates = new ArrayList<>();
                predicates.add(criteriaBuilder.equal(root.get(IS_DELETE), 0));
              
                // 审核不通过5 待处理0 处理中1 待审核3 完成9
                ArrayList<Predicate> statesPredicate = new ArrayList<>();
                statesPredicate.add(criteriaBuilder.equal(root.get("state"), 0));
                statesPredicate.add(criteriaBuilder.equal(root.get("state"), 1));
                Order order = criteriaBuilder.asc(
                        criteriaBuilder.selectCase()
                                .when(criteriaBuilder.equal(root.get("state"), 5), 1)
                                .when(criteriaBuilder.or(statesPredicate.toArray(new Predicate[statesPredicate.size()])), 2)
                                .when(criteriaBuilder.equal(root.get("state"), 3), 4)
                                .when(criteriaBuilder.equal(root.get("state"), 9), 5)
                                .otherwise(99)
                );
                List<Integer> list = Lists.newArrayList(5, 0, 1, 3, 9);
                query.where(criteriaBuilder.and(predicates.toArray(new Predicate[predicates.size()])))
                        .orderBy(order, criteriaBuilder.desc(root.get(START_TIME)));
                return query.getRestriction();
            }
        });