Ribbon拦截器ClientHttpRequestInterceptor-传递header

迈不过友情╰ 2023-07-17 06:18 18阅读 0赞

Spring Cloud的微服务在使用RestTemplate调用的时候，header会丢失，这会导致服务间token等一些信息的丢失

可以使用ClientHttpRequestInterceptor拦截请求，并将header统一设置进去

首先配置一个自定义的ClientHttpRequestInterceptor

@Slf4j
    @Component
    public class MyRibbonInterceptor implements ClientHttpRequestInterceptor {
    
        @Override
        public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution) throws IOException {
    
            log.info("===request: {}, {}", request.getURI().getHost(), request.getURI().getPath());
            request.getHeaders().set("aaa", "aaa");
            return execution.execute(request, body);
        }
    }

在启动程序的时候将这个拦截器注入所有的RestTemplate的Bean

@Bean
        @LoadBalanced
        RestTemplate restTemplate(){
            RestTemplate restTemplate = new RestTemplate();
            restTemplate.setInterceptors(Collections.singletonList(myRibbonInterceptor));
            return restTemplate;
        }

也可以通过程序启动的时候拦截所有的RestTemplate的bean来对所有RestTemplate全局注入

private void setRestTemplateInterceptors(){
            Map<String, RestTemplate> map = applicationContext.getBeansOfType(RestTemplate.class);
            map.keySet().forEach(key -> {
                RestTemplate restTemplate = map.get(key);
                List<ClientHttpRequestInterceptor> list = new ArrayList<>(restTemplate.getInterceptors());
                list.add(myRibbonInterceptor);
                restTemplate.setInterceptors(list);
            });
        }

写一个controller来测试

@GetMapping("/ribbon")
        public String ribbon(@RequestParam String param) {
            return restTemplate.getForObject("http://SPRING-CLOUD-CLIENT-DEMO/demo/get?param="+param, String.class);
        }

启动程序测试

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3lpbmd6aWlzbWU_size_16_color_FFFFFF_t_70]

查看程序打印, 可以看到进入了自定义的MyRibbonInterceptor

INFO 31884 --- [ctor-http-nio-3] c.m.d.g.interceptor.MyRibbonInterceptor  : ===request: SPRING-CLOUD-CLIENT-DEMO, /demo/get

查看restTemplate调用的服务打印

设置的header成功传递

INFO 1 --- [nio-8801-exec-6] c.m.d.client.controller.DemoController   : aaaa: a
      INFO 1 --- [nio-8801-exec-6] c.m.d.client.controller.DemoController   : websession: ECF4D97D02EEAFDDA3C15A7F1F050F61
      INFO 1 --- [nio-8801-exec-6] c.m.d.client.controller.DemoController   : header: aaaa, bbbb
      INFO 1 --- [nio-8801-exec-6] c.m.d.client.controller.DemoController   : header: accept, */*
      INFO 1 --- [nio-8801-exec-6] c.m.d.client.controller.DemoController   : header: user-agent, Java/1.8.0_191
      INFO 1 --- [nio-8801-exec-6] c.m.d.client.controller.DemoController   : header: host, 39.108.15.147:8801
      INFO 1 --- [nio-8801-exec-6] c.m.d.client.controller.DemoController   : header: connection, keep-alive

欢迎关注微信交流  
![在这里插入图片描述][70]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3lpbmd6aWlzbWU_size_16_color_FFFFFF_t_70]: /images/20230528/a8387d0cb269476288b8a393a031f39b.png
[70]: /images/20230528/c79f19f27fd5428c839ce6ad9e53a3da.png