CCMSPUI-379: Update count query to be more efficient

Signed-off-by: Jamie Briggs <jamie.briggs@digital.justice.gov.uk>
ministryofjustice · Feb 20, 2025 · c045010 · c045010
1 parent 0400762
commit c045010
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data-service/src/main/java/uk/gov/laa/ccms/data/repository/BaseEntityManagerRepository.java b/data-service/src/main/java/uk/gov/laa/ccms/data/repository/BaseEntityManagerRepository.java
@@ -63,7 +63,7 @@ private void applySortingClause(CriteriaQuery<T> query, Pageable pageable,
   private long getCount(Specification<T> specification, CriteriaBuilder criteriaBuilder) {
     CriteriaQuery<Long> countQuery = criteriaBuilder.createQuery(Long.class);
     Root<T> countRoot = countQuery.from(getEntityClazz());
-    countQuery.select(criteriaBuilder.count(countRoot));
+    countQuery.select(criteriaBuilder.count(countRoot.get("id")));
     applyWhereClause(countQuery, specification, criteriaBuilder, countRoot);
     return entityManager.createQuery(countQuery).getSingleResult();
   }