如何使用Spring Data JPA Specifications按指定多值顺序排序

你的SQL包含两个核心逻辑：status_id的IN条件过滤和基于status_id的自定义顺序排序，以下是通过Spring Data JPA Specifications实现的具体方案：

1. 实现Specifications的条件过滤 #

在你的DocsSpec类中，添加status_id的IN过滤条件：

public class DocsSpec implements Specification<Docs> {
    private final String filter;
    private final String type;
    // 定义目标status_id集合
    private static final List<String> TARGET_STATUS_IDS = Arrays.asList("105", "55", "65", "228016");

    public DocsSpec(String filter, String type) {
        this.filter = filter;
        this.type = type;
    }

    @Override
    public Predicate toPredicate(Root<Docs> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
        List<Predicate> predicates = new ArrayList<>();
        
        // 保留你原有filter和type的条件逻辑（此处省略你的原有代码）
        // ...
        
        // 添加status_id IN条件
        predicates.add(root.get("statusId").in(TARGET_STATUS_IDS));
        
        return cb.and(predicates.toArray(new Predicate[0]));
    }
}

2. 实现自定义CASE WHEN排序 #

Spring Data JPA的默认Sort不支持CASE WHEN逻辑，需要在toPredicate方法中直接通过CriteriaQuery构建排序规则：

@Override
public Predicate toPredicate(Root<Docs> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
    List<Predicate> predicates = new ArrayList<>();
    
    // 原有条件逻辑 + status_id IN条件
    // ...
    
    // 构建CASE WHEN排序表达式
    Expression<Integer> orderCase = cb.selectCase()
        .when(cb.equal(root.get("statusId"), "55"), 1)
        .when(cb.equal(root.get("statusId"), "228016"), 2)
        .when(cb.equal(root.get("statusId"), "65"), 3)
        .when(cb.equal(root.get("statusId"), "105"), 4)
        .otherwise(5); // 其他状态默认排最后
    
    // 设置查询的排序规则（升序对应SQL中的顺序）
    query.orderBy(cb.asc(orderCase));
    
    return cb.and(predicates.toArray(new Predicate[0]));
}

3. 调整调用代码 #

由于排序逻辑已在Spec中定义，无需再通过PageRequest传入Sort参数：

p2 = repository.findAll(new DocsRepository.DocsSpec(filter, type), 
    PageRequest.of(page, size));

如果需要保留原有status字段的排序作为补充，可在query.orderBy中添加多条件排序：

query.orderBy(cb.asc(orderCase), cb.desc(root.get("status")));

内容的提问来源于stack exchange，提问作者Tracker7