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

Added WebClient.Builder support; fixes #779

Browse Source
This commit is contained in:
Marcin Grzejszczak
2017-11-14 15:57:52 +01:00
parent 163051a1f1
commit f29349d7ee
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,7 +1,5 @@
package org.springframework.cloud.sleuth.instrument.web.client;
import reactor.core.publisher.Mono;
import java.net.URI;
import java.util.AbstractMap;
import java.util.Iterator;
@@ -27,6 +25,7 @@ import org.springframework.web.reactive.function.client.ClientResponse;
import org.springframework.web.reactive.function.client.ExchangeFilterFunction;
import org.springframework.web.reactive.function.client.ExchangeFunction;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Mono;
/**
* {@link BeanPostProcessor} to wrap a {@link WebClient} instance into
@@ -56,6 +55,9 @@ class TraceWebClientBeanPostProcessor implements BeanPostProcessor {
.mutate()
.filter(new TraceExchangeFilterFunction(this.beanFactory))
.build();
} else if (bean instanceof WebClient.Builder) {
WebClient.Builder webClientBuilder = (WebClient.Builder) bean;
return webClientBuilder.filter(new TraceExchangeFilterFunction(this.beanFactory));
}
return bean;
}