From 4f60937d27fab9335bef6bef0f22f39829818c1f Mon Sep 17 00:00:00 2001
From: Marcin Grzejszczak <marcin@grzejszczak.pl>
Date: Fri, 29 Mar 2019 11:07:47 +0100
Subject: [PATCH] Polish

---
 .../web/client/TraceWebClientBeanPostProcessor.java      | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/spring-cloud-sleuth-core/src/main/java/org/springframework/cloud/sleuth/instrument/web/client/TraceWebClientBeanPostProcessor.java b/spring-cloud-sleuth-core/src/main/java/org/springframework/cloud/sleuth/instrument/web/client/TraceWebClientBeanPostProcessor.java
index b7cfdcf12..6628f2eab 100644
--- a/spring-cloud-sleuth-core/src/main/java/org/springframework/cloud/sleuth/instrument/web/client/TraceWebClientBeanPostProcessor.java
+++ b/spring-cloud-sleuth-core/src/main/java/org/springframework/cloud/sleuth/instrument/web/client/TraceWebClientBeanPostProcessor.java
@@ -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()