Ribbon拦截器ClientHttpRequestInterceptor-传递header

最新推荐文章于 2024-12-27 16:50:01 发布

yingziisme

最新推荐文章于 2024-12-27 16:50:01 发布

阅读量5.2k

点赞数 5

分类专栏： springcloud 文章标签： springcloud

本文链接：https://blog.youkuaiyun.com/yingziisme/article/details/104979724

版权

springcloud 专栏收录该内容

13 篇文章

订阅专栏

本文介绍了在Spring Cloud中，由于RestTemplate调用时header可能会丢失，为了解决这个问题，可以使用ClientHttpRequestInterceptor拦截请求并在其中设置header。通过创建自定义拦截器并注入到RestTemplate Bean中，实现全局header的传递。测试表明，设置的header成功地在服务间传递。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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