feat: add count method for SelectQuery

Implemented a count method in DefaultArangoDBDocumentManager
to support counting documents based on a SelectQuery,
including null query validation.

Signed-off-by: Maximillian Arruda <[email protected]>

Author: dearrudam

jnosql-arangodb/src/main/java/org/eclipse/jnosql/databases/arangodb/communication/DefaultArangoDBDocumentManager.java

@@ -154,6 +154,14 @@ public long count(String documentCollection) {
                 .map(Number::longValue).orElse(0L);
     }
 
+    @Override
+    public long count(SelectQuery query) {
+        requireNonNull(query, "query is required");
+        checkCollection(query.name());
+        AQLQueryResult aqlQuery = QueryAQLConverter.count(query);
+        LOGGER.finest("Executing AQL: " + aqlQuery.query());
+        return aql(aqlQuery.query(), aqlQuery.values(), Long.class).findFirst().orElse(0L);
+    }
 
     @Override
     public Stream<CommunicationEntity> aql(String query, Map<String, Object> params) throws NullPointerException {

jnosql-arangodb/src/test/java/org/eclipse/jnosql/databases/arangodb/communication/ArangoDBDocumentManagerTest.java

@@ -44,6 +44,7 @@
 import static java.util.Arrays.asList;
 import static java.util.Collections.singletonMap;
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.SoftAssertions.assertSoftly;
 import static org.eclipse.jnosql.communication.driver.IntegrationTest.MATCHES;
 import static org.eclipse.jnosql.communication.driver.IntegrationTest.NAMED;
 import static org.eclipse.jnosql.communication.semistructured.DeleteQuery.delete;
@@ -223,6 +224,22 @@ void shouldCount() {
         assertTrue(entityManager.count(COLLECTION_NAME) > 0);
     }
 
+    @Test
+    void shouldCountWithSelectQuery() {
+        CommunicationEntity entity = entityManager.insert(createDocumentListNotHavingId());
+        Element key = entity.find(KEY_NAME).get();
+        SelectQuery query = select().from("AppointmentBook").where(key.name()).eq(key.get()).build();
+
+        assertSoftly(softly -> {
+            softly.assertThat(entityManager.count(query))
+                    .as("should count documents matching the query")
+                    .isEqualTo(1L);
+            softly.assertThatThrownBy(() -> entityManager.count((SelectQuery) null))
+                    .as("must not accept null query")
+                    .isInstanceOf(NullPointerException.class);
+        });
+    }
+
     @Test
     void shouldReadFromDifferentBaseDocumentUsingInstance() {
         entityManager.insert(getEntity());
@@ -271,7 +288,7 @@ void shouldInsertNull() {
         entity.add(Element.of("name", null));
         CommunicationEntity documentEntity = entityManager.insert(entity);
         Optional<Element> name = documentEntity.find("name");
-        SoftAssertions.assertSoftly(soft -> {
+        assertSoftly(soft -> {
             soft.assertThat(name).isPresent();
             soft.assertThat(name).get().extracting(Element::name).isEqualTo("name");
             soft.assertThat(name).get().extracting(Element::get).isNull();
@@ -284,7 +301,7 @@ void shouldUpdateNull(){
         entity.add(Element.of("name", null));
         var documentEntity = entityManager.update(entity);
         Optional<Element> name = documentEntity.find("name");
-        SoftAssertions.assertSoftly(soft -> {
+        assertSoftly(soft -> {
             soft.assertThat(name).isPresent();
             soft.assertThat(name).get().extracting(Element::name).isEqualTo("name");
             soft.assertThat(name).get().extracting(Element::get).isNull();
@@ -359,7 +376,7 @@ void shouldInsertUUID() {
         entity.add("uuid", UUID.randomUUID());
         var documentEntity = entityManager.insert(entity);
         Optional<Element> uuid = documentEntity.find("uuid");
-        SoftAssertions.assertSoftly(soft -> {
+        assertSoftly(soft -> {
             soft.assertThat(uuid).isPresent();
             Element element = uuid.orElseThrow();
             soft.assertThat(element.name()).isEqualTo("uuid");
@@ -380,7 +397,7 @@ void shouldFindBetween() {
 
         var result = entityManager.select(query).toList();
 
-        SoftAssertions.assertSoftly(softly -> {
+        assertSoftly(softly -> {
             softly.assertThat(result).hasSize(2);
             softly.assertThat(result).map(e -> e.find("age").orElseThrow().get(Integer.class)).contains(22, 23);
             softly.assertThat(result).map(e -> e.find("age").orElseThrow().get(Integer.class)).doesNotContain(25);
@@ -400,7 +417,7 @@ void shouldFindBetween2() {
 
         var result = entityManager.select(query).toList();
 
-        SoftAssertions.assertSoftly(softly -> {
+        assertSoftly(softly -> {
             softly.assertThat(result).hasSize(2);
             softly.assertThat(result).map(e -> e.find("age").orElseThrow().get(Integer.class)).contains(22, 23);
             softly.assertThat(result).map(e -> e.find("age").orElseThrow().get(Integer.class)).doesNotContain(25);
@@ -416,7 +433,7 @@ void shouldFindContains() {
                 "lia")), COLLECTION_NAME, Collections.emptyList());
 
         var result = entityManager.select(query).toList();
-        SoftAssertions.assertSoftly(softly -> {
+        assertSoftly(softly -> {
             softly.assertThat(result).hasSize(1);
             softly.assertThat(result.get(0).find("name").orElseThrow().get(String.class)).isEqualTo("Poliana");
         });
@@ -431,7 +448,7 @@ void shouldStartsWith() {
                 "Pol")), COLLECTION_NAME, Collections.emptyList());
 
         var result = entityManager.select(query).toList();
-        SoftAssertions.assertSoftly(softly -> {
+        assertSoftly(softly -> {
             softly.assertThat(result).hasSize(1);
             softly.assertThat(result.get(0).find("name").orElseThrow().get(String.class)).isEqualTo("Poliana");
         });
@@ -446,7 +463,7 @@ void shouldEndsWith() {
                 "ana")), COLLECTION_NAME, Collections.emptyList());
 
         var result = entityManager.select(query).toList();
-        SoftAssertions.assertSoftly(softly -> {
+        assertSoftly(softly -> {
             softly.assertThat(result).hasSize(1);
             softly.assertThat(result.get(0).find("name").orElseThrow().get(String.class)).isEqualTo("Poliana");
         });
@@ -483,7 +500,6 @@ private CommunicationEntity createDocumentList() {
     }
 
     private CommunicationEntity createDocumentListNotHavingId() {
-        String id = UUID.randomUUID().toString();
         CommunicationEntity entity = CommunicationEntity.of("AppointmentBook");
         entity.add(Element.of("_id", "ids"));
         List<List<Element>> documents = new ArrayList<>();