Remove client specific code.

Original Pull Request #2352 Closes #2351
spring-projects · Nov 3, 2022 · 96c9aa8 · 96c9aa8
1 parent 049bcf5
commit 96c9aa8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/...ramework/data/elasticsearch/repository/support/SimpleReactiveElasticsearchRepository.java b/...ramework/data/elasticsearch/repository/support/SimpleReactiveElasticsearchRepository.java
@@ -15,13 +15,13 @@
  */
 package org.springframework.data.elasticsearch.repository.support;
 
+import org.springframework.data.elasticsearch.core.query.BaseQuery;
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
 
 import org.reactivestreams.Publisher;
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.domain.Sort;
-import org.springframework.data.elasticsearch.client.erhlc.NativeSearchQueryBuilder;
 import org.springframework.data.elasticsearch.core.AbstractReactiveElasticsearchTemplate;
 import org.springframework.data.elasticsearch.core.MultiGetItem;
 import org.springframework.data.elasticsearch.core.ReactiveElasticsearchOperations;
@@ -158,7 +158,11 @@ public Flux<T> findAllById(Publisher<ID> idStream) {
 		return Flux.from(idStream) //
 				.map(this::convertId) //
 				.collectList() //
-				.map(ids -> new NativeSearchQueryBuilder().withIds(ids).build()) //
+				.map(ids -> {
+					var query = new BaseQuery();
+					query.setIds(ids);
+					return query;
+				}) //
 				.flatMapMany(query -> {
 					IndexCoordinates index = entityInformation.getIndexCoordinates();
 					return operations.multiGet(query, entityInformation.getJavaType(), index) //