QueryDSL 快速获取实体ID与关联数映射方法

本文介绍如何使用 QueryDSL 4.2.1 在不加载完整实体的前提下，通过原生聚合查询构建 Map<主实体ID, 关联实体数量>，避免 N+1 查询与内存开销，提升大数据量场景下的性能。

本文介绍如何使用 QueryDSL 4.2.1 在不加载完整实体的前提下，通过原生聚合查询构建 `Map<主实体ID, 关联实体数量>`，避免 N+1 查询与内存开销，提升大数据量场景下的性能。

在实际开发中，常需统计某类主实体（如 Order）与其关联子实体（如 OrderItem）的数量关系，但又无需加载全部实体对象——此时若使用 fetchJoin() + Java 端分组，不仅冗余传输数据，还易引发内存压力。QueryDSL 提供了面向 JPA 的类型安全聚合能力，可直接在数据库层完成分组计数，并映射为简洁的 Map<K, V> 结构。

核心思路是：以关联实体（qE2）为查询主体，左连接其所属主实体（qE2.qE1），按主实体 ID 分组并计数，最后将结果流式转换为 Map。以下为推荐实现（基于 JPAQueryFactory）：

Map<Long, Long> idToCountMap = new JPAQueryFactory(entityManager)
    .select(qE2.qE1.id, qE2.count())           // 选择主实体ID + 当前分组计数
    .from(qE2)                                 // 从关联实体表开始查询
    .join(qE2.qE1)                             // 显式 inner join 主实体（确保仅统计有归属的记录）
    .where(qE1_exp)                            // 应用主实体过滤条件（如 status = 'ACTIVE'）
    .groupBy(qE2.qE1.id)                       // 按主实体ID分组
    .fetch()                                   // 执行查询，返回 List<Tuple>
    .stream()
    .collect(Collectors.toMap(
        tuple -> tuple.get(qE2.qE1.id),        // key: 主实体ID
        tuple -> tuple.get(qE2.count())        // value: 该ID对应的关联实体数量
    ));

⚠️ 注意事项：

• 若需包含无任何关联子实体的主实体（即 count = 0），应将 join(qE2.qE1) 改为 leftJoin(qE2.qE1)，并调整 select 为 qE1.id, Expressions.asNumber(0L).when(qE1.id.isNotNull()).otherwise(qE2.count())（需配合 caseWhen 处理空值），或改用主实体 qE1 为主表 + left join qE2 实现；
• qE1_exp 必须为 BooleanExpression 类型（如 qE1.status.eq("ACTIVE")），不可传入 null，否则抛 NullPointerException；
• qE2.count() 默认等价于 count(*)；若需去重计数（如 COUNT(DISTINCT qE2.id)），请显式使用 Expressions.countDistinct(qE2.id)；
• 该方案完全绕过实体实例化，SQL 层即生成形如 SELECT e1.id, COUNT(*) FROM e2 JOIN e1 ON e2.e1_id = e1.id WHERE ... GROUP BY e1.id 的高效语句。

总结：此模式兼顾类型安全、可读性与性能，是 QueryDSL 中处理“一对多基数统计 + 轻量映射”的标准实践。建议在 Repository 层封装为通用方法，例如 Map<ID, Long> countAssociationsByParent(Class<T> parentType, BooleanExpression whereClause)，进一步提升复用性。