Skip total count if select result is empty

Author: igdianov

schema/src/main/java/com/introproventures/graphql/jpa/query/schema/impl/GraphQLJpaQueryDataFetcher.java

@@ -34,6 +34,7 @@
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import graphql.schema.GraphQLScalarType;
+import java.util.Collection;
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
@@ -117,7 +118,12 @@ public PagedResult<Object> get(DataFetchingEnvironment environment) {
         }
 
         if (totalSelection.isPresent() || pagesSelection.isPresent()) {
-            final Long total = queryFactory.queryTotalCount(environment, restrictedKeys);
+            final var selectResult = pagedResult.getSelect();
+
+            final long total = recordsSelection.isEmpty() ||
+                selectResult.filter(Predicate.not(Collection::isEmpty)).isPresent()
+                ? queryFactory.queryTotalCount(environment, restrictedKeys)
+                : 0L;
 
             pagedResult.withTotal(total);
         }

schema/src/main/java/com/introproventures/graphql/jpa/query/schema/impl/PagedResult.java

@@ -20,6 +20,7 @@
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 
 public class PagedResult<T> {
 
@@ -135,6 +136,10 @@ public Builder<T> withSelect(List<T> select) {
             return this;
         }
 
+        public Optional<List<T>> getSelect() {
+            return Optional.ofNullable(select);
+        }
+
         /**
          * Builder method for select parameter.
          * @param select field to set

schema/src/main/java/com/introproventures/graphql/jpa/query/schema/impl/ResultStreamWrapper.java

@@ -63,6 +63,8 @@ public Object invoke(Object proxy, Method method, Object[] args) throws Throwabl
                 return System.identityHashCode(proxy);
             } else if ("spliterator".equals(method.getName())) {
                 return stream.spliterator();
+            } else if ("isEmpty".equals(method.getName())) {
+                return size == 0;
             }
             throw new UnsupportedOperationException(method + " is not supported");
         }