fix:fix rule-based router when using RestTemplate. (#1201)

Tencent · Nov 8, 2023 · 5c49e44 · 5c49e44
1 parent f186d2c
commit 5c49e44
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -19,3 +19,4 @@
 - [feat:support configuration encryption.](https://github.com/Tencent/spring-cloud-tencent/pull/1183)
 - [feat:optimize examples.](https://github.com/Tencent/spring-cloud-tencent/pull/1187)
 - [feat: support nacos namespace mapping](https://github.com/Tencent/spring-cloud-tencent/pull/1192)
+- [fix:fix rule-based router when using RestTemplate.](https://github.com/Tencent/spring-cloud-tencent/pull/1201)
diff --git a/...router/src/main/java/com/tencent/cloud/polaris/router/config/RouterAutoConfiguration.java b/...router/src/main/java/com/tencent/cloud/polaris/router/config/RouterAutoConfiguration.java
@@ -36,6 +36,7 @@
 import com.tencent.cloud.polaris.router.interceptor.RuleBasedRouterRequestInterceptor;
 import com.tencent.cloud.polaris.router.resttemplate.RouterLabelRestTemplateInterceptor;
 import com.tencent.cloud.polaris.router.spi.SpringWebRouterLabelResolver;
+import com.tencent.cloud.rpc.enhancement.resttemplate.EnhancedRestTemplateInterceptor;
 
 import org.springframework.beans.factory.SmartInitializingSingleton;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -125,7 +126,13 @@ public RouterLabelRestTemplateInterceptor routerLabelRestTemplateInterceptor(
 		public SmartInitializingSingleton addRouterLabelInterceptorForRestTemplate(RouterLabelRestTemplateInterceptor interceptor) {
 			return () -> restTemplates.forEach(restTemplate -> {
 				List<ClientHttpRequestInterceptor> list = new ArrayList<>(restTemplate.getInterceptors());
-				list.add(interceptor);
+				int addIndex = list.size();
+				for (int i = 0; i < list.size(); i++) {
+					if (list.get(i) instanceof EnhancedRestTemplateInterceptor) {
+						addIndex = i;
+					}
+				}
+				list.add(addIndex, interceptor);
 				restTemplate.setInterceptors(list);
 			});
 		}

diff --git a/...c/main/java/com/tencent/cloud/rpc/enhancement/config/RpcEnhancementAutoConfiguration.java b/...c/main/java/com/tencent/cloud/rpc/enhancement/config/RpcEnhancementAutoConfiguration.java
@@ -52,7 +52,6 @@
 import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication;
 import org.springframework.boot.context.properties.EnableConfigurationProperties;
 import org.springframework.boot.web.servlet.FilterRegistrationBean;
-import org.springframework.cloud.client.loadbalancer.LoadBalanced;
 import org.springframework.cloud.client.serviceregistry.Registration;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -176,7 +175,6 @@ public EnhancedLoadBalancerClientAspect enhancedLoadBalancerClientAspect() {
 	@ConditionalOnClass(name = "org.springframework.web.client.RestTemplate")
 	protected static class PolarisRestTemplateAutoConfiguration {
 
-		@LoadBalanced
 		@Autowired(required = false)
 		private List<RestTemplate> restTemplates = Collections.emptyList();