adds counter

Author: otaviojava

cassandra-driver/src/main/java/org/jnosql/diana/cassandra/column/DefaultCassandraColumnFamilyManagerAsync.java

@@ -15,6 +15,7 @@
 package org.jnosql.diana.cassandra.column;
 
 import com.datastax.driver.core.ConsistencyLevel;
+import com.datastax.driver.core.ResultSet;
 import com.datastax.driver.core.ResultSetFuture;
 import com.datastax.driver.core.Session;
 import com.datastax.driver.core.Statement;
@@ -29,6 +30,7 @@
 import java.time.Duration;
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.ExecutionException;
 import java.util.concurrent.Executor;
 import java.util.function.Consumer;
 import java.util.stream.StreamSupport;
@@ -223,6 +225,16 @@ public void select(ColumnQuery query, Consumer<List<ColumnEntity>> consumer)
     public void count(String columnFamily, Consumer<Long> callback) {
         requireNonNull(columnFamily, "columnFamily is required");
         requireNonNull(callback, "callback is required");
+        ResultSetFuture resultSet = session.executeAsync(QueryUtils.count(columnFamily, keyspace));
+        Runnable counter = () -> {
+            try {
+                Object object = resultSet.get().one().getObject(0);
+                callback.accept(Number.class.cast(object).longValue());
+            } catch (InterruptedException | ExecutionException e) {
+                throw new ExecuteAsyncQueryException(e);
+            }
+        };
+        resultSet.addListener(counter, executor);
     }
 
     @Override

cassandra-driver/src/test/java/org/jnosql/diana/cassandra/column/CassandraColumnFamilyManagerAsyncTest.java

@@ -33,6 +33,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicLong;
 import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Consumer;
 
@@ -307,6 +308,23 @@ public void shouldInsertColumnsAsyncWithConsistenceLevel() {
         columnEntityManager.save(columnEntity, CONSISTENCY_LEVEL);
     }
 
+    @Test
+    public void shouldCount() {
+        ColumnEntity entity = getColumnFamily();
+        columnEntityManager.insert(entity);
+        AtomicLong counter = new AtomicLong(0);
+        AtomicBoolean condition = new AtomicBoolean(false);
+
+        Consumer<Long> callback = (l) ->{
+            condition.set(true);
+            counter.set(l);
+        };
+        columnEntityManager.count(COLUMN_FAMILY, callback);
+        await().untilTrue(condition);
+        assertTrue(counter.get() > 0);
+    }
+
+
     private ColumnEntity getColumnFamily() {
         Map<String, Object> fields = new HashMap<>();
         fields.put("name", "Cassandra");