SpringCloud 之OpenFeign远程调用丢失Header头

Gateway网关丢失请求头解决办法:
在搭建微服务时,使用了SpringSecurity Oauth2认证授权,使用密码方式,从认证中心获取了token后,要将token携带在请求头中,但是发现经过gateway网关后,token丢失了。

解决:通过使用过滤器,重新构建一个request,再向服务发送请求。

@Component
public class RequestAuthFilter implements GlobalFilter, Ordered {

	@Override
	public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
		ServerHttpRequest request = exchange.getRequest();
		String url = request.getURI().getPath();

		//忽略以下url
		if(url.startsWith("/oauth") || url.startsWith("/login")){
			return chain.filter(exchange);
		}

		String token = request.getHeaders().getFirst(HttpHeaders.AUTHORIZATION);
		if(StringUtils.isBlank(token)){
			ServerHttpResponse response = exchange.getResponse();
			response.getHeaders().add(HttpHeaders.CONTENT_TYPE,MediaType.APPLICATION_JSON_UTF8_VALUE);
			exchange.getResponse().setStatusCode(HttpStatus.UNAUTHORIZED);
			return exchange.getResponse().setComplete();
		}else{
			ServerHttpRequest req = request.mutate().headers(header -> header.add(HttpHeaders.AUTHORIZATION, token)).build();
			ServerWebExchange webExchange = exchange.mutate().request(req).build();
			return chain.filter(webExchange);
		}
	}

	@Override
	public int getOrder() {
		return 100;
	}
}

OpenFeigh远程调用丢失Header头:

Feign 在远程调用之前要构造请求,不会带上Header头,会调用很多拦截器。

在这里插入图片描述

在这里插入图片描述

import feign.RequestInterceptor;
import feign.RequestTemplate;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;

@Configuration
public class GuliFeignConfig {
    @Bean
    public RequestInterceptor requestInterceptor() {
        return new RequestInterceptor() {
            @Override
            public void apply(RequestTemplate requestTemplate) {
                //1、RequestContextHolder拿到刚进来的这个请求
                ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
                if (attributes != null) {
                    System.out.println("requestInterceptor线程。。。" + Thread.currentThread().getId());
                    HttpServletRequest request = attributes.getRequest();//老请求
                    if (request != null) {
                        //同步请求头数据。Cookie    给新请求同步了老请求的cookie
                        requestTemplate.header("Cookie", request.getHeader("Cookie"));
                    }
                }
            }
        };
    }
}

或者:

@Component
public class FeignRequestInterceptor implements RequestInterceptor {
    @Override
    public void apply(RequestTemplate requestTemplate) {
        if (Objects.nonNull(sessionHelper.getUserId())) {
            requestTemplate.header("userId", "123");
        }
    }
}

通过Feign拦截器解决了丢失Header头的问题。
但是如果通过异步的方式使用 OpenFeign 还是会丢失 Header 头。
我们发现在拦截器中获取 Request 上下文,最终是从 ThreadLocal 中获取的,如果是一条直线执行下去是没问题的,但是异步的情况下,执行到拦截器中的时候,是开启了另一个线程了,就会获取不到当前的上下文了,导致丢失Header头。

在这里插入图片描述

解决方案:
提前获取当前线程的上下文,然后在异步远程调用的时候将获取到的上下文设置到本地线程,这样在拦截器中就能获取到

// 获取 Request 上下文
RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();

CompletableFuture<Void> getAddressFuture = CompletableFuture.runAsync(() -> {
    // 将 Request 上下文设置到本地线程
    RequestContextHolder.setRequestAttributes(requestAttributes);
   	// 远程调用
}, executor);
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值