gateway网关配置优先请求到本机的服务

熊一族

已于 2023-06-02 16:19:57 修改

阅读量687

点赞数

文章标签： gateway java 开发语言

于 2023-06-02 16:19:31 首次发布

本文为博主原创文章，未经博主允许不得转载。

本文链接：https://blog.youkuaiyun.com/FlowerDance17/article/details/131009298

版权

该代码段展示了一个在SpringCloudGateway中实现的全局过滤器`PreRequestFilter`，用于处理每个请求。它获取客户端IP，添加到请求头中，并根据配置决定是否生成和添加TraceId进行追踪。TraceId使用`UUIDUtil`生成，并存储在MDC中。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

@Slf4j
@Component
@RequiredArgsConstructor
public class PreRequestFilter implements GlobalFilter, Ordered {

    private final GridinterRequestProperties gridinterRequestProperties;

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        String clientIp = IPUtil.getIpAddr(exchange.getRequest());
        ServerHttpRequest.Builder mutate = exchange.getRequest().mutate();
        mutate.headers(h->h.add(IPUtil.IP_KEY,clientIp));
        // 是否开启traceId追踪
        if (gridinterRequestProperties.getTrace()) {
            // ID生成
            String traceId = UUIDUtil.shortUuid();
            MDC.put(GridinterConstant.LOG_TRACE_ID, traceId);
            mutate.headers(h -> h.add(GridinterConstant.GRIDINTER_TRACE_ID, traceId));
        }
        ServerHttpRequest serverHttpRequest = mutate.build();
        ServerWebExchange build = exchange.mutate().request(serverHttpRequest).build();
        return chain.filter(build);
    }

    @Override
    public int getOrder() {
        return Ordered.HIGHEST_PRECEDENCE;
    }
}

关注博主即可阅读全文