feat: add default execute and count methods

Implemented default methods for execute and count in
QueryExecutor interface to streamline query execution
and counting operations without requiring consistency level.

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

Author: dearrudam

jnosql-cassandra/src/main/java/org/eclipse/jnosql/databases/cassandra/communication/QueryExecutor.java

@@ -29,9 +29,17 @@ static QueryExecutor of(SelectQuery query) {
         return QueryExecutorType.DEFAULT;
     }
 
-    Stream<CommunicationEntity> execute(String keyspace, SelectQuery query, DefaultCassandraColumnManager manager);
+    default Stream<CommunicationEntity> execute(String keyspace, SelectQuery query, DefaultCassandraColumnManager manager) {
+        return execute(keyspace, query, null, manager);
+    }
 
     Stream<CommunicationEntity> execute(String keyspace, SelectQuery query, ConsistencyLevel level,
-                                 DefaultCassandraColumnManager manager);
+                                        DefaultCassandraColumnManager manager);
+
+    default long count(String keyspace, SelectQuery query, DefaultCassandraColumnManager manager) {
+        return count(keyspace, query, null, manager);
+    }
+
+    long count(String keyspace, SelectQuery query, ConsistencyLevel level, DefaultCassandraColumnManager manager);
 
 }

jnosql-cassandra/src/main/java/org/eclipse/jnosql/databases/cassandra/communication/QueryExecutorType.java

@@ -30,14 +30,9 @@
 
 enum QueryExecutorType implements QueryExecutor {
     PAGING_STATE {
-        @Override
-        public Stream<CommunicationEntity> execute(String keyspace, SelectQuery query, DefaultCassandraColumnManager manager) {
-            return execute(keyspace, query, null, manager);
-        }
-
         @Override
         public Stream<CommunicationEntity> execute(String keyspace, SelectQuery q, ConsistencyLevel level,
-                                            DefaultCassandraColumnManager manager) {
+                                                   DefaultCassandraColumnManager manager) {
 
             CassandraQuery query = CassandraQuery.class.cast(q);
 
@@ -73,14 +68,9 @@ public Stream<CommunicationEntity> execute(String keyspace, SelectQuery q, Consi
 
     },
     DEFAULT {
-        @Override
-        public Stream<CommunicationEntity> execute(String keyspace, SelectQuery query, DefaultCassandraColumnManager manager) {
-            return execute(keyspace, query, null, manager);
-        }
-
         @Override
         public Stream<CommunicationEntity> execute(String keyspace, SelectQuery query, ConsistencyLevel level,
-                                            DefaultCassandraColumnManager manager) {
+                                                   DefaultCassandraColumnManager manager) {
 
             Select cassandraSelect = QueryUtils.select(query, keyspace);
 
@@ -98,5 +88,17 @@ public Stream<CommunicationEntity> execute(String keyspace, SelectQuery query, C
             }
             return resultSet.all().stream().map(CassandraConverter::toDocumentEntity);
         }
+    };
+
+    @Override
+    public long count(String keyspace, SelectQuery query, ConsistencyLevel level, DefaultCassandraColumnManager manager) {
+
+        Select cassandraSelect = QueryUtils.select(query, keyspace).countAll();
+        SimpleStatement select = cassandraSelect.build();
+        if (Objects.nonNull(level)) {
+            select = select.setConsistencyLevel(level);
+        }
+        ResultSet resultSet = manager.getSession().execute(select);
+        return resultSet.one().getLong(0);
     }
 }