过滤器连接多个表的规范，包括嵌入的主键

Question

我想构建一个Specification 来过滤不直接包含在实体中但在“第二次连接”中的属性。

实体看起来（非常简化）如下：

@Entity
public class A {
    @Id
    private UUID id;

    @OneToMany(mappedBy=pk.a)
    private List<B> b;
}

@Entity
public class B {
    @EmbeddedId
    private PK pk;
}

@Embeddable
public class PK {
    @ManyToOne
    private A a;
    
    @ManyToOne
    private C c;
}

@Entity
public class C {
    @Id
    private UUID id;

    @OneToMany(mappedBy=pk.b)
    private List<B> b;
}

我想获取所有具有特定 C.id 的 A，并且我已经有一个按名称工作的存储库方法：

List<A> findAllByBPKCId(String id);

由于我需要更多过滤器，因此我需要移至 JpaSpecificationExecutor，但我无法构建 Specification 以执行与以前的存储库方法相同的操作。

到目前为止，我尝试的是： 调用

List<A> findAll(Specification specification);

与

new Specification() {
    @Override
    public Predicate toPredicate(Root root, CriteriaQuery query, CriteriaBuilder criteraBuilder) {
        return criteriaBuilder.equal(root.join("b").join(pk.c).get("id"), ID_TO_COMPARE);
    }
}

但是它不会找到“pk.c”，我需要在加入 c 之前先以某种方式“获取”pk。它会引发 IllegalArgumentException，因为它“无法找到具有给定名称 [pk.b] 的属性”。

如何加入嵌套在嵌入式“复合主键”中的实体？

Answer 1

你需要先加入B和PK，然后加入C，即PK在加入的时候应该和其他实体一样考虑。

请考虑以下代码：

new Specification() {
    @Override
    public Predicate toPredicate(Root root, CriteriaQuery query, CriteriaBuilder criteraBuilder) {
        Join<A, B> joinAB = root.join("b");
        Join<B, PK> joinBPK = joinAB.join("pk");
        Join<PK, C> joinPKC = joinBPK.join("c");

        return criteraBuilder.equal(joinPKC.get("id"), cb.literal(ID_TO_COMPARE));
    }
}

可以使用 lambdas 简化此代码：

public static Specification<A> getSpecification(String ID_TO_COMPARE) {
  return (root, query, cb) -> {
    Join<A, B> joinAB = root.join("b");
    Join<B, PK> joinBPK = joinAB.join("pk");
    Join<PK, C> joinPKC = joinBPK.join("c");

    return cb.equal(joinPKC.get("id"), cb.literal(ID_TO_COMPARE));
  };
}

另外，如果你有机会，使用criteria metamodel，它在构造你的查询时提供编译类型字段检查，让事情变得清晰：

public static Specification<A> getSpecification(String ID_TO_COMPARE) {
  return (root, query, cb) -> {
    Join<A, B> joinAB = root.join(A_.b);
    Join<B, PK> joinBPK = joinAB.join(B_.pk);
    Join<PK, C> joinPKC = joinBPK.join(PK_.c);

    return cb.equal(joinPKC.get(C_.id), cb.literal(ID_TO_COMPARE));
  };
}