feign远程调用丢失请求头源码分析与解决

2021-01-19 19:02:41 阅读：166 来源： 互联网

标签：feign 调用请求登录 request 源码 cookie 方法

前言

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

问题

有时候，我们先行登录，然后访问服务A的某个方法，请求头中携带cookie，标识我们已经登录。但若是我们访问的目标方法在执行过程中使用feign进行原程调用服务B（假设不存在跨域），而服务B也要先判断登录状态，我们可能发现服务B会调用失败，或者说拿不到数据，理由是服务B认为我们并未登录。而这时，如果我们直接从浏览器访问服务B的这个方法却能得到一个成功的响应。

也就是说：

浏览器--->服务A成功；服务A-->服务B失败；浏览器-->服务B失败
结合上面所说，服务AB都会先判断用户登录状态，浏览器直接访问AB时都会带上登录成功后保存的cookie，而服务A通过Feign远程调用B，却被认为未登录，显然，这部分请求头数据丢失。

feign源码分析

我们来看下feign远程调用是如何执行的，我们在feign远程调用之处打上断点
在这里插入图片描述

step into进入方法执行，会发现是一个代理对象的invoke方法在执行，首先判断是方法名，

如果是 toString()，hashCode()，equals()这几个方法，那就是本地直接完成了。
如果是真正的远程调用，就会最后进入最后一行。

上一步之后再次step into，发现还是一个invoke方法，方法内，首先根据请求参数创建一个RequestTemplate，核心部分是 while(true) 里面的 executeAndDecode()，while 其实是加了一层重试机制，这里不多说。
进入executeAndDecode方法，我们看到 targetRequest() 构造出了一个request对象，而最终的response就是这个request请求的执行结果。

同时我们能够看到这个request对象的请求头中是空的，当然也就不存在cookie，也就无法识别我们是否登录。

总结

使用feign进行远程调用时，首先判断目标方法类型，如果是 toString()，hashCode()，equals()这几个方法，那就是本地直接完成了；
如果是真正的远程调用，执行的是 executeAndDecode 方法，在这个方法体内，会通过 targetRequest 方法创建出一个新的 request 对象，这个新的request会按照我们指定的参数和路径去发送请求，并获得响应结果。
这个新的request对象的请求头为空（所以会丢失原来的请求头）

解决

问题在于feign自己创建出resttemplate，再用它构建一个新的request对象去发送请求，而这个新的request不包含任何请求头信息。我们应该在它创造出这个request之后，在它真正发送请求之前，把原始请求头中的数据给它复制过去。

我们来看一下feign最后构建出创建request对象的 targetRequest方法
在这里插入图片描述
我们发现这里面会有调用了一系列 RequestInterceptor的apply方法对其进行增强，最后才返回，只不过默认情况下这些拦截器是空的。

因此，我们需要需要自己实现一个 RequestInterceptor，在它的apply方法中将原始请求头中的数据同步到feign创建出的新的request中，并且将这个拦截器注入容器中，这样feign在执行目标方法之前会被其拦截，对其先进行增强。

@Component
public class FeignBeforeExecInterceptor implements RequestInterceptor {

    @Override
    public void apply(RequestTemplate template) {
        ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = requestAttributes.getRequest();
        // 拿到原始请求头数据
        String cookie = request.getHeader("Cookie");
        if (!StringUtils.isEmpty(cookie)) {
            // 同步
            template.header("Cookie", cookie);
        }
    }
}

比较难处理的地方在于，我们如何拿到原始的request对象，spring提供了一个叫 RequestContextHolder对象帮我们解决这个难题，通过它的 getRequestAttributes 方法或者 currentRequestAttributes 方法就能获取到原始请求数据。关于这两个方法的区别，可简单认为，前者如果获取失败，会返回null；而后者会抛出异常。
在这里插入图片描述

问题

还有个问题是这个 RequestContextHolder是如何保存原始请求的，以至于我们在任何时候都能很方便的拿到，而不是像只能在controller层通过方法参数获取。其实如果你细心看上面的源码图片中的注释的话，就能看到它写的是获取与当前线程绑定的请求数据

我们知道，服务器（tomcat）会为每一个请求分配一个线程，从filter到controller到service到db再返回，全都都是同一个线程，所以，只要从一开始就把原始请求和这个线程绑定在一起，那么只要在这个线程内，我就能随时拿到这个数据。

是不是很熟悉，这不就是ThreadLocal嘛！再瞅一眼源代码证明一下？
在这里插入图片描述

标签：feign,调用,请求,登录,request,源码,cookie,方法
来源： https://www.cnblogs.com/codervivi/p/14299451.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

feign远程调用丢失请求头源码分析与解决

前言

问题

feign源码分析

总结

解决

问题