adds arangodb implementation

Author: otaviojava

arangodb-driver/src/main/java/org/jnosql/diana/arangodb/document/DefaultArangoDBDocumentCollectionManager.java

@@ -28,8 +28,10 @@
 import org.jnosql.diana.api.writer.ValueWriterDecorator;
 
 import java.time.Duration;
+import java.util.Collections;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Optional;
 import java.util.stream.StreamSupport;
 
@@ -103,6 +105,14 @@ public List<DocumentEntity> select(DocumentQuery query) throws NullPointerExcept
                 .collect(toList());
     }
 
+    @Override
+    public long count(String documentCollection) {
+        Objects.requireNonNull(documentCollection, "document collection is required");
+        String aql = "RETURN LENGTH(" + documentCollection + ")";
+        ArangoCursor<Object> query = arangoDB.db(database).query(aql, Collections.emptyMap(), null, Object.class);
+        return StreamSupport.stream(query.spliterator(), false).findFirst().map(Long.class::cast).orElse(0L);
+    }
+
 
     @Override
     public List<DocumentEntity> aql(String query, Map<String, Object> values) throws NullPointerException {

arangodb-driver/src/main/java/org/jnosql/diana/arangodb/document/DefaultArangoDBDocumentCollectionManagerAsync.java

@@ -14,6 +14,7 @@
  */
 package org.jnosql.diana.arangodb.document;
 
+import com.arangodb.ArangoCursor;
 import com.arangodb.ArangoCursorAsync;
 import com.arangodb.ArangoDB;
 import com.arangodb.ArangoDBAsync;
@@ -27,8 +28,10 @@
 import org.jnosql.diana.api.document.DocumentQuery;
 
 import java.time.Duration;
+import java.util.Collections;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.concurrent.CompletableFuture;
 import java.util.function.Consumer;
 import java.util.stream.StreamSupport;
@@ -144,6 +147,20 @@ public void select(DocumentQuery query, Consumer<List<DocumentEntity>> callBack)
 
     }
 
+    @Override
+    public void count(String documentCollection, Consumer<Long> callback) {
+        Objects.requireNonNull(documentCollection, "document collection is required");
+        Objects.requireNonNull(callback, "callback is required");
+        String aql = "RETURN LENGTH(" + documentCollection + ")";
+        CompletableFuture<ArangoCursorAsync<Object>> query = arangoDBAsync.db(database).query(aql, Collections.emptyMap(),
+                null, Object.class);
+        query.thenAccept(r -> {
+            long count = StreamSupport.stream(r.spliterator(), false).findFirst().map(Long.class::cast).orElse(0L);
+            callback.accept(count);
+        });
+
+    }
+
 
     @Override
     public void aql(String query, Map<String, Object> values, Consumer<List<DocumentEntity>> callBack)
@@ -186,5 +203,4 @@ private void createConsumer(DocumentEntity entity, Consumer<DocumentEntity> call
     }
 
 
-
 }

arangodb-driver/src/test/java/org/jnosql/diana/arangodb/document/ArangoDBDocumentCollectionManagerAsyncTest.java

@@ -28,13 +28,17 @@
 import java.util.Map;
 import java.util.Random;
 import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicLong;
 import java.util.concurrent.atomic.AtomicReference;
+import java.util.function.Consumer;
 
 import static org.awaitility.Awaitility.await;
 import static org.jnosql.diana.api.document.query.DocumentQueryBuilder.delete;
 import static org.jnosql.diana.api.document.query.DocumentQueryBuilder.select;
 import static org.jnosql.diana.arangodb.document.DocumentConfigurationUtils.getConfiguration;
+import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 
 public class ArangoDBDocumentCollectionManagerAsyncTest {
@@ -147,6 +151,24 @@ public void shouldRunAQL() {
         assertFalse(entities.isEmpty());
     }
 
+    @Test
+    public void shouldCount() {
+        DocumentEntity entity = getEntity();
+        AtomicBoolean condition = new AtomicBoolean();
+        AtomicLong value = new AtomicLong(0L);
+        entityManagerAsync.insert(entity, d -> condition.set(true));
+
+        await().untilTrue(condition);
+        condition.set(false);
+        Consumer<Long> callback = l -> {
+            condition.set(true);
+            value.set(l);
+        };
+        entityManagerAsync.count(COLLECTION_NAME, callback);
+        await().untilTrue(condition);
+        assertTrue(value.get() > 0);
+
+    }
 
     private DocumentEntity getEntity() {
         DocumentEntity entity = DocumentEntity.of(COLLECTION_NAME);

arangodb-driver/src/test/java/org/jnosql/diana/arangodb/document/ArangoDBDocumentCollectionManagerTest.java

@@ -188,6 +188,14 @@ private DocumentEntity createSubdocumentList() {
         return entity;
     }
 
+    @Test
+    public void shouldCount() {
+        DocumentEntity entity = getEntity();
+        DocumentEntity entitySaved = entityManager.insert(entity);
+
+        assertTrue(entityManager.count(COLLECTION_NAME) > 0);
+    }
+
 
     private DocumentEntity getEntity() {
         DocumentEntity entity = DocumentEntity.of(COLLECTION_NAME);