adds count implementation at mongodb

Author: otaviojava

mongodb-driver/src/main/java/org/jnosql/diana/mongodb/document/MongoDBDocumentCollectionManager.java

@@ -130,6 +130,13 @@ public List<DocumentEntity> select(DocumentQuery query) {
 
     }
 
+    @Override
+    public long count(String documentCollection) {
+        Objects.requireNonNull(documentCollection, "documentCollection is required");
+        MongoCollection<Document> collection = mongoDatabase.getCollection(documentCollection);
+        return collection.count();
+    }
+
     private Bson getSort(Sort sort) {
         boolean isAscending = Sort.SortType.ASC.equals(sort.getType());
         return isAscending?Sorts.ascending(sort.getName()): Sorts.descending(sort.getName());

mongodb-driver/src/main/java/org/jnosql/diana/mongodb/document/MongoDBDocumentCollectionManagerAsync.java

@@ -35,10 +35,10 @@
 
 import java.time.Duration;
 import java.util.List;
-import java.util.Objects;
 import java.util.concurrent.CopyOnWriteArrayList;
 import java.util.function.Consumer;
 
+import static java.util.Objects.requireNonNull;
 import static org.jnosql.diana.mongodb.document.MongoDBUtils.ID_FIELD;
 import static org.jnosql.diana.mongodb.document.MongoDBUtils.getDocument;
 
@@ -60,7 +60,7 @@ public class MongoDBDocumentCollectionManagerAsync implements DocumentCollection
 
     @Override
     public void insert(DocumentEntity entity) throws ExecuteAsyncQueryException, UnsupportedOperationException {
-        Objects.requireNonNull(entity, "entity is required");
+        requireNonNull(entity, "entity is required");
         insert(entity, v -> {
         });
     }
@@ -74,8 +74,8 @@ public void insert(DocumentEntity entity, Duration ttl) {
     public void insert(DocumentEntity entity, Consumer<DocumentEntity> callBack)
             throws ExecuteAsyncQueryException, UnsupportedOperationException {
 
-        Objects.requireNonNull(entity, "entity is required");
-        Objects.requireNonNull(callBack, "callBack is required");
+        requireNonNull(entity, "entity is required");
+        requireNonNull(callBack, "callBack is required");
         insert(entity, (aVoid, throwable) -> callBack.accept(entity));
     }
 
@@ -87,7 +87,7 @@ public void insert(DocumentEntity entity, Duration ttl, Consumer<DocumentEntity>
     @Override
     public void update(DocumentEntity entity)
             throws ExecuteAsyncQueryException, UnsupportedOperationException {
-        Objects.requireNonNull(entity, "entity is required");
+        requireNonNull(entity, "entity is required");
         update(entity, (d, throwable) -> {
         });
     }
@@ -96,14 +96,14 @@ public void update(DocumentEntity entity)
     public void update(DocumentEntity entity, Consumer<DocumentEntity> callBack)
             throws ExecuteAsyncQueryException, UnsupportedOperationException {
 
-        Objects.requireNonNull(entity, "entity is required");
-        Objects.requireNonNull(callBack, "callBack is required");
+        requireNonNull(entity, "entity is required");
+        requireNonNull(callBack, "callBack is required");
         update(entity, (d, throwable) -> callBack.accept(DocumentEntity.of(entity.getName(), Documents.of(d))));
     }
 
     @Override
     public void delete(DocumentDeleteQuery query) {
-        Objects.requireNonNull(query, "query is required");
+        requireNonNull(query, "query is required");
         delete(query, (deleteResult, throwable) -> {
         });
     }
@@ -112,8 +112,8 @@ public void delete(DocumentDeleteQuery query) {
     public void delete(DocumentDeleteQuery query, Consumer<Void> callBack)
             throws ExecuteAsyncQueryException, UnsupportedOperationException {
 
-        Objects.requireNonNull(query, "query is required");
-        Objects.requireNonNull(callBack, "callBack is required");
+        requireNonNull(query, "query is required");
+        requireNonNull(callBack, "callBack is required");
         delete(query, (deleteResult, throwable) -> callBack.accept(null));
 
     }
@@ -122,8 +122,8 @@ public void delete(DocumentDeleteQuery query, Consumer<Void> callBack)
     public void select(DocumentQuery query, Consumer<List<DocumentEntity>> callBack)
             throws ExecuteAsyncQueryException, UnsupportedOperationException {
 
-        Objects.requireNonNull(query, "query is required");
-        Objects.requireNonNull(callBack, "callBack is required");
+        requireNonNull(query, "query is required");
+        requireNonNull(callBack, "callBack is required");
 
         String collectionName = query.getDocumentCollection();
         MongoCollection<Document> collection = asyncMongoDatabase.getCollection(collectionName);
@@ -145,6 +145,14 @@ public void select(DocumentQuery query, Consumer<List<DocumentEntity>> callBack)
         result.forEach(documentBlock, voidSingleResultCallback);
     }
 
+    @Override
+    public void count(String documentCollection, Consumer<Long> callback) {
+        requireNonNull(documentCollection, "documentCollection is required");
+        requireNonNull(callback, "callback is required");
+        MongoCollection<Document> collection = asyncMongoDatabase.getCollection(documentCollection);
+        collection.count((l, e) -> callback.accept(l));
+    }
+
     private DocumentEntity createEntity(String collectionName, Document document) {
         return DocumentEntity.of(collectionName, MongoDBUtils.of(document));
     }