Skip to content

Commit 6935791

Browse files
committed
chore: change to follow the dependent changed codes
Signed-off-by: Maximillian Arruda <[email protected]>
1 parent a4e2121 commit 6935791

File tree

1 file changed

+2
-29
lines changed
  • jnosql-elasticsearch/src/main/java/org/eclipse/jnosql/databases/elasticsearch/communication

1 file changed

+2
-29
lines changed

jnosql-elasticsearch/src/main/java/org/eclipse/jnosql/databases/elasticsearch/communication/EntityConverter.java

Lines changed: 2 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,6 @@
1818

1919
import co.elastic.clients.elasticsearch.ElasticsearchClient;
2020
import co.elastic.clients.elasticsearch._types.SortOrder;
21-
import co.elastic.clients.elasticsearch._types.query_dsl.BoolQuery;
22-
import co.elastic.clients.elasticsearch._types.query_dsl.MatchQuery;
23-
import co.elastic.clients.elasticsearch._types.query_dsl.Query;
2421
import co.elastic.clients.elasticsearch.core.SearchRequest;
2522
import co.elastic.clients.elasticsearch.core.SearchResponse;
2623
import org.eclipse.jnosql.communication.document.Document;
@@ -30,7 +27,6 @@
3027

3128
import java.io.IOException;
3229
import java.util.HashMap;
33-
import java.util.List;
3430
import java.util.Map;
3531
import java.util.function.Consumer;
3632
import java.util.stream.Stream;
@@ -59,18 +55,14 @@ static Map<String, Object> getMap(DocumentEntity entity) {
5955
}
6056

6157
static Stream<DocumentEntity> query(DocumentQuery query, ElasticsearchClient client, String index) {
62-
QueryConverterResult select = QueryConverter.select(query);
58+
QueryConverterResult select = QueryConverter.select(client, index, query);
6359

6460
try {
65-
Stream<DocumentEntity> idQueryStream = Stream.empty();
6661
Stream<DocumentEntity> statementQueryStream = Stream.empty();
67-
if (select.hasId()) {
68-
idQueryStream = executeId(client, index, select);
69-
}
7062
if (select.hasStatement()) {
7163
statementQueryStream = executeStatement(query, client, index, select);
7264
}
73-
return Stream.concat(idQueryStream, statementQueryStream).distinct();
65+
return statementQueryStream.distinct();
7466
} catch (IOException e) {
7567
throw new ElasticsearchException("An error to execute a query on elasticsearch", e);
7668
}
@@ -122,25 +114,6 @@ private static boolean isSudDocumentList(Object value) {
122114
allMatch(d -> d instanceof Iterable && isSudDocument(d));
123115
}
124116

125-
private static Stream<DocumentEntity> executeId(ElasticsearchClient client, String index,
126-
QueryConverterResult select) throws IOException {
127-
128-
List<Query> ids = select.getIds().stream()
129-
.map(id -> MatchQuery.of(m -> m
130-
.field(EntityConverter.ID_FIELD)
131-
.query(id))._toQuery()
132-
).collect(toList());
133-
134-
SearchRequest searchRequest = SearchRequest.of(sb -> sb
135-
.index(index)
136-
.query(BoolQuery.of(q -> q
137-
.should(ids))._toQuery()));
138-
SearchResponse<Map> responses = client.search(searchRequest, Map.class);
139-
140-
return getDocumentEntityStream(client, responses);
141-
142-
}
143-
144117
static Stream<DocumentEntity> getDocumentEntityStream(ElasticsearchClient client, SearchResponse<Map> responses) {
145118
return responses.hits().hits().stream()
146119
.map(hits -> ElasticsearchEntry.of(hits.id(), hits.source()))

0 commit comments

Comments
 (0)