如何使用Criteria API拆分查询解决MultipleBagFetchException

核心实现逻辑 #

Hibernate的持久化上下文（对应当前EntityManager的一级缓存）会对同一实体实例做自动去重挂载。基于这个特性，我们不需要手动做实体和集合的匹配，完全匹配官方推荐的单批次单集合抓取方案，分两次独立执行Criteria查询，每次只抓取一个集合关联：

1. 第一次查询加载所有符合业务过滤条件的Lesson实体，同步抓取第一个集合（如students）
2. 第二次查询使用完全相同的过滤条件，仅抓取第二个集合（如guests）

执行第二次查询时，Hibernate检测到查询结果中的Lesson实体已经存在于一级缓存中，会自动把当前抓取的guests集合内容，挂载到已存在的Lesson实例上，不会重复创建新的实体对象。

具体代码实现 #

@Repository
@Transactional(readOnly = true) // 保证整个逻辑运行在同一持久化上下文内
public class LessonCriteriaRepositoryImpl implements LessonCriteriaRepository {

    @PersistenceContext
    private EntityManager em;

    @Override
    public List<Lesson> findAll() {
        CriteriaBuilder builder = em.getCriteriaBuilder();

        // 第一次查询：加载匹配的Lesson + 抓取students集合
        CriteriaQuery<Lesson> firstQuery = builder.createQuery(Lesson.class);
        Root<Lesson> firstRoot = firstQuery.from(Lesson.class);
        List<Predicate> firstPredicates = buildSharedPredicates(builder, firstRoot);
        
        firstQuery.select(firstRoot)
                .where(firstPredicates.toArray(new Predicate[0]))
                .distinct(true);
        // 单次仅抓取一个集合，不会触发异常
        firstRoot.fetch(Lesson_.students, JoinType.LEFT);

        List<Lesson> matchedLessons = em.createQuery(firstQuery).getResultList();

        // 第二次查询：同条件加载 + 抓取guests集合
        CriteriaQuery<Lesson> secondQuery = builder.createQuery(Lesson.class);
        Root<Lesson> secondRoot = secondQuery.from(Lesson.class);
        List<Predicate> secondPredicates = buildSharedPredicates(builder, secondRoot);
        
        secondQuery.select(secondRoot)
                .where(secondPredicates.toArray(new Predicate[0]))
                .distinct(true);
        // 单次仅抓取guests一个集合
        secondRoot.fetch(Lesson_.guests, JoinType.LEFT);

        // 执行查询，Hibernate自动将guests关联填充到已存在的Lesson实例
        em.createQuery(secondQuery).getResultList();

        // 返回的结果中两个集合均已被完全加载，无懒加载触发风险
        return matchedLessons;
    }

    /**
     * 抽离公共业务过滤条件，保证两次查询的筛选范围完全一致
     */
    private List<Predicate> buildSharedPredicates(CriteriaBuilder builder, Root<Lesson> root) {
        List<Predicate> predicates = new ArrayList<>();
        // 此处编写所有实际业务的过滤规则，比如课程状态、开课时间筛选等
        // 示例：predicates.add(builder.equal(root.get(Lesson_.status), LessonStatus.PUBLISHED));
        return predicates;
    }
}

关键注意事项 #

• 必须保证两次查询运行在同一个持久化上下文内，Spring环境下给Repository方法添加@Transactional(readOnly = true)即可，若上下文提前关闭，两次查询的实体不共享缓存，就无法实现集合的自动挂载。
• 两次查询的过滤条件必须完全一致，否则会出现部分Lesson的关联集合未被加载的问题。将Predicate构建逻辑抽成独立方法复用，是避免条件不一致的最稳妥方式。
• 这种实现生成的两条独立SQL，不存在多集合连接导致的笛卡尔积问题，相比单条多表关联查询的性能更优，不会因为重复数据占用过多内存。
• 若后续有更多集合关联需要抓取，按照相同逻辑依次新增独立的单集合fetch查询即可，全程不会触发MultipleBagFetchException。

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