首页 > 解决方案 > 列表中的列表。标准 API

问题描述

我有两个实体:

实体:

@Entity
@Data
public class AEntity{
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String name;

}

B实体:

@Entity
@Data
public class BEntity{
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    @ManyToMany(fetch = FetchType.EAGER)
    private List<AEntity> aEntities = new LinkedList();
}

我想通过aEntities使用标准 API 包含 AEntity 名称列表来从数据库中加载所有 BEntities。

例子:

Database:
AEntities -> {1, "A1"}, {2, "A2"}, {3, "A3"}
BEntities -> {1, AEntities: {1, 2, 3}}, {2, AEntities: {1, 3}}, {3, AEntities: {3}}

如果我调用mySearchMethod(List.of("A1", "A3"))它应该返回: {1, AEntities: {1, 2, 3}}, {2, AEntities: {1, 3}} 因为只有这些 BEntities 包含名称为“A1”和“A3”的 AEntities

试图用in表达式来实现这个,但它无法在列表中找到列表:(

标签: javahibernatecriteria-api

解决方案


我还没有尝试过,但它可能会这样工作:

public List<BEntity> mySearchMethod(List<String> keyList){
    CriteriaBuilder cb = em.getCriteriaBuilder();
    CriteriaQuery<BEntity> cq = cb.createQuery(BEntity.class);
    Root<BEntity> r = cq.from(BEntity.class);
        
    cq.select(r);
    ArrayList<Predicate> predicates = new ArrayList<>();
    for(String key : keyList) {
        predicates.add(cb.isMember(key, r.get(BEntity_.aEntities)));
    }
    if(predicates.size() > 0) {
        if(predicates.size() == 1) {
            cq.where(predicates.get(0));
        }else {
            cq.where(cb.and(predicates.toArray(new Predicate[predicates.size()])));
        }
    }
    TypedQuery<BEntity> query = em.createQuery(cq);

    return query.getResultList();
}

推荐阅读