Feign——多线程调用Feign导致的Request对象为NULL的问题解决

一、在Feign模块或者核心模块添加一个工具类ThreadLocalUtils，代码如下

package com.zhiyu.common.security.utils;


import org.springframework.util.CollectionUtils;

import java.util.HashMap;
import java.util.Map;

/**
 * 定义多线程的共享工具类——可继承父线程数据的线程变量
 */
public class ThreadLocalUtils {

    /**
     * 定义线程变量
     */
    private static final InheritableThreadLocal<Map<String, String>> headerMap =
            new InheritableThreadLocal<Map<String,String>>(){
                @Override
                protected Map<String, String> initialValue() {
                    return new HashMap<>();
                }
            };

    /**
     * 获取所有共享变量
     * @return 共享Map
     */
    public static Map<String,String> get(){
        return headerMap.get();
    }

    /**
     * 获取共享变量指定key的value值
     * @param key key
     * @return
     */
    public static String get(String key){
        return headerMap.get().get(key);
    }

    /**
     * 设置共享变量值
     * @param header
     */
    public static void setHeaderMap(Map<String, String> header) {
        if (!CollectionUtils.isEmpty(header)) {
            for (Map.Entry<String, String> entry : header.entrySet()) {
                headerMap.get().put(entry.getKey(), entry.getValue());
            }
        }
    }

    /**
     * 设置共享变量值
     * @param key
     * @param value
     */
    public static void set(String key,String value){
        headerMap.get().put(key,value);
    }

}

二、在需要使用异步线程的地方加入如下代码

// 在主线程中获取requestAttributes 
RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
// 获取header，保存到ThreadLocalUtils中
Map<String, String> headers = ServletUtils.getHeaders(ServletUtils.getRequest());
        ThreadLocalUtils.setHeaderMap(headers);

// 在异步线程中使用
//共享主线程请求上下文信息————防止子线程RequestAttributes为空
RequestContextHolder.setRequestAttributes(requestAttributes);

三、在Feign的拦截器上添加如下代码

package com.zhiyu.common.security.feign;

import java.util.Map;
import javax.servlet.http.HttpServletRequest;

import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.zhiyu.common.security.utils.ThreadLocalUtils;
import org.apache.commons.lang3.ObjectUtils;
import org.springframework.stereotype.Component;
import com.zhiyu.common.core.constant.SecurityConstants;
import com.zhiyu.common.core.utils.ServletUtils;
import com.zhiyu.common.core.utils.StringUtils;
import com.zhiyu.common.core.utils.ip.IpUtils;
import feign.RequestInterceptor;
import feign.RequestTemplate;
import org.springframework.web.context.request.RequestAttributes;

/**
 * feign 请求拦截器
 *
 * @author zhiyu
 */
@Component
public class FeignRequestInterceptor implements RequestInterceptor {
    @Override
    public void apply(RequestTemplate requestTemplate) {
        // 原本是从主线程的request对象中获取到headers
        HttpServletRequest httpServletRequest = ServletUtils.getRequest();
        // 异步线程可以从我们刚刚存储的共享变量中获取headers
        Map<String, String> headerMaps = ThreadLocalUtils.get();
        if (StringUtils.isNotNull(httpServletRequest)) {
            Map<String, String> headers = ServletUtils.getHeaders(httpServletRequest);
            if (CollectionUtils.isEmpty(headers) && CollectionUtils.isNotEmpty(headerMaps)) {
                headers = headerMaps;
            }
            // 传递用户信息请求头，防止丢失
            String userId = headers.get(SecurityConstants.DETAILS_USER_ID);
            if (StringUtils.isNotEmpty(userId)) {
                requestTemplate.header(SecurityConstants.DETAILS_USER_ID, userId);
            }
            String userKey = headers.get(SecurityConstants.USER_KEY);
            if (StringUtils.isNotEmpty(userKey)) {
                requestTemplate.header(SecurityConstants.USER_KEY, userKey);
            }
            String userName = headers.get(SecurityConstants.DETAILS_USERNAME);
            if (StringUtils.isNotEmpty(userName)) {
                requestTemplate.header(SecurityConstants.DETAILS_USERNAME, userName);
            }
            String authentication = headers.get(SecurityConstants.AUTHORIZATION_HEADER);
            if (StringUtils.isNotEmpty(authentication)) {
                requestTemplate.header(SecurityConstants.AUTHORIZATION_HEADER, authentication);
            }
            // 配置客户端IP
            requestTemplate.header("X-Forwarded-For", IpUtils.getIpAddr());
        }
    }
}