如何使用相同的criteriaQuery对象获取结果列表并进行选择计数

Question

我使用criteriaBuilder为特定类型创建criteriaQuery。

CriteriaQuery<? extends File> criteriaQuery = criteriaBuilder.createQuery(Folder.class);

我需要在使用EntityManager的createQuery方法时限制查询的结果。

final TypedQuery<? extends File> query = this.entityManager.createQuery(criteriaQuery)
                .setFirstResult(start).setMaxResults(filePerPage);

但我也需要知道有多少结果我可以得到没有限制，因为我需要它来计算页数的分页系统。

所以我使用了另一个criteriaQueryObject来创建一个选择计数

CriteriaQuery<Long> criteriaQueryCount = criteriaBuilder.createQuery(Long.class);

//creation of the criteriaBuilders from specific types
CriteriaQuery<? extends File> criteriaQuery = criteriaBuilder.createQuery(Folder.class);

CriteriaQuery<Long> criteriaQueryCount = criteriaBuilder.createQuery(Long.class);

//creation of the roots
Root<? extends File> root = criteriaQuery.from(Folder.class),
                     rootCount = criteriaQuery.from(Folder.class);

//list of predicates feeded next
List<Predicate> criteriaList = new ArrayList<>(),
                criteriaCountList = new ArrayList<>();

//adding a join and the predicates
CriteriaBuilder.In<Long> inTagIds = criteriaBuilder.in(root.join("tags", JoinType.INNER).get("id"));
CriteriaBuilder.In<Long> inTagIdsCount = criteriaBuilder.in(rootCount.join("tags", JoinType.INNER).get("id"));

tags.forEach(t -> {
       inTagIds.value(t.getId());
       inTagIdsCount.value(t.getId());
});

criteriaList.add(inTagIds);
criteriaCountList.add(inTagIds);

.... I have almost 10 others filters like this one.

//generating the requests
criteriaQuery.where(criteriaBuilder.and(criteriaList.toArray(new Predicate[0])));

criteriaQuery.distinct(true);

criteriaQueryCount.select(criteriaBuilder.count(rootCount));

criteriaQueryCount.where(criteriaBuilder.and(criteriaCountList.toArray(new Predicate[0])));

final TypedQuery<? extends File> query = this.entityManager.createQuery(criteriaQuery)
                .setFirstResult(searchDto.start).setMaxResults(this.filePerPage);

Long count = this.entityManager.createQuery(criteriaQueryCount).getSingleResult();

List<? extends File> searchRs = query.getResultList();

Answer 1

在一个方法中做到这一点的唯一方法是让它使用非类型化的CriteriaQuery（并抑制警告），并返回非类型化的TypedQuery。您可以在执行计数查询时传入一个布尔值，该布尔值告诉它不执行排序等操作。调用方法处理键入，并且必须抑制警告。

希望有一个更优雅的方式，但这是它。它实际上在幕后将所有criteriaquery垃圾转化为JPQL，因此在这种情况下，返回到巨型字符串块几乎更好。