orm - JPA 2.0 : count for arbitrary CriteriaQuery?

Question

我正在尝试实现以下便捷方法:

/**
 * Counts the number of results of a search.
 * @param criteria The criteria for the query.
 * @return The number of results of the query.
 */
public int findCountByCriteria(CriteriaQuery<?> criteria);

在 Hibernate 中，这是由

完成的

criteria.setProjection(Projections.rowCount());

JPA 中相当于上面的内容是什么？我发现了许多简单的计数示例，但它们都没有使用应确定行数的 CriteriaQuery。

编辑:

不幸的是，我发现@Pascal的答案不正确。这个问题非常微妙，只有在使用联接时才会出现:

// Same query, but readable:
// SELECT *
// FROM Brain b
// WHERE b.iq = 170

CriteriaQuery<Person> query = cb.createQuery(Person.class);
Root<Person> root = query.from(Person.class);
Join<Object, Object> brainJoin = root.join("brain");
Predicate iqPredicate = cb.equal(brainJoin.<Integer>get("iq"), 170);
query.select(root).where(iqPredicate);

当调用findCountByCriteria(query)时，它会因以下异常而终止:

org.hibernate.hql.ast.QuerySyntaxException: Invalid path: 'generatedAlias1.iq' [select count(generatedAlias0) from xxx.tests.person.dom.Person as generatedAlias0 where generatedAlias1.iq=170]

还有其他方法可以提供这样的 CountByCriteria 方法吗？

Answer 1

我写了一个实用程序类，JDAL JpaUtils 来做到这一点:

• 计数结果:Long count = JpaUtils.count(em, criteriaQuery);
• 复制条件查询:JpaUtils.copyCriteria(em, criteriaQueryFrom, criteriaQueryTo);
• 获取计数标准:CriteriaQuery<Long> countCriteria = JpaUtils.countCriteria(em, criteria)

等等...

如果您对源代码感兴趣，请参阅JpaUtils.java