defines search hit

Author: otaviojava

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

@@ -15,6 +15,7 @@
 package org.jnosql.diana.elasticsearch.document;
 
 
+import org.elasticsearch.search.SearchHit;
 import org.jnosql.diana.api.document.Document;
 import org.jnosql.diana.api.document.DocumentEntity;
 
@@ -92,4 +93,11 @@ private boolean isADocumentIterable(Object value) {
                         .allMatch(Map.class::isInstance);
     }
 
+
+    static ElasticsearchEntry of(SearchHit searchHit) {
+        return new ElasticsearchEntry(searchHit.getId(),
+                searchHit.getType(),
+                searchHit.getSourceAsMap());
+    }
+
 }

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

@@ -92,7 +92,7 @@ private static void executeStatement(DocumentQuery query, RestHighLevelClient cl
         SearchResponse response = client.search(searchRequest);
         Stream.of(response.getHits())
                 .flatMap(h -> Stream.of(h.getHits()))
-                .map(h -> new ElasticsearchEntry(h.getId(), h.getIndex(), h.getSourceAsMap()))
+                .map(ElasticsearchEntry::of)
                 .filter(ElasticsearchEntry::isNotEmpty)
                 .map(ElasticsearchEntry::toEntity)
                 .forEach(entities::add);