WebClient 超時后如何優雅地處理異常
在使用 WebClient 時，超時是一個常見的問題。當請求超時時，WebClient 會拋出 TimeoutException。為了優雅地處理這些異常，可以采用以下幾種方法：【起飛嘎嘎飛LSIXSO】
一、使用 onErrorResume 或 onErrorReturn 處理超時
onErrorResume 和 onErrorReturn 是 WebClient 提供的兩個操作符，用于在發生錯誤時提供替代的響應。
（一）onErrorResume
onErrorResume 允許你在發生錯誤時返回一個替代的 Mono 或 Flux。
java復制
import org.springframework.web.reactive.function.client.WebClient;
import java.time.Duration;
public class WebClientTimeoutExample {
public static void main(String[] args) {
WebClient client = WebClient.builder()
.baseUrl("https://api.example.com")
.build();
client.get()
.uri("/data")
.retrieve()
.bodyToMono(String.class)
.timeout(Duration.ofSeconds(5)) // 設置請求級超時為 5 秒
.onErrorResume(e -> Mono.just("Default response")) // 超時后返回默認響應
.subscribe(
response -> System.out.println("Response: " + response),
error -> System.err.println("Error: " + error.getMessage())
);
}
}（二）onErrorReturn
onErrorReturn 允許你在發生錯誤時返回一個特定的值。
java復制
client.get()
.uri("/data")
.retrieve()
.bodyToMono(String.class)
.timeout(Duration.ofSeconds(5))
.onErrorReturn("Default response")
.subscribe(
response -> System.out.println("Response: " + response),
error -> System.err.println("Error: " + error.getMessage())
);二、使用 retryWhen 實現重試機制
retryWhen 允許你定義一個重試策略，當請求失敗時自動重試。這可以提高系統的可靠性和容錯能力。
java復制
import org.springframework.web.reactive.function.client.WebClient;
import java.time.Duration;
public class WebClientTimeoutExample {
public static void main(String[] args) {
WebClient client = WebClient.builder()
.baseUrl("https://api.example.com")
.build();
client.get()
.uri("/data")
.retrieve()
.bodyToMono(String.class)
.timeout(Duration.ofSeconds(5)) // 設置請求級超時為 5 秒
.retryWhen(Retry.backoff(3, Duration.ofSeconds(2))) // 重試 3 次，每次間隔 2 秒
.subscribe(
response -> System.out.println("Response: " + response),
error -> System.err.println("Error: " + error.getMessage())
);
}
}（一）自定義異常處理
你可以在 retryWhen 中定義更復雜的重試策略，例如在重試前執行某些操作。
java復制
client.get()
.uri("/data")
.retrieve()
.bodyToMono(String.class)
.timeout(Duration.ofSeconds(5))
.retryWhen(Retry.backoff(3, Duration.ofSeconds(2))
.doBeforeRetry(retrySignal -> System.out.println("Retrying...")))
.subscribe(
response -> System.out.println("Response: " + response),
error -> System.err.println("Error: " + error.getMessage())
);三、使用 onStatus 處理特定狀態碼
onStatus 允許你根據 HTTP 狀態碼處理錯誤。這可以用于處理 4xx 和 5xx 錯誤。
java復制
import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.http.HttpStatus;
public class WebClientTimeoutExample {
public static void main(String[] args) {
WebClient client = WebClient.builder()
.baseUrl("https://api.example.com")
.build();
client.get()
.uri("/data")
.retrieve()
.onStatus(HttpStatus::is4xxClientError, response -> Mono.error(new RuntimeException("Client error")))
.onStatus(HttpStatus::is5xxServerError, response -> Mono.error(new RuntimeException("Server error")))
.bodyToMono(String.class)
.timeout(Duration.ofSeconds(5)) // 設置請求級超時為 5 秒
.subscribe(
response -> System.out.println("Response: " + response),
error -> System.err.println("Error: " + error.getMessage())
);
}
}四、全局異常處理
你可以通過定義一個全局的異常處理邏輯來處理所有 WebClient 請求的異常。
java復制
import org.springframework.web.reactive.function.client.ExchangeFilterFunction;
import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.http.HttpStatus;
public class WebClientConfig {
public WebClient createWebClient() {
return WebClient.builder()
.filter(ExchangeFilterFunction.ofResponseProcessor(clientResponse -> {
if (clientResponse.statusCode().is4xxClientError()) {
return clientResponse.bodyToMono(String.class)
.flatMap(errorBody -> Mono.error(new RuntimeException("Client error: " + errorBody)));
} else if (clientResponse.statusCode().is5xxServerError()) {
return clientResponse.bodyToMono(String.class)
.flatMap(errorBody -> Mono.error(new RuntimeException("Server error: " + errorBody)));
}
return Mono.just(clientResponse);
}))
.build();
}
}五、總結
在使用 WebClient 時，超時是一個常見的問題。通過使用 onErrorResume、onErrorReturn、retryWhen 和 onStatus 等操作符，可以優雅地處理這些異常。這些方法不僅可以提高系統的可靠性，還可以提供更好的用戶體驗。
