在Spring Data JPA中实现分组获取最新关联数据的方案探讨

业务背景与原生SQL对应逻辑 #

实体A与B、B与C均为一对多（1-N）关联，需求是筛选出A.other_field = 1234的A实体，并为其关联的每个B获取最新的C记录，对应的原生SQL如下：

select * from (
    select C.*, Row_number() OVER ( partition BY B.b_id ORDER BY C.c_id DESC ) as rn
    from C
    join B on B.b_id = C.b_id
    join A on A.a_id = B.a_id
    where A.other_field = 1234
) as subquery where rn = 1

可选实现方案（不止原生查询一种） #

1. 关联子查询方式 #

通过子查询找到每个B对应的最大c_id，再关联C表获取完整记录，JPQL实现如下：

SELECT c FROM C c
JOIN c.b b
JOIN b.a a
WHERE a.other_field = 1234
AND c.c_id = (
    SELECT MAX(c2.c_id) FROM C c2 WHERE c2.b = b
)

注：若业务中存在多个C记录共享同一最大c_id（自增主键场景不会出现），会返回多条结果，需根据业务调整过滤逻辑。

2. JPA Criteria API构建查询 #

如果需要动态生成查询，可通过Criteria API实现与上述子查询等价的逻辑：

CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<C> query = cb.createQuery(C.class);
Root<C> cRoot = query.from(C.class);
Join<C, B> bJoin = cRoot.join("b");
Join<B, A> aJoin = bJoin.join("a");

// 子查询获取每个B的最大c_id
Subquery<Long> maxCIdSubquery = query.subquery(Long.class);
Root<C> cSubRoot = maxCIdSubquery.from(C.class);
maxCIdSubquery.select(cb.max(cSubRoot.get("cId")))
              .where(cb.equal(cSubRoot.get("b"), bJoin));

query.select(cRoot)
     .where(cb.equal(aJoin.get("otherField"), 1234),
            cb.equal(cRoot.get("cId"), maxCIdSubquery));

List<C> result = entityManager.createQuery(query).getResultList();

3. 实体模型优化+特定JPA注解 #

若业务频繁需要获取B的最新C，可在B实体中新增一对一关联，借助提供商扩展注解实现：

@Entity
public class B {
    // 其他字段
    @OneToMany(mappedBy = "b")
    private List<C> cs;

    @OneToOne
    @JoinColumn(name = "b_id", insertable = false, updatable = false)
    @Where(clause = "c_id = (SELECT MAX(c_id) FROM C WHERE b_id = b.b_id)")
    private C latestC;
}

注：该方案依赖Hibernate等JPA提供商的@Where注解，移植性稍弱，但使用时可直接通过B实体获取最新C。

方案选择建议 #

原生窗口函数查询在数据量较大时性能更优，数据库可更好地优化执行计划；子查询或实体模型方案在代码可读性、易用性上有优势，可根据项目JPA版本、数据库支持情况及业务需求选择。

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