解决：使用Feign进行远程调用造成请求头丢失的问题

古城微笑少年丶 2022-11-19 09:57 720阅读 0赞

### 场景： ###

用户已经登录了网站，在浏览器页面我使用cookie保存session Id，由于使用feign进行远程调用的时候造成请求头丢失，所以被调用方无法获取到cookie，因此被调用方法认为用户未登录，所以我们无法在被调用方获取用户信息

### 使用单线程的解决办法： ###

创建`MyFeignConfig.java`配置类：

@Configuration
    public class MyFeignConfig { 
        @Bean
        public RequestInterceptor requestInterceptor() { 
            return new RequestInterceptor() { 
                @Override
                public void apply(RequestTemplate requestTemplate) { 
                    // 1、拿到request请求，以下两个方法来自spring，其中getRequestAttributes()方法内部使用ThreadLocal来做
                    ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
                    // 有些多线程进行远程调用不需要使用请求头中的信息，如果不在当前线程的ThreadLocal中设置RequestAttributes，那这里获取到的就是null
                    if (attributes != null) { 
                        HttpServletRequest request = attributes.getRequest();
                        // 避免多线程状态下request没有值出现NullPointerException异常
                        // Cookie名称来自前端，这是固定写法
                        String cookie = request.getHeader("Cookie");
                        // 将新请求的cookie传给被远程调用的模块，让它们知道我们已经登录了
                        requestTemplate.header("Cookie", cookie);
                    }
                }
            };
        }
    }

### 使用多线程的解决办法： ###

1、创建`MyFeignConfig.java`配置类，代码就在上面，不在赘述

2、在主线程中写上：

RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();

3、在分线程中调用远程方法之前写上：

RequestContextHolder.setRequestAttributes(requestAttributes);

注意：每一个分线程都要写哦！

### 解释 ###

我们在使用远程调用方法的地方打断点，如下：  
![在这里插入图片描述][2021012021350846.png]  
直接进入该方法，然后在进入的方法中在进入`invoke()`方法，如下：

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70]  
然后进入`invoke()`方法之后，进入`executeAndDecode()`方法

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70 1]  
然后你就可以看到`Request request = this.targetRequest(template);`，进入`targetRequest()`方法：

![在这里插入图片描述][20210120213940705.png]  
然后你就可以看到拦截器：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70 2]  
如果使用图来解释就是这样的，如下：

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70 3]  
因此我们需要创建一个名称为`RequestInterceptor`的拦截器，然后我们需要获取到请求头中的cookie信息，我们可以这么做：

ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
    HttpServletRequest request = attributes.getRequest();
    String cookie = request.getHeader("Cookie");

其中`getRequestAttributes()`方法细节如下：

@Nullable
    public static RequestAttributes getRequestAttributes() { 
    	RequestAttributes attributes = requestAttributesHolder.get();
    	if (attributes == null) { 
    		attributes = inheritableRequestAttributesHolder.get();
    	}
    	return attributes;
    }

其中`requestAttributesHolder`如下所示：

private static final ThreadLocal<RequestAttributes> requestAttributesHolder = new NamedThreadLocal<>("Request attributes");

可以看到`get()`方法利用的是`ThreadLocal`来存储`RequestAttributes`的实现类对象，而该对象中存储着request信息，所以如果我们不用多线程，就不需要考虑那么多了，毕竟同一个线程的ThreadLocal是不会发生改变的

如果你不使用多线程，直接创建一个`MyFeignConfig.java`配置类就行了，具体代码在上面的`使用单线程的解决办法`中

如果你使用多线程，仅仅使用`MyFeignConfig.java`配置类还是不够的，假如我使用`CompletableFuture`来实现多线程，在单线程状态下我们可以从`ThreadLocal`中取出我们想要的请求头信息，但是这个`ThreadLocal`是主线程的，我们在开线程的话就不能用了，这样以为这我们不能取出请求信息了，所以我们需要在主线程中取出RequestAttributes接口的实现类对象，然后在其他线程中在把它放入`ThreadLocal`，当然这个`ThreadLocal`就不是主线程的了，那么我们就可以继续获取请求头中的信息了，具体做法如下：

// 1、获取主线程中的请求信息，请求中有请求头信息，请求信息放在ThreadLocal中，如果我们新开辟线程，我们需要保证其他线程也可以用请求信息
    RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
    
    // 使用多线程获取用户的所有收获地址信息
    CompletableFuture<Void> addressFuture = CompletableFuture.runAsync(() -> { 
        // 2、将之前的请求信息传到新线程中，这样在MyThreadConfig.java中就可以获取到真正的请求信息了，这个需要在调用远程方法之前写上
        RequestContextHolder.setRequestAttributes(requestAttributes);
        List<MemberAddressVo> address = memberFeignService.getAddress(userInfo.getId());
        orderConfirmVo.setAddress(address);
    }, executor);

在上面我们已经看了`getRequestAttributes()`方法，现在我们来看一下`setRequestAttributes()`方法，源码如下：

public static void setRequestAttributes(@Nullable RequestAttributes attributes) { 
    	setRequestAttributes(attributes, false);
    }

执行进入`setRequestAttributes(attributes, false)`，源码如下：

public static void setRequestAttributes(@Nullable RequestAttributes attributes, boolean inheritable) { 
    	if (attributes == null) { 
    		resetRequestAttributes();
    	}
    	else { 
    		if (inheritable) { 
    			inheritableRequestAttributesHolder.set(attributes);
    			requestAttributesHolder.remove();
    		}
    		else { 
    			requestAttributesHolder.set(attributes);
    			inheritableRequestAttributesHolder.remove();
    		}
    	}
    }

由于`attributes`不是null，而`inheritable`是false，所以我们最终会执行这两行代码：

requestAttributesHolder.set(attributes);
    inheritableRequestAttributesHolder.remove();

我们来看一下`requestAttributesHolder`，如下：

private static final ThreadLocal<RequestAttributes> requestAttributesHolder = new NamedThreadLocal<>("Request attributes");

所以`set()`方法就是把主线程中的`RequestAttributes`接口的实体类对象放在分线程的`ThreadLocal`中，所以我们依然可以使用`MyFeignConfig`中的`requestInterceptor`() 方法中的如下代码获取`RequestAttributes`接口的实体类对象，然后从中获取请求头信息，如下：

// 其实获得的就是RequestAttributes类型数据，也就是主线程中的该类型数据
    ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();

用图解释如下：

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70 4]

[2021012021350846.png]: /images/20221119/7e7f2ce9fe6c4be0b4c7c139bbf4e6af.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70]: /images/20221119/a87ae62f4ae74056a9d9c931de2d81d4.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70 1]: /images/20221119/9bba5c44e1ea4e4889c9c1093d32d6a0.png
[20210120213940705.png]: /images/20221119/9c69b23b4bc048e3bb7ae54a3c9f36a9.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70 2]: /images/20221119/139862c8427249ecaacc78e5b8b27105.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70 3]: /images/20221119/b6425f9ef49a4345aaa0a4e4fb899609.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQyNDQ5OTYz_size_16_color_FFFFFF_t_70 4]: /images/20221119/5d7120f6dafd43b3bafbbae6392300ba.png