Ribbon拦截器ClientHttpRequestInterceptor-传递header

yingzi_code

发布于 2020-03-20 03:19:09

1.8K00

代码可运行

文章被收录于专栏：生活不止眼前的代码生活不止眼前的代码

运行总次数：0

代码可运行

Ribbon拦截器ClientHttpRequestInterceptor-传递header

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);
    }

启动程序测试

查看程序打印, 可以看到进入了自定义的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