[#1110] fixed serviceName config is not effective for instanceIsolati…

…on policy (#1112)
huaweicloud · Nov 27, 2023 · f058477 · f058477
1 parent f58153f
commit f058477
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...i-governance/src/main/java/com/huaweicloud/governance/adapters/web/RestTemplateUtils.java b/...i-governance/src/main/java/com/huaweicloud/governance/adapters/web/RestTemplateUtils.java
@@ -18,6 +18,7 @@
 package com.huaweicloud.governance.adapters.web;
 
 import org.apache.servicecomb.governance.marker.GovernanceRequestExtractor;
+import org.springframework.cloud.client.loadbalancer.ServiceRequestWrapper;
 import org.springframework.http.HttpRequest;
 
 import com.huaweicloud.common.context.InvocationContextHolder;
@@ -58,6 +59,9 @@ public String instanceId() {
 
       @Override
       public String serviceName() {
+        if (request instanceof ServiceRequestWrapper) {
+          return ((ServiceRequestWrapper) request).getRequest().getURI().getHost();
+        }
         return request.getURI().getHost();
       }