@SpringBootTest中Spring WebMvc应用Context-Propagation失效问题

以下是针对该问题的具体解决方案：

1. 校验ThreadLocalAccessor实现与注册 #

确保ThreadLocalAccessor逻辑正确且被Spring容器管理：

• 实现时指定对应泛型（比如ThreadLocal<Map<String, Object>>对应ThreadLocalAccessor<Map<String, Object>>）
• 重写key()返回唯一标识，get()返回ThreadLocal值，set()存入值，reset()清空ThreadLocal
• 将Accessor注册为Spring Bean，让Spring自动处理上下文绑定

示例实现：

@Component
public class RequestResponseContextAccessor implements ThreadLocalAccessor<Map<String, Object>> {
    public static final String KEY = "request-response-context";
    private static final ThreadLocal<Map<String, Object>> CONTEXT_THREAD_LOCAL = new ThreadLocal<>();

    @Override
    public String key() {
        return KEY;
    }

    @Override
    public Map<String, Object> getValue() {
        return CONTEXT_THREAD_LOCAL.get();
    }

    @Override
    public void setValue(Map<String, Object> value) {
        CONTEXT_THREAD_LOCAL.set(value);
    }

    @Override
    public void reset() {
        CONTEXT_THREAD_LOCAL.remove();
    }
}

2. 调整WebClient过滤器的上下文绑定逻辑 #

通过Reactor Context桥接ThreadLocal，避免线程切换时上下文丢失：

@Bean
public WebClient webClient(RequestResponseContextAccessor accessor) {
    ExchangeFilterFunction requestFilter = ExchangeFilterFunction.ofRequestProcessor(clientRequest ->
        Mono.deferContextual(ctx -> {
            Map<String, Object> contextMap = ctx.getOrDefault(RequestResponseContextAccessor.KEY, new HashMap<>());
            contextMap.put("request", clientRequest);
            accessor.setValue(contextMap);
            return Mono.just(clientRequest).contextWrite(c -> c.put(RequestResponseContextAccessor.KEY, contextMap));
        })
    );

    ExchangeFilterFunction responseFilter = ExchangeFilterFunction.ofResponseProcessor(clientResponse ->
        Mono.deferContextual(ctx -> {
            Map<String, Object> contextMap = ctx.get(RequestResponseContextAccessor.KEY);
            contextMap.put("response", clientResponse);
            accessor.setValue(contextMap);
            return Mono.just(clientResponse);
        }).doFinally(signal -> accessor.reset())
    );

    return WebClient.builder()
        .filter(requestFilter)
        .filter(responseFilter)
        .build();
}

3. 确保测试类中上下文传播钩子提前初始化 #

在测试类静态初始化方法中显式启用Reactor上下文传播钩子：

@SpringBootTest
@AutoConfigureMockMvc
public class YourControllerIntegrationTest {
    @BeforeAll
    static void setupContextPropagation() {
        Hooks.enableAutomaticContextPropagation();
    }

    @Autowired
    private MockMvc mockMvc;

    // 测试方法...
}

4. 规范测试客户端的使用方式 #

• 依赖Spring注入的MockMvc或WebTestClient实例，不要手动构建脱离Spring上下文的客户端：

@Autowired
private WebTestClient webTestClient;

@Test
void testThirdPartyCallContextPropagation() {
    webTestClient.post().uri("/your-api-endpoint")
        .bodyValue(new YourRequestPayload())
        .exchange()
        .expectStatus().isOk()
        .expectBody(YourResponse.class);
}

5. 校验依赖版本兼容性 #

确保micrometer-context-propagation与Spring Boot版本匹配：

• Spring Boot 3.x 对应1.1.x及以上版本
• Spring Boot 2.7.x 对应1.0.x版本
• 推荐依赖Spring Boot的依赖管理，不手动指定版本：

<dependency>
    <groupId>io.micrometer</groupId>
    <artifactId>micrometer-context-propagation</artifactId>
</dependency>

6. 清理测试间的ThreadLocal污染 #

在测试方法后清空ThreadLocal，避免用例间上下文污染：

@Autowired
private RequestResponseContextAccessor accessor;

@AfterEach
void cleanUpThreadLocal() {
    accessor.reset();
}

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