如何計算子查詢返回的行數？

Question

我想要做這樣的事情：

select count(*) from (select ...) 

（因爲這將是SQL），但是在JPA。

上我會怎麼做任何想法？

Answer 1

這應該做的伎倆（如果你想使用JPA標準API）：

CriteriaBuilder cb = getEntityManager().getCriteriaBuilder(); 
CriteriaQuery<Long> query = cb.createQuery(Long.class); 

Root<Entity> root = query.from(Entity.class); 

//Selecting the count 
query.select(cb.count(root)); 

//Create your search criteria 
Criteria criteria = ... 

//Adding search criteria 
query.where(criteria); 

Long count = getEntityManager().createQuery(query).getSingleResult(); 

在另一方面，如果你想使用JP-QL，下面的代碼應該做的伎倆：

//Add the where condition to the end of the query 
Query query = getEntityManager().createQuery("select count(*) from Entity entity where...") 
Long count = query.getSingleResult(); 

Answer 2

我在這個問題上絆倒也。我最終希望執行以下JPQL：

SELECT COUNT(u) 
FROM (
    SELECT DISTINCT u 
    FROM User u 
    JOIN u.roles r 
    WHERE r.id IN (1) 
) 

但是，這是不可能的，也不符合標準API。研究表明，這只是JPA的設計限制。 JPA規範規定，子查詢只在WHERE和HAVING條款支持（因而在FROM沒有）。

重寫下面的JPQL形式查詢：

SELECT COUNT(u) 
FROM User u 
WHERE u IN (
    SELECT DISTINCT u 
    FROM User u 
    JOIN u.roles r 
    WHERE r.id IN (1) 
) 

使用JPA標準API類似如下：

CriteriaQuery<Long> query = cb.createQuery(Long.class); 
Root<User> u = query.from(User.class); 
Subquery<User> subquery = query.subquery(User.class); 
Root<User> u_ = subquery.from(User.class); 
subquery.select(u_).distinct(true).where(u_.join("roles").get("id").in(Arrays.asList(1L))); 
query.select(cb.count(u)).where(cb.in(u).value(subquery)); 
Long count = entityManager.createQuery(query).getSingleResult(); 
// ... 

已經解決了這個功能要求我。這也應該讓您充分了解解決您的特定功能需求。