Spring JPA：实体无关联字段时如何用Specification实现关联查询？

我们拥有A、B、C三个实体，需要基于B和C的字段对A表进行查询，但A中未定义与B、C的关联关系，且不能转为双向映射。以下是基于Spring Specification的实现方案：

实体类定义 #

@Entity
public class A {
    @Id
    Long id;
    String name;
}

@Entity
public class B {
    @Id
    Long id;

    @OneToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "a_id")
    A a;
    
    @Column
    String country;
}

@Entity
public class C {
    @Id
    Long id;

    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "a_id")
    private A a;
}

核心思路 #

由于A中没有定义反向关联，无法直接通过root.join("b")的方式关联B/C表。此时可以通过子查询或者直接构建表关联条件两种方式实现，子查询的方式更直观且易于维护。

方案1：基于子查询实现（推荐） #

基于B表的country字段查询A

public static Specification<A> countryEquals(String country) {
    return (root, query, criteriaBuilder) -> {
        // 构造子查询：筛选出符合country条件的B记录对应的A的id
        Subquery<Long> subquery = query.subquery(Long.class);
        Root<B> bRoot = subquery.from(B.class);
        subquery.select(bRoot.get("a").get("id"))
                .where(criteriaBuilder.equal(bRoot.get("country"), country));
        
        // 主查询：A的id存在于子查询结果集中
        return criteriaBuilder.in(root.get("id")).value(subquery);
    };
}

基于C表的字段查询A（以假设C存在status字段为例）

public static Specification<A> cStatusEquals(String status) {
    return (root, query, criteriaBuilder) -> {
        Subquery<Long> subquery = query.subquery(Long.class);
        Root<C> cRoot = subquery.from(C.class);
        subquery.select(cRoot.get("a").get("id"))
                .where(criteriaBuilder.equal(cRoot.get("status"), status));
        
        return criteriaBuilder.in(root.get("id")).value(subquery);
    };
}

方案2：直接构建表关联条件 #

如果需要用join的方式实现，可以手动构建A与B/C的关联条件，注意添加distinct(true)避免重复结果：

public static Specification<A> countryEqualsWithJoin(String country) {
    return (root, query, criteriaBuilder) -> {
        // 从查询中获取B表的Root
        Root<B> bRoot = query.from(B.class);
        // 开启去重，避免A被重复查询
        query.distinct(true);
        
        // 构建关联条件：A.id = B.a.id，同时匹配country
        return criteriaBuilder.and(
                criteriaBuilder.equal(root.get("id"), bRoot.get("a").get("id")),
                criteriaBuilder.equal(bRoot.get("country"), country)
        );
    };
}

组合多条件查询 #

如果需要同时基于B和C的字段查询A，可以用criteriaBuilder.and()组合多个Specification：

public static Specification<A> combineConditions(String country, String cStatus) {
    return Specification.where(countryEquals(country))
            .and(cStatusEquals(cStatus));
}

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