adds test to ES

Author: otaviojava

elasticsearch-driver/src/main/java/org/jnosql/diana/elasticsearch/document/CountActionListener.java

@@ -0,0 +1,28 @@
+package org.jnosql.diana.elasticsearch.document;
+
+import org.elasticsearch.action.ActionListener;
+import org.elasticsearch.action.search.SearchResponse;
+import org.jnosql.diana.api.JNoSQLException;
+
+import java.util.function.Consumer;
+
+final class CountActionListener implements ActionListener<SearchResponse> {
+
+    private final Consumer<Long> callback;
+    private final String documentCollection;
+
+    CountActionListener(Consumer<Long> callback, String documentCollection) {
+        this.callback = callback;
+        this.documentCollection = documentCollection;
+    }
+
+    @Override
+    public void onResponse(SearchResponse response) {
+        callback.accept(response.getHits().getTotalHits());
+    }
+
+    @Override
+    public void onFailure(Exception e) {
+        throw new ElasticsearchException("An error when do search from QueryBuilder on elasticsearch", e);
+    }
+}

elasticsearch-driver/src/main/java/org/jnosql/diana/elasticsearch/document/DefaultElasticsearchDocumentCollectionManager.java

@@ -22,7 +22,9 @@
 import org.elasticsearch.action.search.SearchResponse;
 import org.elasticsearch.client.RestHighLevelClient;
 import org.elasticsearch.index.query.QueryBuilder;
+import org.elasticsearch.index.query.QueryBuilders;
 import org.elasticsearch.search.builder.SearchSourceBuilder;
+import org.jnosql.diana.api.JNoSQLException;
 import org.jnosql.diana.api.document.Document;
 import org.jnosql.diana.api.document.DocumentDeleteQuery;
 import org.jnosql.diana.api.document.DocumentEntity;
@@ -116,6 +118,21 @@ public List<DocumentEntity> select(DocumentQuery query) throws NullPointerExcept
         return EntityConverter.query(query, client, index);
     }
 
