protect ES

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.action.get.GetResponse;
 import org.elasticsearch.search.SearchHit;
 import org.jnosql.diana.api.document.Document;
 import org.jnosql.diana.api.document.DocumentEntity;
@@ -100,4 +101,9 @@ static ElasticsearchEntry of(SearchHit searchHit) {
                 searchHit.getSourceAsMap());
     }
 
+    static ElasticsearchEntry of(GetResponse searchHit) {
+        return new ElasticsearchEntry(searchHit.getId(),
+                searchHit.getType(),
+                searchHit.getSourceAsMap());
+    }
 }

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

@@ -175,7 +175,7 @@ private static void executeId(DocumentQuery query, RestHighLevelClient client, S
         MultiGetResponse responses = client.multiGet(multiGetRequest);
         Stream.of(responses.getResponses())
                 .map(MultiGetItemResponse::getResponse)
-                .map(g -> new ElasticsearchEntry(g.getId(), g.getType(), g.getSourceAsMap()))
+                .map(ElasticsearchEntry::of)
                 .filter(ElasticsearchEntry::isNotEmpty)
                 .map(ElasticsearchEntry::toEntity)
                 .forEach(entities::add);