如何使用Hibernate与JPA在单查询中过滤关联实体？

嘿，这个问题我之前做项目时也纠结过，JPA/Hibernate里过滤关联实体确实容易踩坑，尤其是想在单查询里搞定的话，我给你几个亲测有效的方案：

方案一：使用Hibernate的@Filter注解 #

这是Hibernate特有的功能，配置好后能局部或全局过滤关联实体，步骤如下：

1. 在实体B上定义过滤器：

@FilterDef(name = "activeBFilter", parameters = @ParamDef(name = "activeStatus", type = Boolean.class))
@Filter(name = "activeBFilter", condition = "is_active = :activeStatus")
public class B {
    private Long id;
    private Boolean isActive;
    @ManyToOne
    private A a;
    // getter、setter省略
}

2. 在实体A的关联字段上启用这个过滤器：

public class A {
    private Long id;
    @OneToMany
    @FilterJoinTable(name = "activeBFilter", parameters = @ParamValue(name = "activeStatus", value = "true"))
    private List<B> bList;
    // getter、setter省略
}

3. 查询前开启过滤器，并且必须用FETCH JOIN确保关联实体一次性加载：

// 先开启过滤器
Session session = entityManager.unwrap(Session.class);
session.enableFilter("activeBFilter").setParameter("activeStatus", true);

// 执行查询，加FETCH JOIN才能避免懒加载触发额外查询
List<A> aList = entityManager.createQuery("SELECT a FROM A a LEFT JOIN FETCH a.bList", A.class)
                             .getResultList();

你之前说@FilterJoinTable会触发额外查询，大概率是没加FETCH JOIN或者忘了开启过滤器，这两个点缺一不可哦。

方案二：用JPQL/Criteria API的LEFT JOIN FETCH + ON条件 #

如果不想用Hibernate专属注解，标准JPA的方式也能搞定，JPQL 2.1及以上支持在JOIN时加ON条件，刚好用来过滤关联实体：

JPQL写法 #

List<A> aList = entityManager.createQuery(
    "SELECT DISTINCT a FROM A a LEFT JOIN FETCH a.bList b ON b.isActive = true",
    A.class
).getResultList();

这里的DISTINCT一定要加，因为LEFT JOIN FETCH会产生笛卡尔积，同一个A会因为关联的B数量重复出现，加DISTINCT可以去重。

Criteria API写法 #

你之前用Criteria的listJoin.on()没成功，可能是没加fetch或者distinct，正确写法如下：

CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<A> query = cb.createQuery(A.class);
Root<A> aRoot = query.from(A.class);

// 关联B并添加过滤条件
ListJoin<A, B> bJoin = aRoot.joinList("bList", JoinType.LEFT);
bJoin.on(cb.equal(bJoin.get("isActive"), true));

// 显式fetch关联，否则还是会懒加载
aRoot.fetch("bList", JoinType.LEFT);

// 去重避免重复的A
query.select(aRoot).distinct(true);

List<A> aList = entityManager.createQuery(query).getResultList();

方案三：DTO投影（适合不需要完整实体的场景） #

如果你的场景不需要返回完整的JPA实体，只需要部分数据，可以用DTO投影的方式，通过子查询直接获取每个A对应的活跃B列表：

1. 先定义一个DTO类：

public class ADto {
    private Long id;
    private List<B> activeBList;

    // 构造函数参数顺序要和JPQL里的字段顺序一致
    public ADto(Long id, List<B> activeBList) {
        this.id = id;
        this.activeBList = activeBList;
    }

    // getter、setter省略
}

2. 执行JPQL查询：

List<ADto> aDtoList = entityManager.createQuery(
    "SELECT new com.yourpackage.ADto(a.id, " +
    "   (SELECT b FROM B b WHERE b.a = a AND b.isActive = true)) " +
    "FROM A a",
    ADto.class
).getResultList();

这个方式不会产生笛卡尔积，也不需要去重，而且是单查询完成，非常适合前端只需要部分数据的场景。

最后提一句 #

如果你的数据量不大，事后用Stream过滤（比如a.getBList().removeIf(b -> !b.getIsActive())）其实也简单，但如果数据量大，还是建议用上面的单查询方案，性能会好很多。

备注：内容来源于stack exchange，提问作者Pioter