增加传递Header的用例

discovery-springcloud-example-service/src/main/java/com/nepxion/discovery/plugin/example/service/rest/ARestImpl.java

@@ -9,19 +9,25 @@ package com.nepxion.discovery.plugin.example.service.rest;
  * @version 1.0
  */
 
+import java.util.Enumeration;
+
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.client.RestTemplate;
+import org.springframework.web.context.request.ServletRequestAttributes;
 
 import com.alibaba.csp.sentinel.annotation.SentinelResource;
 import com.alibaba.csp.sentinel.slots.block.BlockException;
 import com.nepxion.discovery.common.constant.DiscoveryConstant;
+import com.nepxion.discovery.plugin.strategy.service.context.ServiceStrategyContextHolder;
 
 @RestController
 @ConditionalOnProperty(name = DiscoveryConstant.SPRING_APPLICATION_NAME, havingValue = "discovery-springcloud-example-a")
@@ -31,11 +37,34 @@ public class ARestImpl extends AbstractRestImpl {
     @Autowired
     private RestTemplate restTemplate;
 
+    @Autowired
+    private ServiceStrategyContextHolder serviceStrategyContextHolder;
+
     @RequestMapping(path = "/rest", method = RequestMethod.POST)
     @SentinelResource(value = "sentinel-resource", blockHandler = "handleBlock", fallback = "handleFallback")
     public String rest(@RequestBody String value) {
         value = doRest(value);
-        value = restTemplate.postForEntity("http://discovery-springcloud-example-b/rest", value, String.class).getBody();
+
+        // Just for testing
+        ServletRequestAttributes attributes = serviceStrategyContextHolder.getRestAttributes();
+        Enumeration<String> headerNames = attributes.getRequest().getHeaderNames();
+
+        System.out.println("Header name list:");
+        while (headerNames.hasMoreElements()) {
+            String headerName = headerNames.nextElement();
+            System.out.println("* " + headerName);
+        }
+
+        String token = attributes.getRequest().getHeader("token");
+        System.out.println("Old token=" + token);
+
+        System.out.println("New token=Token-A");
+
+        HttpHeaders headers = new HttpHeaders();
+        headers.set("token", "Token-A");
+
+        HttpEntity<String> entity = new HttpEntity<String>(value, headers);
+        value = restTemplate.postForEntity("http://discovery-springcloud-example-b/rest", entity, String.class).getBody();
 
         LOG.info("调用路径：{}", value);
 

discovery-springcloud-example-service/src/main/java/com/nepxion/discovery/plugin/example/service/rest/BRestImpl.java

@@ -9,19 +9,25 @@ package com.nepxion.discovery.plugin.example.service.rest;
  * @version 1.0
  */
 
+import java.util.Enumeration;
+
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.client.RestTemplate;
+import org.springframework.web.context.request.ServletRequestAttributes;
 
 import com.alibaba.csp.sentinel.annotation.SentinelResource;
 import com.alibaba.csp.sentinel.slots.block.BlockException;
 import com.nepxion.discovery.common.constant.DiscoveryConstant;
+import com.nepxion.discovery.plugin.strategy.service.context.ServiceStrategyContextHolder;
 
 @RestController
 @ConditionalOnProperty(name = DiscoveryConstant.SPRING_APPLICATION_NAME, havingValue = "discovery-springcloud-example-b")
@@ -31,11 +37,34 @@ public class BRestImpl extends AbstractRestImpl {
     @Autowired
     private RestTemplate restTemplate;
 
+    @Autowired
+    private ServiceStrategyContextHolder serviceStrategyContextHolder;
+
     @RequestMapping(path = "/rest", method = RequestMethod.POST)
     @SentinelResource(value = "sentinel-resource", blockHandler = "handleBlock", fallback = "handleFallback")
     public String rest(@RequestBody String value) {
         value = doRest(value);
-        value = restTemplate.postForEntity("http://discovery-springcloud-example-c/rest", value, String.class).getBody();
+
+        // Just for testing
+        ServletRequestAttributes attributes = serviceStrategyContextHolder.getRestAttributes();
+        Enumeration<String> headerNames = attributes.getRequest().getHeaderNames();
+
+        System.out.println("Header name list:");
+        while (headerNames.hasMoreElements()) {
+            String headerName = headerNames.nextElement();
+            System.out.println("* " + headerName);
+        }
+
+        String token = attributes.getRequest().getHeader("token");
+        System.out.println("Old token=" + token);
+
+        System.out.println("New token=Token-B");
+
+        HttpHeaders headers = new HttpHeaders();
+        headers.set("token", "Token-B");
+
+        HttpEntity<String> entity = new HttpEntity<String>(value, headers);
+        value = restTemplate.postForEntity("http://discovery-springcloud-example-c/rest", entity, String.class).getBody();
 
         LOG.info("调用路径：{}", value);
 

discovery-springcloud-example-service/src/main/java/com/nepxion/discovery/plugin/example/service/rest/CRestImpl.java

@@ -9,28 +9,49 @@ package com.nepxion.discovery.plugin.example.service.rest;
  * @version 1.0
  */
 
+import java.util.Enumeration;
+
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
 import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.context.request.ServletRequestAttributes;
 
 import com.alibaba.csp.sentinel.annotation.SentinelResource;
 import com.alibaba.csp.sentinel.slots.block.BlockException;
 import com.nepxion.discovery.common.constant.DiscoveryConstant;
+import com.nepxion.discovery.plugin.strategy.service.context.ServiceStrategyContextHolder;
 
 @RestController
 @ConditionalOnProperty(name = DiscoveryConstant.SPRING_APPLICATION_NAME, havingValue = "discovery-springcloud-example-c")
 public class CRestImpl extends AbstractRestImpl {
     private static final Logger LOG = LoggerFactory.getLogger(CRestImpl.class);
 
+    @Autowired
+    private ServiceStrategyContextHolder serviceStrategyContextHolder;
+
     @RequestMapping(path = "/rest", method = RequestMethod.POST)
     @SentinelResource(value = "sentinel-resource", blockHandler = "handleBlock", fallback = "handleFallback")
     public String rest(@RequestBody String value) {
         value = doRest(value);
 
+        // Just for testing
+        ServletRequestAttributes attributes = serviceStrategyContextHolder.getRestAttributes();
+        Enumeration<String> headerNames = attributes.getRequest().getHeaderNames();
+
+        System.out.println("Header name list:");
+        while (headerNames.hasMoreElements()) {
+            String headerName = headerNames.nextElement();
+            System.out.println("* " + headerName);
+        }
+
+        String token = attributes.getRequest().getHeader("token");
+        System.out.println("Token=" + token);
+
         LOG.info("调用路径：{}", value);
 
         return value;