Implements getItemCount with filters

jpos · Sep 20, 2023 · 27f1f3c · 27f1f3c
1 parent a070780
commit 27f1f3c
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/modules/dbsupport/src/main/java/org/jpos/ee/DBManager.java b/modules/dbsupport/src/main/java/org/jpos/ee/DBManager.java
@@ -55,6 +55,32 @@ public int getItemCount()  {
         return db.session().createQuery(query).getSingleResult().intValue();
     }
 
+    public int getItemCount(DBFilter<T>... filters)  {
+        CriteriaBuilder cb = db.session().getCriteriaBuilder();
+        // CriteriaQuery<Card> query = createCriteriaQuery(false, (DBFilter<Card>) null);
+        CriteriaQuery<Long> countQuery = cb.createQuery(Long.class);
+        Root<T> root = countQuery.from(clazz);
+        Predicate combinedPredicate = cb.and(
+            Arrays.stream(filters)
+                .filter(f -> f != null)
+                .map(f -> f.createPredicate(cb, root))
+                .toArray(Predicate[]::new)
+        );
+
+        Predicate[] mgrFilters = buildFilters(root);
+        if (mgrFilters != null) {
+            Predicate[] nonNullPredicates = Arrays.stream(mgrFilters).filter(f -> f != null).toArray(Predicate[]::new);
+            Predicate mgrPredicate = cb.and(nonNullPredicates);
+            combinedPredicate = cb.and(mgrPredicate, combinedPredicate);
+        }
+
+        countQuery.select(cb.count(root));
+        countQuery.where(combinedPredicate);
+        Long totalCount = db.session().createQuery(countQuery).getSingleResult();
+        return totalCount.intValue();
+
+    }
+
     public List<T> getAll(int offset, int limit, Map<String,Boolean> orders) {
         CriteriaBuilder criteriaBuilder = db.session().getCriteriaBuilder();
         CriteriaQuery<T> query = criteriaBuilder.createQuery(clazz);