如何为WebTestClient添加重试机制？

要在WebTestClient的调用链中嵌入重试逻辑，无需外部循环，核心是利用WebTestClient基于Reactor的特性，直接对exchange()返回的Mono<EntityExchangeResult<T>>应用Reactor的重试操作符。以下是具体实现方案：

实现步骤与代码示例 #

1. 复用WebTestClient实例（可选但推荐） #

先单独构建WebTestClient实例，避免每次请求重复初始化：

WebTestClient webTestClient = WebTestClient.bindToServer()
    .baseUrl(gatewayUrl)
    .build();

2. 嵌入重试逻辑到调用链 #

在exchange()之后添加重试策略，针对竞态条件导致的失败场景（比如状态码非200、响应体为空）进行重试：

String whoAmIBody = webTestClient.get()
    .uri("/whoami")
    .header(HttpHeaders.ACCEPT, "application/json")
    .header(HttpHeaders.CONTENT_TYPE, "application/json")
    .header(HttpHeaders.AUTHORIZATION, "Bearer " + responseBody.getAccessToken())
    .exchange()
    // 配置重试策略：最多3次重试，初始间隔1秒，仅在指定条件下触发
    .retryWhen(Retry.backoff(3, Duration.ofSeconds(1))
        .filter(exchangeResult -> {
            // 自定义重试触发条件：状态码非2xx 或 响应体为空
            HttpStatus status = exchangeResult.getStatus();
            boolean shouldRetry = !status.is2xxSuccessful();
            if (!shouldRetry) {
                String body = exchangeResult.getResponseBody(String.class);
                shouldRetry = body == null;
            }
            return shouldRetry;
        })
        .doBeforeRetry(retrySignal -> {
            // 可选：打印重试日志，便于调试
            System.out.printf("重试第%d次，原因：%s%n", 
                retrySignal.totalRetries() + 1, retrySignal.failure());
        }))
    .expectStatus()
    .isEqualTo(HttpStatus.OK)
    .expectBody(String.class)
    .returnResult()
    .getResponseBody();

assertThat(whoAmIBody).isNotNull();

关键细节说明 #

• 重试策略选择：Retry.backoff实现指数退避重试（每次间隔递增），比简单的retry(3)更适合应对竞态条件，避免短时间内重复请求加剧资源竞争。
• 自定义重试条件：通过filter方法精准控制触发重试的场景，只针对实际由竞态条件导致的失败重试，避免无意义重试。
• 链化逻辑保持：所有操作都在WebTestClient的调用链中完成，无需额外的循环或包裹代码，符合链式调用的风格。

依赖导入 #

确保项目中引入Reactor相关依赖（Spring Boot项目默认已包含），需要导入的类：

import reactor.util.retry.Retry;
import java.time.Duration;
import org.springframework.http.HttpStatus;

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