feat: add count method to DefaultCouchbaseDocumentManager

Implemented a count method that executes a N1QL count query
on the provided SelectQuery, returning the total number of
matching documents in the Couchbase bucket.

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

Author: dearrudam

jnosql-couchbase/src/main/java/org/eclipse/jnosql/databases/couchbase/communication/DefaultCouchbaseDocumentManager.java

@@ -25,6 +25,7 @@
 import com.couchbase.client.java.query.QueryOptions;
 import com.couchbase.client.java.query.QueryResult;
 import org.eclipse.jnosql.communication.semistructured.CommunicationEntity;
+import org.eclipse.jnosql.communication.semistructured.DefaultSelectQuery;
 import org.eclipse.jnosql.communication.semistructured.DeleteQuery;
 import org.eclipse.jnosql.communication.semistructured.Element;
 import org.eclipse.jnosql.communication.semistructured.SelectQuery;
@@ -205,6 +206,26 @@ public long count(String documentCollection) {
         });
     }
 
+    @Override
+    public long count(SelectQuery query) {
+        Objects.requireNonNull(query, "query is required");
+        return waitBucketBeReadyAndGet(() -> {
+            N1QLQuery n1QLQuery = N1QLBuilder.countOf(query, database, bucket.defaultScope().name()).get();
+            QueryResult result;
+            if (n1QLQuery.isParameterEmpty()) {
+                result = cluster.query(n1QLQuery.query());
+            } else {
+                result = cluster.query(n1QLQuery.query(), QueryOptions
+                        .queryOptions().parameters(n1QLQuery.params()));
+            }
+            return result.rowsAsObject()
+                    .stream()
+                    .findFirst()
+                    .map(json -> json.getLong("$1"))
+                    .orElse(0L);
+        });
+    }
+
     @Override
     public Stream<CommunicationEntity> n1qlQuery(final String n1ql, final JsonObject params) throws NullPointerException {
         requireNonNull(n1ql, "n1qlQuery is required");