如何将指定SQL转换为Java Specification并解决OR逻辑问题

原需求SQL #

select * from student s where s.group_id in (
select student_id from class c where c.place='23' and c.student_id IS NOT NULL)
or s.tab_num = '23';

问题代码 #

需要将上述SQL转换为Java Specification，涉及Student和Class两张表，现有代码无法将子查询结果与cb.or()正确结合，代码如下：

public static Predicate studentPredicate(CriteriaBuilder cb, Root<?> r) {

CriteriaQuery<Class> subQuery = cb.createQuery(Class.class);

Root<Class> subRoot = subQuery.from(Class.class);

subQuery.select(subRoot.get("student_id")).where(cb.equal(subRoot.get("place"), 23L), cb.isNotNull(subRoot.get("student_id"));

CriteriaQuery<Dictionary> id = outerQuery.where(outerRoot.get("group_id").in(subQuery));

return cb.or(cb.equal(r.get("tabNum"), 23)); // here is an issue
}

修正方案 #

原代码存在三个核心问题：子查询类型定义错误、引用未定义的outerQuery/outerRoot、未将子查询条件纳入OR组合。以下是修正后的完整实现：

public static Predicate studentPredicate(CriteriaBuilder cb, Root<Student> r) {
    // 构建子查询：筛选place为'23'且student_id非空的student_id集合
    Subquery<Long> subQuery = cb.createSubquery(Long.class);
    Root<Class> subRoot = subQuery.from(Class.class);
    
    subQuery.select(subRoot.get("student_id"))
            .where(
                cb.equal(subRoot.get("place"), "23"), // 匹配原SQL的字符串类型，若实体类place为数值则改为23L
                cb.isNotNull(subRoot.get("student_id"))
            );
    
    // 分别构建两个条件
    Predicate groupIdInSubquery = cb.in(r.get("group_id")).value(subQuery);
    Predicate tabNumMatch = cb.equal(r.get("tabNum"), "23");
    
    // OR组合两个条件并返回
    return cb.or(groupIdInSubquery, tabNumMatch);
}

关键说明 #

• 子查询使用createSubquery()创建，明确指定返回类型为student_id对应的实际类型（如Long，需根据实体类字段调整）
• 直接利用方法参数中的Root<Student>构建外层查询条件，无需额外定义outerQuery
• 先单独生成两个Predicate，再通过cb.or()将子查询的IN条件与tab_num匹配条件组合，完全对应原SQL逻辑

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