OpenFeign远程调用丢失请求头问题以及异步调用丢失上下文问题

一,OpenFeign远程调用丢失请求头问题

原理

下面是OpenFeign远程调用的方法的源码部分,可以看到,是Spring调用buildTemplateFromArgs.create(argv)方法,构建了一个新的请求,这个请求没有任何的请求头信息,也就是我们请求头的数据会丢失.

但是,我们看到,在构建这个新请求的时候,会进入到executeAndDecode(template, options)方法,进入到这个方法以后,我们看到,继续调用调用Request request = targetRequest(template)对请求做了处理,

可以看到,遍历我们所有的拦截器Interceptor,给构造的请求附加参数,所以,给我们的容器中加上一个拦截器Interceptor就可以解决请求头丢失问题,代码如下

@Configuration
public class NaisiFeignConfig {

    @Bean
    public RequestInterceptor requestInterceptor(){
        return new RequestInterceptor() {
            @Override
            public void apply(RequestTemplate template) {
                //2,RequestContextHolder拿到刚进来的这个请求
                //RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
                ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
                System.out.println("RequestInterceptor线程"+Thread.currentThread().getId());
                //老请求
                //空指针Exception
                HttpServletRequest request = requestAttributes.getRequest();
                if (request != null){
                    //同步请求头数据,Cookie
                    String cookie = request.getHeader("Cookie");
                    //给新请求同步了老请求的Cookie
                    template.header("Cookie",cookie);
                }
            }
        };
    }
}

这时,新请求一进来就会先调用我们的拦截器,我们在拦截器给请求设置上所需要的Header请求头信息就可以.

二,异步调用丢失上下文问题

上面是OpenFeign远程调用丢失请求头问题,此时,我们发现,如果是异步请求,我们的拦截器就不能设置上所需要的Header信息.

原因

因为我们是异步请求,请求不一样

ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();

这里是使用ThreadLocal,即可以拿到同一个线程携带的数据,但是不同线程的数据,我们是拿不到的,

解决方法

我们在发远程调用请求之前,给异步请求添加上上下文信息即可

//获取之前的请求
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        //1,远程查询所有的收货地址列表
        CompletableFuture<Void> getAddress = CompletableFuture.runAsync(() -> {
            System.out.println("address线程"+Thread.currentThread().getId());
            //每一个线程都来共享之前请求的数据
            RequestContextHolder.setRequestAttributes(requestAttributes);
            List<MemberAddressVo> address = memberFeignService.getAddress(memberRespVo.getId());
            confirmVo.setAddress(address);
        }, executor);

此时就可以解决上下文丢失问题