feign调用请求header信息丢失问题

1.定义FeignSupportConfig

@Configuration
public class FeignSupportConfig {
    @Bean
    public RequestInterceptor requestInterceptor(){
        return new MyRequestInterceptor();
    }

    @Bean
    Logger.Level feignLoggerLevel() {
        return Logger.Level.FULL;
    }
}

2.自定义RequestInterceptor

@Component
public class MyRequestInterceptor implements RequestInterceptor {

    private static final String X_FORWARDED_FOR = "X-Forwarded-For";

    private static final String PROXY_CLIENT_IP = "Proxy-Client-IP";

    private static final String WL_PROXY_CLIENT_IP = "WL-Proxy-Client-IP";

    private static final String X_REAL_IP = "X-Real-IP";


    @Override
    public void apply(RequestTemplate requestTemplate) {
        if (RequestContextHolder.getRequestAttributes() == null) {
            return;
        }
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        String tokenHead = attributes.getRequest().getHeader(X_FORWARDED_FOR);
        if (!Strings.isNullOrEmpty(tokenHead) && !"null".equalsIgnoreCase(tokenHead)) {
            requestTemplate.header(X_FORWARDED_FOR, tokenHead);
        }

        String proxyHead = attributes.getRequest().getHeader(PROXY_CLIENT_IP);
        if (!Strings.isNullOrEmpty(proxyHead) && !"null".equalsIgnoreCase(proxyHead)) {
            requestTemplate.header(PROXY_CLIENT_IP, proxyHead);
        }

        String wlProxyHead = attributes.getRequest().getHeader(WL_PROXY_CLIENT_IP);
        if (!Strings.isNullOrEmpty(wlProxyHead) && !"null".equalsIgnoreCase(wlProxyHead)) {
            requestTemplate.header(WL_PROXY_CLIENT_IP, wlProxyHead);
        }

        String realHead = attributes.getRequest().getHeader(X_REAL_IP);
        if (!Strings.isNullOrEmpty(realHead) && !"null".equalsIgnoreCase(realHead)) {
            requestTemplate.header(X_REAL_IP, realHead);
        }
    }
}

3.【核心】自定义HystrixConcurrencyStrategy,否则RequestContextHolder.getRequestAttributes()为null

@Component
@Slf4j
public class RequestAttributeStrategy extends HystrixConcurrencyStrategy {

    public RequestAttributeStrategy() {
        HystrixPlugins.reset();
        HystrixPlugins.getInstance().registerConcurrencyStrategy(this);
    }

    @Override
    public <T> Callable<T> wrapCallable(Callable<T> callable) {
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        return new WrappedCallable<>(callable, requestAttributes);
    }

    static class WrappedCallable<T> implements Callable<T> {

        private final Callable<T> target;
        private final RequestAttributes requestAttributes;

        public WrappedCallable(Callable<T> target, RequestAttributes requestAttributes) {
            this.target = target;
            this.requestAttributes = requestAttributes;
        }

        @Override
        public T call() throws Exception {
            try {
                RequestContextHolder.setRequestAttributes(requestAttributes);
                return target.call();
            } finally {
                RequestContextHolder.resetRequestAttributes();
            }
        }
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值