在 JPA 2 中，使用 CriteriaQuery，如何对结果进行计数

java jpa-2.0 criteriaquery

2022-08-31 08:13:59

我对JPA 2很陌生，它是CriteritionBuilder / CriteriaQuery API：

CriteriaQuery javadoc

Java EE 6 教程中的 CriteriaQuery

我想在不实际检索它们的情况下计算 CriteriaQuery 的结果。有没有可能，我没有找到任何这样的方法，唯一的方法是这样做：

CriteriaBuilder cb = entityManager.getCriteriaBuilder();

CriteriaQuery<MyEntity> cq = cb
        .createQuery(MyEntityclass);

// initialize predicates here

return entityManager.createQuery(cq).getResultList().size();

这不可能是正确的方法...

有解决方案吗？

答案 1

类型的查询将返回。您希望查询 .MyEntityMyEntityLong

CriteriaBuilder qb = entityManager.getCriteriaBuilder();
CriteriaQuery<Long> cq = qb.createQuery(Long.class);
cq.select(qb.count(cq.from(MyEntity.class)));
cq.where(/*your stuff*/);
return entityManager.createQuery(cq).getSingleResult();

显然，您将需要使用您在示例中跳过的任何限制和分组等来构建表达式。

答案 2

我已经使用cb.createQuery（）（没有结果类型参数）对此进行了整理：

public class Blah() {

    CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
    CriteriaQuery query = criteriaBuilder.createQuery();
    Root<Entity> root;
    Predicate whereClause;
    EntityManager entityManager;
    Class<Entity> domainClass;

    ... Methods to create where clause ...

    public Blah(EntityManager entityManager, Class<Entity> domainClass) {
        this.entityManager = entityManager;
        this.domainClass = domainClass;
        criteriaBuilder = entityManager.getCriteriaBuilder();
        query = criteriaBuilder.createQuery();
        whereClause = criteriaBuilder.equal(criteriaBuilder.literal(1), 1);
        root = query.from(domainClass);
    }

    public CriteriaQuery<Entity> getQuery() {
        query.select(root);
        query.where(whereClause);
        return query;
    }

    public CriteriaQuery<Long> getQueryForCount() {
        query.select(criteriaBuilder.count(root));
        query.where(whereClause);
        return query;
    }

    public List<Entity> list() {
        TypedQuery<Entity> q = this.entityManager.createQuery(this.getQuery());
        return q.getResultList();
    }

    public Long count() {
        TypedQuery<Long> q = this.entityManager.createQuery(this.getQueryForCount());
        return q.getSingleResult();
    }
}

希望它能帮助:)