使用Hibernate Criteria API获取子查询值时遇TableGroup定位错误求助

核心问题点 #

1. 返回类型不匹配：方法声明返回Long，但实际调用getResultList()返回的是List<Long>，编译和运行时都会出现类型错误。
2. 内部API操作风险：代码强制转换为SqmSubQuery、SqmQuerySpec等Hibernate私有API，这类API依赖未公开的上下文状态（如TableGroup），仅用simpleContext()复制查询结构无法正确初始化表关联，导致抛出Could not locate TableGroup错误。
3. 子查询与主查询关联错误：主查询直接引用子查询的subRoot（属于子查询上下文），主查询无法识别该根对象；同时覆盖子查询的multiselect会破坏复制而来的查询结构完整性。

解决步骤 #

1. 修正返回类型 #

将方法签名从：

protected Long getIdList(...)

改为：

protected List<Long> getIdList(...)

2. 改用标准JPA Criteria API（推荐） #

避免依赖Hibernate内部API，用标准子查询关联方式实现需求：

protected List<Long> getIdList(
        @NonNull final CriteriaBuilder cb,
        @NonNull final CriteriaQuery<?> rootQuery,
        @NonNull final Pageable pageable) {
    // 关联原查询的根，创建子查询并选择id
    Subquery<Long> subQuery = rootQuery.subquery(Long.class);
    Root<?> subRoot = subQuery.correlate(((AbstractQuery<?>) rootQuery).getRoots().iterator().next());
    subQuery.select(subRoot.get("id"));
    // 复制原查询的过滤条件
    subQuery.where(rootQuery.getRestriction());

    // 创建主查询，基于子查询结果获取分页后的id列表
    CriteriaQuery<Long> listQuery = cb.createQuery(Long.class);
    Root<Long> subQueryRoot = listQuery.from(subQuery);
    listQuery.select(subQueryRoot.get("id"));

    return entityManager
            .createQuery(listQuery)
            .setFirstResult(PageableUtils.getOffsetAsInteger(pageable))
            .setMaxResults(pageable.getPageSize())
            .getResultList();
}

3. 若必须使用Hibernate SQM API（不推荐） #

如果因业务需求必须复制原查询的SQM结构，需确保复制上下文加载完整的会话工厂元数据：

// 替换原复制上下文的创建方式
final SqmCopyContext copyContext = SqmCopyContext.withSessionFactory(
        entityManager.getEntityManagerFactory().unwrap(SessionFactory.class)
);
final SqmQuerySpec<Tuple> subQuerySpec = rootQuerySpec.copy(copyContext);

同时，不要直接调用subQuery.multiselect，而是修改复制后的subQuerySpec的选择项：

subQuerySpec.setSelection(subQuerySpec.getRoots().iterator().next().get("id"));

额外提示 #

• 优先使用标准JPA API，减少对Hibernate版本的依赖；
• 若使用Hibernate 6.x+版本，SQM API内部逻辑变动较大，需确保复制上下文包含完整的会话工厂元数据。

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