spring cloud alibaba实用攻略一 Gateway中基于ThreadLocal实现Controller参数的AOP注解注入

项目在GateWay服务中集成了权限认证以及用户信息等接口,需要经常使用Headers里的信息,但reactor环境下没有实现类似于mvc环境下的RequestContextHolder。通过ReactiveRequestContextFilter获取ServerHttpRequest,基于ThreadLocal实现了ReactiveRequestContextHolder，以便于在AOP中获取Request信息并注入Controller。

ReactiveRequestContextFilter

package com.taylor.gateway.config.context;

import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.Ordered;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.server.ServerWebExchange;
import org.springframework.web.server.WebFilter;
import org.springframework.web.server.WebFilterChain;
import reactor.core.publisher.Mono;
import reactor.util.context.Context;

/**
 *
 * reactor does not contain a {@link RequestContextHolder} object that has the same effect as mvc.
 * If you need to get the current request from the context, you need to manually configure it.
 *
 * Bind ServerWebExchange to {@link Context} via {@link ReactiveRequestContextHolder} after intercepting with filter.
 *
 * from {https://gitee.com/596392912/mica/tree/master/mica-boot/src/main/java/net/dreamlu/mica/reactive/context}
 * 使用 {@link Context} 的方式不能解决此问题 存入成功之后使用 {@link Mono#subscriberContext} 进行获取是无法成功的
 *
 * 改为ThreadLocal的方式
 *
 */
@Configuration
@ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.REACTIVE)
public class ReactiveRequestContextFilter implements WebFilter, Ordered {
   

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
   
        ServerHttpRequest request = exchange.getRequest();
        ReactiveRequestContextHolder.put(request);
        return chain.filter(exchange)
                //Clear request save in ThreadLocal to prevent memory overflow
                .doFinally(s -> ReactiveRequestContextHolder.reset());

        //Invalid way
//        return chain.filter(exchange)
//                .subscriberContext(ctx -> ctx.put(ReactiveRequestContextHolder.CONTEXT_KEY, request));
    }

    @Override
    public int getOrder() {
   
        return -20;
    }
}

ReactiveRequestContextHolder

package com.taylor.gateway.config.context;

import org.springframework.core.NamedThreadLocal;
import org.springframework.http.server.reactive.ServerHttpRequest;

import java.util.Objects;

/**
 *
 * Used to bind the current request and context and provide a way to get the current request
 *
 */
public