[fix] Remove JOIN FETCH when using COUNT query (#68)

Previous versions of Spring still need this check.
darrachequesne · Dec 24, 2017 · b99ed24 · b99ed24
1 parent da86c5d
commit b99ed24
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/main/java/org/springframework/data/jpa/datatables/SpecificationBuilder.java b/src/main/java/org/springframework/data/jpa/datatables/SpecificationBuilder.java
@@ -27,6 +27,11 @@ private class DataTablesSpecification<S> implements Specification<S> {
         public Predicate toPredicate(Root<S> root, CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) {
             initPredicatesRecursively(tree, root, root, criteriaBuilder);
 
+            boolean isCountQuery = query.getResultType() == Long.class;
+            if (isCountQuery) {
+                root.getFetches().clear();
+            }
+
             return createFinalPredicate(criteriaBuilder);
         }