SpringCloud/spring-cloud-sleuth
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Polish

Browse Source
This commit is contained in:
Marcin Grzejszczak
2019-03-29 11:07:47 +01:00
parent 558900155a
commit 4f60937d27
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
spring-cloud-sleuth-core/src/main/java/org/springframework/cloud/sleuth/instrument/web/client/TraceWebClientBeanPostProcessor.java
View File

@@ -75,16 +75,19 @@ final class TraceWebClientBeanPostProcessor implements BeanPostProcessor {
throws BeansException {
if (bean instanceof WebClient) {
WebClient webClient = (WebClient) bean;
return webClient.mutate()
.filters(addTraceExchangeFilterFunctionIfNotPresent()).build();
return wrapBuilder(webClient.mutate()).build();
}
else if (bean instanceof WebClient.Builder) {
WebClient.Builder webClientBuilder = (WebClient.Builder) bean;
return webClientBuilder.filters(addTraceExchangeFilterFunctionIfNotPresent());
return wrapBuilder(webClientBuilder);
}
return bean;
}
private WebClient.Builder wrapBuilder(WebClient.Builder webClientBuilder) {
return webClientBuilder.filters(addTraceExchangeFilterFunctionIfNotPresent());
}
private Consumer<List<ExchangeFilterFunction>> addTraceExchangeFilterFunctionIfNotPresent() {
return functions -> {
if (functions.stream()