java

关注公众号 jb51net

关闭
首页 > 软件编程 > java > Feign远程调用丢失请求头

Feign远程调用丢失请求头问题

作者:头未秃

本文介绍了在服务端项目中如何解决资源访问限制问题,首先介绍了问题的产生,然后详细解析了源码,最后提出了解决方案,解决方案包括同步和异步两种,同步时直接向Spring容器注入RequestInterceptor拦截器

前言

我们在写服务端项目的时候,总会限制对某些资源的访问,最常见的就是要求用户先登录才能访问资源,当用户登录后就会将此次会话信息保存进session,同时返回给浏览器指定的cookie键值,下次浏览器再次访问,请求头中就会携带这个cookie,我们也以次来识别用户的登录状态,做出正确响应。

问题

有时候,我们先行登录,然后访问服务A的某个方法,请求头中携带cookie,标识我们已经登录。

但若是我们访问的目标方法在执行过程中使用feign进行远程调用服务B,而服务B也要先判断登录状态,我们可能发现服务B会调用失败,或者说拿不到数据,理由是服务B认为我们并未登录。

而这时,如果我们直接从浏览器访问服务B的这个方法却能得到一个成功的响应。

查看源码

1、使用feign进行远程调用时,首先判断目标方法类型,如果是 toString(),hashCode(),equals()这几个方法,那就是本地直接完成了

2、执行真正的远程调用的方法

3、根据模板template来创建请求request(默认的情况feign是不会帮我们把原请求头参数复制到新请求的请求头上的),然后进行客户端client调用

4、但是创建请求时,会先调用所有的拦截器RequestInterceptors

5、所以我们可以自己向容器中注册一个拦截器RequestInterceptor,在这个拦截器中重写apply方法,在apply方法中把老请求的cookie复制到新request的请求头中,完成请求头的同步。(然后targetRequest 方法就会调用拦截器的apply方法,进行返回)

总结:

feign远程调用,自己创建一个新的request对象,按照指定的路径和参数发起新的请求,并得到响应结果。但是这个新的request对象请求头为空,所以丢失了原先请求中的数据。

feign在创建新的request对象时,会调用一系列容器中的RequestInterceptor对象,执行其apply方法,对这个创建好的request进行增强,再去真正执行请求。但是默认情况下容器中不存在这类拦截器对象。

我们可以自己向容器中注册一个RequestInterceptor,在其apply方法体内,获取到原始request,将其数据取出,赋值到新的request中,完成请求头的同步。RequestContextHolder借助ThreadLocal将每一个原始请求与tomcat为其分配的线程绑定,之后,只要在同个线程内,随时随地都可轻易获取到原始request。而我们是在apply方法体内,通过 RequestContextHolder.getRequestAttributes() 获取的。RequestContextHolder是借助ThreadLocal将每一个原始请求与tomcat为其分配的线程绑定,之后,只要在同个线程内,随时随地都可轻易获取到原始request。

解决

1、同步时

直接向spring容器注入RequestInterceptor拦截器即可

/**
* Feign
* @return
*/
@Bean
public RequestInterceptor requestInterceptor(){
   return requestTemplate -> {
       //1、从RequestContextHolder获取原始请求的请求数据(请求参数、请求头等)
       ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.currentRequestAttributes();
       HttpServletRequest request = attributes.getRequest();
       //2、同步请求头数据
       Enumeration<String> headerNames = request.getHeaderNames();
       if (headerNames != null) {
           while (headerNames.hasMoreElements()) {
               String name = headerNames.nextElement();
               String values = request.getHeader(name);
               // 跳过 content-length,没有跳过会导致请求参数无法接收
               if (HttpHeaderConsts.CONTENT_LENGTH.equalsIgnoreCase(name)) {
                   continue;
               }
               requestTemplate.header(name, values);
           }
       }
   };
}

2、异步时

如果请求中使用了异步,也就是多线程,就算配置了上面的配置也会导致feign请求头丢失,因为请求头信息是通过threadLocal保存的,也就是只有在同一个线程中才能使用请求中的请求头并且同步初始请求头信息到feign请求中

解决办法

在异步调用时主动将请求头覆盖到异步线程的请求上下文中

// 通过请求上下文获取请求信息
RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
CompletableFuture<Void> addressFuture = CompletableFuture.runAsync(() -> {
    // 远程查询所有的收货地址列表
    // 在该线程中添加请求信息
    RequestContextHolder.setRequestAttributes(requestAttributes);
    List<MemberAddressVo> address = memberFeignService.getAddress(memberRespVo.getId());
    orderConfirmVo.setAdress(address);
}, executor);
​
CompletableFuture<Void> cartFuture = CompletableFuture.runAsync(() -> {
    // 远程查询购物车所有选中的购物想
    // 在该线程中添加请求信息
    RequestContextHolder.setRequestAttributes(requestAttributes);
    List<OrderItemVo> currentUserItems = cartFeignService.getCurrentUserItems();
    orderConfirmVo.setItems(currentUserItems);
}, executor);

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文