Cwikius-Spring/spring-data-elasticsearch
1
0
Fork 0
mirror of https://github.com/spring-projects/spring-data-elasticsearch.git synced 2025-06-01 09:42:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Polishing

Browse Source
This commit is contained in:
Peter-Josef Meisch 2021-03-16 22:36:57 +01:00
parent 005d6a4d6f
commit 98a8d1a5ac
No known key found for this signature in database
GPG Key ID: DE108246970C7708
5 changed files with 50 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/main/java/org/springframework/data/elasticsearch/client/reactive/DefaultReactiveElasticsearchClient.java
View File

@ -22,7 +22,6 @@ import io.netty.handler.ssl.IdentityCipherSuiteFilter;
import io.netty.handler.ssl.JdkSslContext;
import io.netty.handler.timeout.ReadTimeoutHandler;
import io.netty.handler.timeout.WriteTimeoutHandler;
import org.elasticsearch.action.get.MultiGetItemResponse;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import reactor.netty.http.client.HttpClient;
@ -64,6 +63,7 @@ import org.elasticsearch.action.delete.DeleteRequest;
import org.elasticsearch.action.delete.DeleteResponse;
import org.elasticsearch.action.get.GetRequest;
import org.elasticsearch.action.get.GetResponse;
import org.elasticsearch.action.get.MultiGetItemResponse;
import org.elasticsearch.action.get.MultiGetRequest;
import org.elasticsearch.action.get.MultiGetResponse;
import org.elasticsearch.action.index.IndexRequest;
@ -409,8 +409,7 @@ public class DefaultReactiveElasticsearchClient implements ReactiveElasticsearch
@Override
public Flux<SearchHit> searchTemplate(HttpHeaders headers, SearchTemplateRequest searchTemplateRequest) {
return sendRequest(searchTemplateRequest, requestCreator.searchTemplate(), SearchTemplateResponse.class, headers)
.map(r -> r.getResponse().getHits())
.flatMap(Flux::fromIterable);
.map(response -> response.getResponse().getHits()).flatMap(Flux::fromIterable);
}
/*
@ -876,10 +875,9 @@ public class DefaultReactiveElasticsearchClient implements ReactiveElasticsearch
String mediaType = response.headers().contentType().map(MediaType::toString).orElse(XContentType.JSON.mediaType());
return response.body(BodyExtractors.toMono(byte[].class)) //
.switchIfEmpty(Mono
.error(new ElasticsearchStatusException(String.format("%s request to %s returned error code %s and no body.",
request.getMethod(), request.getEndpoint(), statusCode), status))
)
.switchIfEmpty(Mono.error(
new ElasticsearchStatusException(String.format("%s request to %s returned error code %s and no body.",
request.getMethod(), request.getEndpoint(), statusCode), status)))
.map(bytes -> new String(bytes, StandardCharsets.UTF_8)) //
.flatMap(content -> contentOrError(content, mediaType, status))
.flatMap(unused -> Mono

22
src/test/java/org/springframework/data/elasticsearch/client/reactive/ReactiveElasticsearchClientIntegrationTests.java
View File

@ -454,10 +454,7 @@ public class ReactiveElasticsearchClientIntegrationTests {
params.put("firstname", "inline");
request.setScriptParams(params);
client.searchTemplate(request)
.as(StepVerifier::create)
.expectNextCount(1)
.verifyComplete();
client.searchTemplate(request).as(StepVerifier::create).expectNextCount(1).verifyComplete();
}
@Test // #1725
@ -471,14 +468,13 @@ public class ReactiveElasticsearchClientIntegrationTests {
testDoc.put("lastname", "template");
add(testDoc).to(INDEX_I);
client.execute(c -> c.post()
.uri(builder -> builder.path("_scripts/searchbyfirstname").build())
client.execute(c -> c.post().uri(builder -> builder.path("_scripts/searchbyfirstname").build())
.contentType(MediaType.APPLICATION_JSON)
.bodyValue(
"{\"script\":{\"lang\":\"mustache\",\"source\":{\"query\":{\"match\":{\"firstname\":\"{{firstname}}\"}}}}}")
.retrieve()
.bodyToMono(Void.class))
.block();
.retrieve().bodyToMono(Void.class)).block();
client.indices().refreshIndex(request -> request.indices(INDEX_I)).block();
SearchTemplateRequest request = new SearchTemplateRequest(new SearchRequest(INDEX_I));
request.setScriptType(ScriptType.STORED);
@ -487,10 +483,10 @@ public class ReactiveElasticsearchClientIntegrationTests {
params.put("firstname", "stored");
request.setScriptParams(params);
client.searchTemplate(request)
.as(StepVerifier::create)
.expectNextCount(1)
.verifyComplete();
client.searchTemplate(request) //
.as(StepVerifier::create) //
.expectNextCount(1) //
.verifyComplete(); //
}
@Test // DATAES-488