如何为Spring Webflux WebClient Wiretap日志更新ThreadContext

核心原因是WebClient的Wiretap运行在Reactor的异步线程池中，ThreadContext（即Logback的MDC）是线程局部变量，主线程的上下文不会自动传递到异步线程。以下是几种可行的解决方式：

方式一：借助Reactor Context手动传递并注入ThreadContext #

1. 添加请求过滤器，将主线程ThreadContext存入Reactor Context #

ExchangeFilterFunction threadContextPropagator = ExchangeFilterFunction.ofRequestProcessor(clientRequest -> {
    // 复制当前主线程的ThreadContext内容
    Map<String, String> threadContextMap = ThreadContext.getCopyOfContextMap();
    // 将上下文绑定到当前请求的Reactor Context中
    return Mono.just(clientRequest)
            .contextWrite(ctx -> ctx.put("THREAD_CONTEXT", threadContextMap));
});

2. 配置WebClient时注册过滤器，并自定义Wiretap日志逻辑 #

WebClient webClient = WebClient.builder()
        .filter(threadContextPropagator)
        .clientConnector(new ReactorClientHttpConnector(HttpClient.create()
                .wiretap("your.webclient.logger.name", LogLevel.INFO, AdvancedByteBufFormat.TEXTUAL,
                        (logger, event) -> {
                            // 从Reactor Context中取出上下文，注入到当前Wiretap线程的ThreadContext
                            Mono.subscriberContext()
                                    .map(ctx -> ctx.getOrDefault("THREAD_CONTEXT", Collections.emptyMap()))
                                    .subscribe(contextMap -> {
                                        ThreadContext.putAll(contextMap);
                                        // 执行日志输出
                                        logger.info(event.toString());
                                        // 清理ThreadContext，避免线程池复用导致上下文污染
                                        ThreadContext.clearAll();
                                    });
                        })))
        .build();

方式二：使用Reactor官方的MDC上下文传播工具（推荐） #

如果你的项目依赖Logback（ThreadContext就是其MDC实现），可以直接使用Reactor提供的MdcContextPropagator来自动完成上下文传递：

1. 引入依赖（Spring Boot项目可直接使用） #

<dependency>
    <groupId>io.projectreactor.tools</groupId>
    <artifactId>reactor-tools</artifactId>
</dependency>

2. 注册上下文传播器 #

@Configuration
public class ReactorContextConfig {
    @PostConstruct
    public void registerMdcPropagator() {
        ContextRegistry.getInstance().register(MdcContextPropagator.create());
    }
}

完成配置后，Reactor会自动将主线程的MDC（ThreadContext）内容传递到所有异步线程，包括Wiretap运行的线程，无需修改Wiretap的日志逻辑，日志会自动包含主线程设置的上下文值。

方式三：自定义ContextPropagator（适用于自定义ThreadContext场景） #

如果需要自定义上下文传递逻辑，可以实现Reactor的ContextPropagator接口：

public class ThreadContextPropagator implements ContextPropagator<Map<String, String>> {
    @Override
    public Map<String, String> extract(ContextView contextView) {
        // 提取当前线程的ThreadContext内容
        return ThreadContext.getCopyOfContextMap();
    }

    @Override
    public void inject(Map<String, String> context, ContextView contextView, Supplier<Context> contextSupplier) {
        // 将上下文注入到当前线程的ThreadContext
        if (context != null && !context.isEmpty()) {
            ThreadContext.putAll(context);
        }
    }

    @Override
    public Object key() {
        // 定义传播器的唯一标识
        return ThreadContextPropagator.class;
    }
}

然后在配置类中注册：

@Configuration
public class ReactorConfig {
    @PostConstruct
    public void setupPropagation() {
        ContextRegistry.getInstance().register(new ThreadContextPropagator());
    }
}

内容的提问来源于stack exchange，提问作者mememoremore