解决升级SpringBoot3 JPA报Could not locate TableGroup问题

产品技术架构从SpringBoot2.x升级到SpringBoot3.x后，对原有代码进行单元测试时发现，之前通过CriteriaQuery查询对象实现的分页查询功能，在进行记录数count查询时，会抛出SqlTreeCreationException: Could not locate TableGroup异常。

通过排查发现，SpringBoot2.x依赖的是Hibernate 5.x，SpringBoot3.x依赖的是Hibernate 6.x，在进行count记录数查询时，查询条件直接拿的分页查询的Predicate，而Hibernate 6.x中在一个 CriteriaQuery 的 where 子句中使用另一个 CriteriaQuery 的Root会导致 "找不到 TableGroup" 错误，示例代码如下：

// 分页查询
criteriaQuery = criteriaBuilder.createQuery(entityClass);
entityRoot = criteriaQuery.from(entityClass);
// 设置查询条件
List<Predicate> predicates = new ArrayList<>();
predicates.add(criteriaBuilder.equal(criteriaBuilder.equal(entityRoot.get(Reflections.fnToFieldName(fn)), "张三"));
criteriaQuery.where(predicates.toArray(new Predicate[0]));
// ...省略部分查询代码

// 记录数查询
CriteriaQuery<Long> countQuery = criteriaBuilder.createQuery(Long.class);
Root<T> countRoot = countQuery.from(entityClass);
// 设置当前记录数统计的Root别名，和条件查询entity的Root别名保持一致
// 解决select _user1.id from user _user1 where _user0.name = '' and _user0.sfsc = 'N';别名不一致问题
countRoot.alias(entity.getAlias());
countQuery.select(criteriaBuilder.count(countRoot));
// 直接使用上面的predicates，会抛出异常
countQuery.where(predicates.toArray(new Predicate[0]));
// ...省略部分查询代码

JpaSystemException: Could not locate TableGroup

解决方案

// 分页查询
HibernateCriteriaBuilder criteriaBuilder = entityManager.unwrap(Session.class).getCriteriaBuilder();
JpaCriteriaQuery<Entity> criteriaQuery = criteriaBuilder.createQuery(Entity.class);
Root<Entity> entityRoot = criteriaQuery.from(Entity.class);
// 设置查询条件
List<Predicate> predicates = new ArrayList<>();
predicates.add(criteriaBuilder.equal(criteriaBuilder.equal(entityRoot.get(Reflections.fnToFieldName(fn)), "张三"));
criteriaQuery.where(predicates.toArray(new Predicate[0]));
// ...省略部分查询代码

// 记录数查询
Long total = entityManager.createQuery(criteriaQuery.createCountQuery()).getSingleResult();