Fix handling transformers properties. (#1037)

Author: OlgaMaciaszek

spring-cloud-loadbalancer/src/main/java/org/springframework/cloud/loadbalancer/annotation/LoadBalancerClientConfiguration.java

@@ -30,7 +30,6 @@
 import org.springframework.cloud.client.ServiceInstance;
 import org.springframework.cloud.client.discovery.DiscoveryClient;
 import org.springframework.cloud.client.discovery.ReactiveDiscoveryClient;
-import org.springframework.cloud.loadbalancer.config.XForwardedConfigurationCondition;
 import org.springframework.cloud.loadbalancer.core.ReactorLoadBalancer;
 import org.springframework.cloud.loadbalancer.core.RetryAwareServiceInstanceListSupplier;
 import org.springframework.cloud.loadbalancer.core.RoundRobinLoadBalancer;
@@ -100,7 +99,6 @@ public ServiceInstanceListSupplier zonePreferenceDiscoveryClientServiceInstanceL
 		@Bean
 		@ConditionalOnBean({ XForwardedHeadersTransformer.class, LoadBalancerClientFactory.class })
 		@ConditionalOnMissingBean
-		@Conditional(XForwardedConfigurationCondition.class)
 		public XForwardedHeadersTransformer xForwarderHeadersTransformer(LoadBalancerClientFactory clientFactory) {
 			return new XForwardedHeadersTransformer(clientFactory);
 		}

spring-cloud-loadbalancer/src/main/java/org/springframework/cloud/loadbalancer/config/BlockingLoadBalancerClientAutoConfiguration.java

@@ -35,13 +35,8 @@
 import org.springframework.cloud.loadbalancer.blocking.retry.BlockingLoadBalancedRetryFactory;
 import org.springframework.cloud.loadbalancer.core.LoadBalancerServiceInstanceCookieTransformer;
 import org.springframework.cloud.loadbalancer.support.LoadBalancerClientFactory;
-import org.springframework.cloud.loadbalancer.support.LoadBalancerEnvironmentPropertyUtils;
 import org.springframework.context.annotation.Bean;
-import org.springframework.context.annotation.Condition;
-import org.springframework.context.annotation.ConditionContext;
-import org.springframework.context.annotation.Conditional;
 import org.springframework.context.annotation.Configuration;
-import org.springframework.core.type.AnnotatedTypeMetadata;
 import org.springframework.retry.support.RetryTemplate;
 import org.springframework.web.client.RestTemplate;
 
@@ -68,15 +63,13 @@ public LoadBalancerClient blockingLoadBalancerClient(LoadBalancerClientFactory l
 	}
 
 	@Bean
-	@Conditional(AddServiceInstanceCookieCondition.class)
 	@ConditionalOnMissingBean(LoadBalancerServiceInstanceCookieTransformer.class)
 	public LoadBalancerServiceInstanceCookieTransformer loadBalancerServiceInstanceCookieTransformer(
 			LoadBalancerProperties properties) {
 		return new LoadBalancerServiceInstanceCookieTransformer(properties.getStickySession());
 	}
 
 	@Bean
-	@Conditional(XForwardedConfigurationCondition.class)
 	@ConditionalOnMissingBean(XForwardedHeadersTransformer.class)
 	@ConditionalOnBean(LoadBalancerClientFactory.class)
 	public XForwardedHeadersTransformer xForwarderHeadersTransformer(
@@ -98,14 +91,4 @@ LoadBalancedRetryFactory loadBalancedRetryFactory(
 
 	}
 
-	static class AddServiceInstanceCookieCondition implements Condition {
-
-		@Override
-		public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
-			return LoadBalancerEnvironmentPropertyUtils.trueForClientOrDefault(context.getEnvironment(),
-					"sticky-session.add-service-instance-cookie");
-		}
-
-	}
-
 }

spring-cloud-loadbalancer/src/main/java/org/springframework/cloud/loadbalancer/config/XForwardedConfigurationCondition.java

@@ -57,6 +57,9 @@ public HttpRequest transformRequest(HttpRequest request, ServiceInstance instanc
 		}
 		LoadBalancerProperties.StickySession stickySession = factory != null
 				? factory.getProperties(instance.getServiceId()).getStickySession() : stickySessionProperties;
+		if (!stickySession.isAddServiceInstanceCookie()) {
+			return request;
+		}
 		String instanceIdCookieName = stickySession.getInstanceIdCookieName();
 		if (!StringUtils.hasText(instanceIdCookieName)) {
 			return request;

spring-cloud-loadbalancer/src/main/java/org/springframework/cloud/loadbalancer/core/LoadBalancerServiceInstanceCookieTransformer.java

@@ -16,6 +16,7 @@
 
 package org.springframework.cloud.loadbalancer.core;
 
+import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 
 import org.springframework.cloud.client.DefaultServiceInstance;
@@ -43,6 +44,11 @@ class LoadBalancerServiceInstanceCookieTransformerTests {
 
 	HttpRequest request = new MockClientHttpRequest();
 
+	@BeforeEach
+	void setUp() {
+		stickySessionProperties.setAddServiceInstanceCookie(true);
+	}
+
 	@Test
 	void shouldAddServiceInstanceCookieHeader() {
 		HttpRequest newRequest = transformer.transformRequest(request, serviceInstance);