Feign 请求头设置与传递问题

c790d3bfd09d88f5ad32429b1361d6f6.png

Feign其实也是通过HTTP请求来实现的通信。那么自然绕不开HTTP相关的东西,比如很多系统中权限校验都是通过Header中的参数来实现,需要将前端传过来的header转发到目标服务,这里主要记录一下关于Header的设置。

下面提到的同步/异步 只是记录一下遇到问题的情景, 异步方法的实现方式同样适用于同步方法

同步

同步方法一: 通过拦截器

这种方法是比较通用而且也比较常见的,通过实现Feign的RequestInterceptor接口,重写里面的apply方法,为RequestTemplate添加请求头信息

代码如下

@Component
public class FeignClientsConfigurationCustom implements RequestInterceptor {
    @Override
    public void apply(RequestTemplate requestTemplate) {
    // 此种方式是线程安全的
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder
                .getRequestAttributes();
    // 不为空时取出请求中的header 原封不动的设置到feign请求中
        if (null != attributes) {
            HttpServletRequest request = attributes.getRequest();
            if (null != request) {
        // 遍历设置 也可从request取出整个Header 写到RequestTemplate 中
                Enumeration<String> headerNames = request.getHeaderNames();
                if (headerNames != null) {
                    while (headerNames.hasMoreElements()) {
                        String name = headerNames.nextElement();
                        String values = request.getHeader(name);
                        requestTemplate.header(name, values);
                    }
                }
            }
        }
    }
}

完成上述代码之后 在FeignClient注解中加入 configuration = FeignClientsConfigurationCustom.class 即可

@FeignClient(name = "testClient", configuration = FeignClientsConfigurationCustom.class)
public interface testServer{
}

同步方法二:@Headers注解

这种方法比较适用于一些不变的参数,如Content-Type等

@FeignClient(name = "testClient", configuration = FeignClientsConfigurationCustom.class)
public interface testServer{

    @GetMapping("/test")
    @Headers({"Content-Type: application/json","Accept: application/json"})
    String test(@RequestParam String param);
}

异步

此处说的异步场景是通过Spring中的@Async实现的

// 带有@Async注解,异步调用Feign
@Autowired
private TestService testService;

@GetMapping("/test")
public String test(){
    // 此方法是一个异步方法, 在该方法中调用了Feign服务
    testService.testMethod();
    return "Hello world!";
}

此时的业务场景是 前端向我发送一个服务,我异步地调用其他微服务的方法, 由于此方法执行耗时可能会比较长,而且对用户来说没有下一步操作,所以直接return掉,那么主线程因return而关闭,此时在刚刚的FeignConfig中就无法获取到请求头了。

26eb071665a4863b13e0b40d733068b8.png

如图所示,红色方框圈起来的地方都不为null,但是最终获取到的Header是一个空的Map。

所以猜测是因为主线程退出触发了JVM的回收机制

那么此时的情况将是 主线程已经退出,子线程没有执行完

所以显然,此时不能通过这种方式传递Header了。

异步方法一:线程私有变量ThreadLocal。

既然无法直接通过获取HttpServletRequest来获取Header,那么可以稍微改造一下,在原来的基础上添加一个拦截器。
所有的请求过来的时候,在拦截器中将Header先取出来,然后设置到本线程私有的Map中。

原来的apply方法在提交请求的时候再通过ThreadLocal提供的remove方法,清除掉。

只要把对该Map的操作封装一个工具类,工具类中实现get/set方法即可。

其实这种方式就是换了一个地方保存请求头,因此实用性与便捷性都还可以。

代码与截图待补充

异步方法二:通过传参

该方式是在方法执行前,先将需要的参数取出来,比如我需要一个token 就在Header中取出token,需要一个Content-Type就取出Content-Type。

然后将取出来的值作为参数传递到待执行的方法中。

该方法所调用的Feign接口需要做一个改造,在参数中添加带有@RequestHeader的注解,该注解表示将变量放在请求头,而不是请求的参数或者请求体里面。

@FeignClient(name = "testClient", configuration = FeignClientsConfigurationCustom.class)
public interface testServer{

    @GetMapping("/test")
    String test(@RequestParam String param,@RequestHeader String token);
}

代码与截图待补充

这种方法对于原来代码的改动较小,如果异步的场景比较少的话可以选择这种方法。
但是如果项目中用到了较多的异步方法,那么就需要异步方法一里面的拦截器 + ThreadLocal + RequestInterceptor了。

source: //www.write1bug.cn/archives/feign请求头设置问题同步异步

6902f2903485ddc4d2dd1e561b5c54c1.png

喜欢,在看

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值