feat: add count method based on query to CassandraColumnManager

Implemented a count method in CassandraColumnManager to
allow counting records based on a query and consistency level.
Added corresponding test to verify functionality.

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

Author: dearrudam

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

@@ -101,6 +101,17 @@ public interface CassandraColumnManager extends DatabaseManager {
      */
     Stream<CommunicationEntity> select(SelectQuery query, ConsistencyLevel level) throws NullPointerException;
 
+    /**
+     * Count based on a query using a consistency level
+     *
+     * @param query the query
+     * @param level the consistency level
+     * @return the query using a consistency level
+     * @throws NullPointerException when either query or level are null
+     */
+    long count(SelectQuery query, ConsistencyLevel level) throws NullPointerException;
+
+
     /**
      * Executes CQL
      *

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

@@ -159,6 +159,17 @@ public long count(String columnFamily) {
         return execute.one().getLong(0);
     }
 
+    @Override
+    public long count(SelectQuery query) {
+        return count(query, null);
+    }
+
+    @Override
+    public long count(SelectQuery query, ConsistencyLevel level) {
+        requireNonNull(query, "query is required");
+        QueryExecutor executor = QueryExecutor.of(query);
+        return executor.count(keyspace, query, level, this);
+    }
 
     @Override
     public void close() {

jnosql-cassandra/src/test/java/org/eclipse/jnosql/databases/cassandra/communication/CassandraColumnManagerTest.java

@@ -451,6 +451,14 @@ void shouldCount() {
         assertTrue(contacts > 0);
     }
 
+    @Test
+    void shouldCountBySelectQuery() {
+        entityManager.insert(getColumnFamily());
+        var query = select().from(Constants.COLUMN_FAMILY).where("id").eq(10L).build();
+        assertThat(entityManager.count(query))
+                .isEqualTo(1);
+    }
+
    @Test
     void shouldPagingState() {
         for (long index = 1; index <= 10; index++) {