java - JPA Criteria 查询计数结果涉及连接

Question

我正在使用 hibernate 和 JPA 标准 API，并尝试创建一个可重用的实用方法来确定查询将返回多少行。

目前我有这个:

Long countResults(CriteriaQuery cq, String alias){
    CriteriaBuilder cb = em().getCriteriaBuilder();
    CriteriaQuery<Long> countQuery = cb.createQuery(Long.class);
    Root ent = countQuery.from(cq.getResultType());
    ent.alias(alias);
    countQuery.select(cb.count(ent));
    Predicate restriction = cq.getRestriction();
    if(restriction != null){
        countQuery.where(restriction);
    }
    return em().createQuery(countQuery).getSingleResult();
}

我像这样使用:

CriteriaBuilder cb = em().getCriteriaBuilder();
CriteriaQuery<User> cq = cb.createQuery(User.class);
Root<User> root = cq.from(modelClass());
root.alias("ct");
cq.select(root);

TypedQuery<User> query = em().createQuery(cq);
long count = countResults(cq, "ct");

这很好用。
但是，当我使用更复杂的查询时

Join<UserThing, Thing> j = root.join(User_.things).join(UserThing_.thing);
cq.where(somePredicate);

我的电话 countResults()产生异常，如 org.hibernate.hql.internal.ast.InvalidPathException: Invalid path: 'myAlias.name' , <AST>:0:0: unexpected end of subtree , left-hand operand of a binary operator was null
我猜这与加入有关，并且我需要以某种方式对其进行别名，但到目前为止我还没有取得任何成功。

帮助？

Answer 1

我有同样的问题，我解决了:

CriteriaQuery<Long> countCriteria = cb.createQuery(Long.class);
Root<EntityA> countRoot = countCriteria.from(cq.getResultType());
Set<Join<EntityA, ?>> joins = originalEntityRoot.getJoins();
for (Join<EntityA, ?> join :  joins) {
    countRoot.join(join.getAttribute().getName());
}
countCriteria.select(cb.count(countRoot));
if(finalPredicate != null)
    countCriteria.where(finalPredicate);

TypedQuery<Long> queryCount = entityManager.createQuery(countCriteria);
Long count = queryCount.getSingleResult();

在哪里

originalEntityRoot 是我使用 where 子句进行查询的主根。