Merge pull request #7 from jshandorov1/v132

Add count to PersistenceService

Author: bgizdov

pom.xml

@@ -12,7 +12,7 @@
 
     <groupId>in.kuros</groupId>
     <artifactId>jFirebase</artifactId>
-    <version>1.3.1</version>
+    <version>1.3.2</version>
 
 
     <description>This project aims at building a wrapper over nosql technologies - Firebase</description>

src/main/java/in/kuros/jfirebase/PersistenceService.java

@@ -45,6 +45,8 @@ public interface PersistenceService {
 
     List<WriteResult> writeInBatch(Consumer<WriteBatch> batchConsumer);
 
+    <T> long count(Query<T> query);
+
     static void init() {
     }
 }

src/main/java/in/kuros/jfirebase/provider/firebase/PersistenceServiceImpl.java

@@ -278,6 +278,17 @@ public List<WriteResult> writeInBatch(final Consumer<WriteBatch> batchConsumer)
         }
     }
 
+    @Override
+    public <T> long count(Query<T> query) {
+        try {
+            AggregateQuery aggregateQuery = QueryAdapter.toFirebaseQuery(firestore, query).count();
+            final ApiFuture<AggregateQuerySnapshot> querySnapshot = aggregateQuery.get();
+            return querySnapshot.get().getCount();
+        } catch (final Exception e) {
+            throw new PersistenceException(e);
+        }
+    }
+
     @SuppressWarnings("unchecked")
     private <T> Class<T> getClass(final T entity) {
         return (Class<T>) entity.getClass();