feat: add count method to DefaultTinkerpopGraphDatabaseManager

Implemented a count method for SelectQuery to enhance
query capabilities and improve performance in counting
entities within the graph database.

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

Author: dearrudam

jnosql-tinkerpop/src/main/java/org/eclipse/jnosql/databases/tinkerpop/communication/DefaultTinkerpopGraphDatabaseManager.java

@@ -127,37 +127,32 @@ public Iterable<CommunicationEntity> update(Iterable<CommunicationEntity> entiti
     public void delete(DeleteQuery query) {
         Objects.requireNonNull(query, "delete is required");
         GraphTraversal<Vertex, Vertex> traversal = graph.traversal().V().hasLabel(query.name());
-        query.condition().ifPresent(c ->{
+        query.condition().ifPresent(c -> {
             GraphTraversal<Vertex, Vertex> predicate = TraversalExecutor.getPredicate(c);
             traversal.filter(predicate);
         });
 
-       traversal.drop().iterate();
+        traversal.drop().iterate();
         GraphTransactionUtil.transaction(graph);
     }
 
     @Override
     public Stream<CommunicationEntity> select(SelectQuery query) {
-        Objects.requireNonNull(query, "query is required");
-        GraphTraversal<Vertex, Vertex> traversal = graph.traversal().V().hasLabel(query.name());
-        query.condition().ifPresent(c ->{
-            GraphTraversal<Vertex, Vertex> predicate = TraversalExecutor.getPredicate(c);
-            traversal.filter(predicate);
-        });
+        GraphTraversal<Vertex, Vertex> traversal = buildGraphTraversalOf(query);
 
-        if(query.limit()> 0) {
+        if (query.limit() > 0) {
             traversal.limit(query.limit());
-        } else if(query.skip() > 0) {
+        } else if (query.skip() > 0) {
             traversal.skip(query.skip());
         }
-       query.sorts().forEach(
-               s -> {
-                   if (s.isAscending()) {
-                       traversal.order().by(s.property(), asc);
-                   } else {
-                       traversal.order().by(s.property(), desc);
-                   }
-               });
+        query.sorts().forEach(
+                s -> {
+                    if (s.isAscending()) {
+                        traversal.order().by(s.property(), asc);
+                    } else {
+                        traversal.order().by(s.property(), desc);
+                    }
+                });
         return traversal.toStream().map(CommunicationEntityConverter.INSTANCE);
     }
 
@@ -168,6 +163,22 @@ public long count(String entity) {
         return count.next();
     }
 
+    @Override
+    public long count(SelectQuery query) {
+        GraphTraversal<Vertex, Vertex> traversal = buildGraphTraversalOf(query);
+        return traversal.count().next();
+    }
+
+    private GraphTraversal<Vertex, Vertex> buildGraphTraversalOf(SelectQuery query) {
+        Objects.requireNonNull(query, "query is required");
+        GraphTraversal<Vertex, Vertex> traversal = graph.traversal().V().hasLabel(query.name());
+        query.condition().ifPresent(c -> {
+            GraphTraversal<Vertex, Vertex> predicate = TraversalExecutor.getPredicate(c);
+            traversal.filter(predicate);
+        });
+        return traversal;
+    }
+
     @Override
     public void close() {
         try {