JPA CriteriaBuilderを使用して追加のカウント列を含むクエリを作成する方法
Aug 18 2020
データベースにあるすべてのPオブジェクトを返すJPAクエリを作成するのに苦労しています。その隣に、propertyA = 1を持つSの子の数を表示したいと思います。
SQLクエリ
p_tableからp。*、(s_tableからcount(s.id)を選択s WHERE p.id = s.p_idおよびs.propertyA = 1)を選択します。
マッピング:
@Entity
@Table(name = "t_table")
public class PTable{
@Id
private String id;
@Version
private Long version;
private String subject;
@OneToMany(cascade = CascadeType.ALL, orphanRemoval = true, fetch = FetchType.EAGER)
@JoinColumn(name = "p_id", referencedColumnName = "id")
private Set<STable> sSet = new HashSet<>();
}
@Entity
@Table(name = "s_table")
public class STable {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
@Column(name = "p_id")
private String pId;
private String propertyA;
}
また、JPAで複雑なクエリを作成するための優れたチュートリアルを紹介します。
CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<PTable> q = cb.createQuery(PTable.class);
Root<PTable> c = q.from(PTable.class);
回答
JLazar0 Aug 19 2020 at 09:57
CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<MyPojo> cq = cb.createQuery(MyPojo.class);
Root<PTable> rootPTable = cq.from(PTable.class);
Join<PTable, STable> joinSTable = rootPTable.join(PTable_.sSet);
Subquery<Long> sqCount = cq.subquery(Long.class);
Root<STable> sqRootSTable = sqCount.from(STable.class);
Join<STable, PTable> sqJoinPTable = sqRootSTable.join(STable_.pSet);
sqCount.where(cb.and(
cb.equal(sqJoinPTable.get(PTable_.id),rootPTable.get(PTable_.id)),
cb.equal(sqRootSTable.get(STable_.propertyA),"1")));
sqCount.select(cb.count(sqRootSTable));
cq.multiselect(
rootPTable.get(PTable_.id),
rootPTable.get(PTable_.version),
rootPTable.get(PTable_.subject),
joinSTable.get(STable_.id),
sqCount.getSelection(),
);
コンストラクターが持つ結果を取得するには、次のように複数選択パラメーターと順序とタイプが一致するPojoが必要になります。
public MyPojo(String pId, Long version, String subject, Long sId, Long count){
[...]
}
また、関係を正しくマップするようにエンティティを変更する必要があります。次のように、パフォーマンスを向上させるために双方向で遅延します。
PTable
@OneToMany(mappedBy="p",fetch = FetchType.LAZY)
private Set<STable> sSet;
STable
@ManyToOne(fetch = FetchType.LAZY)
@JoinColumn(name="id")
private PTable p;