如何在Spring Data Specification中用CriteriaBuilder合并枚举与Null场景的查询结果

我们通过Spring Data Specification的CriteriaBuilder操作Student与StudentResult两张关联表：

• 当API请求携带枚举值参数（如http://localhost:8080/student?query=grade:5,lastRollResult@SUCCESS#FAILURE）时，通过INNER JOIN关联StudentResult表，查询其result字段；
• 当传入lastRollResult=null时，直接查询Student表的lastRunId字段（此时无对应关联记录，返回null值）。

现在需要实现：当查询参数同时包含枚举值（如SUCCESS/FAILURE）和null时，合并两种场景的查询结果。

当前核心实现代码：

@Override
protected Expression<String> getPath(SearchCriteria criteria, Root<Student> root) {
  if (criteria.getKey().equals("lastRollResult"))  { // 判断是否为null，避免关联表
    if (!"null".equals(criteria.getValue())) {
      Join results = root.join("lastRollResult", JoinType.INNER);
      return results.get("result");
    } else {
      return root.get("lastRunId");
    }
}
  return root.get(criteria.getKey());
}

基础类定义：public abstract class BaseSpecification<M> implements Specification<M>

当前getPath方法仅能处理单一查询值（要么枚举、要么null），无法支持多值合并查询。需要调整逻辑，在Specification的核心方法toPredicate中构建OR条件，同时覆盖两种场景：

核心修改思路 #

1. 放弃在getPath中做分支判断，转而在toPredicate中拆分多值查询条件；
2. 用LEFT JOIN替代INNER JOIN，确保无关联StudentResult的Student记录能被查询到；
3. 分别构建枚举值匹配、null值匹配的子条件，通过CriteriaBuilder.or()合并；
4. 适配多值参数解析，将包含枚举和null的查询参数解析为列表传入。

代码实现 #

调整toPredicate方法 #

@Override
public Predicate toPredicate(Root<M> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
    List<Predicate> predicates = new ArrayList<>();
    
    for (SearchCriteria criteria : criteriaList) {
        if ("lastRollResult".equals(criteria.getKey())) {
            // 假设criteria.getValue()已解析为包含多值的List<Object>
            List<Object> queryValues = (List<Object>) criteria.getValue();
            List<Predicate> subPredicates = new ArrayList<>();
            
            // 1. 处理枚举值查询：关联StudentResult匹配指定结果
            List<Object> enumValues = queryValues.stream()
                    .filter(val -> !"null".equals(val))
                    .collect(Collectors.toList());
            if (!enumValues.isEmpty()) {
                Join<Student, StudentResult> resultJoin = root.join("lastRollResult", JoinType.LEFT);
                subPredicates.add(resultJoin.get("result").in(enumValues));
            }
            
            // 2. 处理null查询：匹配无关联StudentResult的Student记录
            if (queryValues.contains("null")) {
                subPredicates.add(cb.isNull(root.get("lastRunId")));
            }
            
            // 合并子条件为OR，加入总条件列表
            if (!subPredicates.isEmpty()) {
                predicates.add(cb.or(subPredicates.toArray(new Predicate[0])));
            }
        } else {
            // 保留原有普通字段的查询逻辑
            Expression<String> expression = getPath(criteria, root);
            switch (criteria.getOperation().toLowerCase()) {
                case ">":
                    predicates.add(cb.greaterThan(expression, criteria.getValue().toString()));
                    break;
                case "<":
                    predicates.add(cb.lessThan(expression, criteria.getValue().toString()));
                    break;
                case ":":
                    if (expression.getJavaType() == String.class) {
                        predicates.add(cb.like(expression, "%" + criteria.getValue() + "%"));
                    } else {
                        predicates.add(cb.equal(expression, criteria.getValue()));
                    }
                    break;
                // 其他操作符请自行补充
            }
        }
    }
    
    return cb.and(predicates.toArray(new Predicate[0]));
}

简化getPath方法 #

去掉lastRollResult的特殊分支，仅处理普通字段路径：

@Override
protected Expression<String> getPath(SearchCriteria criteria, Root<M> root) {
    return root.get(criteria.getKey());
}

关键说明 #

• LEFT JOIN的必要性：INNER JOIN会过滤掉无关联StudentResult的Student记录，无法匹配null场景，因此必须改用LEFT JOIN；
• OR条件合并：将枚举匹配和null匹配的子条件用OR连接，实现两种场景的结果合并；
• 参数解析适配：需要确保查询参数（如lastRollResult@SUCCESS#FAILURE#null）被正确解析为包含SUCCESS、FAILURE、null的列表，传入SearchCriteria的value字段。

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