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);