+    @Override
+    public long count(String documentCollection) {
+        Objects.requireNonNull(documentCollection, "query is required");
+        SearchRequest searchRequest = new SearchRequest(index);
+        searchRequest.types(documentCollection);
+        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
+        searchSourceBuilder.size(0);
+        try {
+            SearchResponse search = client.search(searchRequest);
+            return search.getHits().getTotalHits();
+        } catch (IOException e) {
+           throw new JNoSQLException("Error on ES when try to execute count to document collection:" + documentCollection, e);
+        }
+    }
+
     @Override
     public List<DocumentEntity> search(QueryBuilder query, String... types) throws NullPointerException {
         Objects.requireNonNull(query, "query is required");

elasticsearch-driver/src/main/java/org/jnosql/diana/elasticsearch/document/DefaultElasticsearchDocumentCollectionManagerAsync.java

@@ -21,10 +21,12 @@
 import org.elasticsearch.action.delete.DeleteRequest;
 import org.elasticsearch.action.index.IndexRequest;
 import org.elasticsearch.action.search.SearchRequest;
+import org.elasticsearch.action.search.SearchResponse;
 import org.elasticsearch.client.RestHighLevelClient;
 import org.elasticsearch.index.query.QueryBuilder;
 import org.elasticsearch.search.builder.SearchSourceBuilder;
 import org.jnosql.diana.api.ExecuteAsyncQueryException;
+import org.jnosql.diana.api.JNoSQLException;
 import org.jnosql.diana.api.document.Document;
 import org.jnosql.diana.api.document.DocumentDeleteQuery;
 import org.jnosql.diana.api.document.DocumentEntity;
@@ -111,7 +113,7 @@ public void delete(DocumentDeleteQuery query, Consumer<Void> callBack) {
         DocumentQuery select = new ElasticsearchDocumentQuery(query);
 
         List<DocumentEntity> entities = EntityConverter.query(select, client, index);
-        if(entities.isEmpty()) {
+        if (entities.isEmpty()) {
             callBack.accept(null);
             return;
         }
@@ -139,10 +141,22 @@ public void onFailure(Exception e) {
     }
 
     @Override
-    public void select(DocumentQuery query, Consumer<List<DocumentEntity>> callBack) {
+    public void select(DocumentQuery query, Consumer<List<DocumentEntity>> callback) {
         requireNonNull(query, "query is required");
-        requireNonNull(callBack, "callBack is required");
-        EntityConverter.queryAsync(query, client, index, callBack);
+        requireNonNull(callback, "callback is required");
+        EntityConverter.queryAsync(query, client, index, callback);
+    }
+
+    @Override
+    public void count(String documentCollection, Consumer<Long> callback) {
+        requireNonNull(documentCollection, "documentCollection is required");
+        requireNonNull(callback, "callback is required");
+        SearchRequest searchRequest = new SearchRequest(index);
+        searchRequest.types(documentCollection);
+        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
+        searchSourceBuilder.size(0);
+        ActionListener<SearchResponse> listener = new CountActionListener(callback, documentCollection);
+        client.searchAsync(searchRequest, listener);
     }
 
 

elasticsearch-driver/src/test/java/org/jnosql/diana/elasticsearch/document/ElasticsearchDocumentCollectionManagerAsyncTest.java

@@ -28,14 +28,17 @@
 import java.time.Duration;
 import java.util.List;
 import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicLong;
 import java.util.concurrent.atomic.AtomicReference;
+import java.util.function.Consumer;
 
 import static org.elasticsearch.index.query.QueryBuilders.termQuery;
 import static org.jnosql.diana.api.document.query.DocumentQueryBuilder.delete;
 import static org.jnosql.diana.api.document.query.DocumentQueryBuilder.select;
 import static org.jnosql.diana.elasticsearch.document.DocumentEntityGerator.COLLECTION_NAME;
 import static org.jnosql.diana.elasticsearch.document.DocumentEntityGerator.INDEX;
 import static org.jnosql.diana.elasticsearch.document.DocumentEntityGerator.getEntity;
+import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
@@ -104,9 +107,10 @@ public void shouldRemoveEntityAsync() throws InterruptedException {
     }
 
     @Test
-    public void shouldUserSearchBuilder() {
+    public void shouldUserSearchBuilder() throws InterruptedException {
         DocumentEntity entity = getEntity();
         entityManager.insert(entity);
+        Thread.sleep(1_000L);
         TermQueryBuilder query = termQuery("name", "Poliana");
         AtomicReference<List<DocumentEntity>> result = new AtomicReference<>();
         AtomicBoolean atomicBoolean = new AtomicBoolean(false);
@@ -122,9 +126,10 @@ public void shouldUserSearchBuilder() {
     }
 
     @Test
-    public void shouldReturnAll() {
+    public void shouldReturnAll() throws InterruptedException {
         DocumentEntity entity = getEntity();
         entityManagerAsync.insert(entity);
+        Thread.sleep(1_000L);
         DocumentQuery query = select().from(COLLECTION_NAME).build();
         AtomicBoolean condition = new AtomicBoolean(false);
         AtomicReference<List<DocumentEntity>> result = new AtomicReference<>();
@@ -139,6 +144,23 @@ public void shouldReturnAll() {
 
     }
 
+    @Test
+    public void shouldCount() throws InterruptedException {
+        AtomicBoolean condition = new AtomicBoolean(false);
+        AtomicLong value = new AtomicLong(0L);
+        DocumentEntity entity = getEntity();
+        entityManagerAsync.insert(entity);
+
+        Thread.sleep(1_000L);
+        Consumer<Long> callback = l -> {
+            condition.set(true);
+            value.set(l);
+        };
+        entityManagerAsync.count(DocumentEntityGerator.COLLECTION_NAME, callback);
+        Awaitility.await().untilTrue(condition);
+        assertTrue(value.get() > 0);
+
+    }
 
     @Test
     public void shouldInsertTTL() {

elasticsearch-driver/src/test/java/org/jnosql/diana/elasticsearch/document/ElasticsearchDocumentCollectionManagerTest.java

@@ -49,7 +49,6 @@
 public class ElasticsearchDocumentCollectionManagerTest {
 
 
-
     private ElasticsearchDocumentCollectionManager entityManager;
 
     @BeforeEach
@@ -74,7 +73,7 @@ public void shouldInsert() {
 
     @Test
     public void shouldInsertTTL() {
-        assertThrows(UnsupportedOperationException.class, () ->{
+        assertThrows(UnsupportedOperationException.class, () -> {
             entityManager.insert(getEntity(), Duration.ofSeconds(1L));
         });
     }
@@ -170,7 +169,6 @@ public void shouldFindAll() {
     }
 
 
-
     @Test
     public void shouldSaveSubDocument() {
         DocumentEntity entity = getEntity();
@@ -224,6 +222,17 @@ public void shouldRetrieveListSubdocumentList() {
         assertTrue(contacts.stream().allMatch(d -> d.size() == 3));
     }
 
+    @Test
+    public void shouldCount() {
+        DocumentEntity entity = getEntity();
+        DocumentEntity entity2 = getEntity();
+        entity2.add(Document.of("_id", "test"));
+        entityManager.insert(entity);
+        entityManager.insert(entity2);
+        assertTrue(entityManager.count(COLLECTION_NAME) > 0);
+    }
+
+
     private DocumentEntity createSubdocumentList() {
         DocumentEntity entity = DocumentEntity.of(COLLECTION_NAME);
         entity.add(Document.of("_id", "ids"));