Spring Boot 中 WebClient 的实践详解

一、什么是 WebClient？

WebClient 是 Spring WebFlux 提供的非阻塞式 HTTP 客户端，它支持同步和异步的调用方式，适合高并发场景下的服务通信。与传统的 RestTemplate 相比，WebClient 的特点包括：

非阻塞式 I/O：更高的性能，适合处理大量请求。
强大的功能：支持流式处理、拦截器、请求超时等高级功能。
灵活性：支持多种编码方式和请求类型。

二、引入依赖

在使用 WebClient 之前，需要确保你的 Spring Boot 项目已包含相关依赖。以下是常见的 Maven 依赖：

xml 复制代码

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-webflux</artifactId>
</dependency>

三、配置 WebClient

1. 基本配置

WebClient 可以通过静态方法 WebClient.create() 创建，也可以通过 WebClient.Builder 定制。

以下是一个最基本的配置：

kotlin 复制代码

import org.springframework.web.reactive.function.client.WebClient;

@Configuration
public class WebClientConfig {

    @Bean
    public WebClient webClient() {
        return WebClient.create("https://api.example.com");
    }
}

2. 高级配置

为了增强 WebClient 的灵活性，可以使用 WebClient.Builder 来配置全局属性，比如超时设置、全局拦截器等：

kotlin 复制代码

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.reactive.function.client.ExchangeStrategies;
import org.springframework.web.reactive.function.client.WebClient;

@Configuration
public class WebClientConfig {

    @Bean
    public WebClient webClient(WebClient.Builder builder) {
        return builder
                .baseUrl("https://api.example.com")
                .defaultHeader("Authorization", "Bearer your-token")
                .exchangeStrategies(
                    ExchangeStrategies.builder()
                        .codecs(configurer -> configurer
                            .defaultCodecs()
                            .maxInMemorySize(16 * 1024 * 1024)) // 设置最大内存限制为16MB
                        .build())
                .build();
    }
}

四、WebClient 的使用场景

1. 发起 GET 请求

以下示例展示了如何使用 WebClient 发起一个简单的 GET 请求：

kotlin 复制代码

import org.springframework.web.reactive.function.client.WebClient;

@Service
public class ApiService {

    private final WebClient webClient;

    public ApiService(WebClient webClient) {
        this.webClient = webClient;
    }

    public String fetchData() {
        return webClient.get()
                .uri("/data")
                .retrieve()
                .bodyToMono(String.class)
                .block(); // 同步方式获取结果
    }
}

2. 发起 POST 请求

对于 POST 请求，可以发送 JSON 数据：

arduino 复制代码

import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Mono;

@Service
public class ApiService {

    private final WebClient webClient;

    public ApiService(WebClient webClient) {
        this.webClient = webClient;
    }

    public String postData(Object requestData) {
        return webClient.post()
                .uri("/submit")
                .body(Mono.just(requestData), Object.class)
                .retrieve()
                .bodyToMono(String.class)
                .block();
    }
}

五、优化和最佳实践

1. 超时设置

为避免长时间等待，建议为 WebClient 配置超时时间：

java 复制代码

import java.time.Duration;

@Bean
public WebClient webClientWithTimeout(WebClient.Builder builder) {
    return builder
            .baseUrl("https://api.example.com")
            .defaultHeaders(headers -> headers.set("Authorization", "Bearer token"))
            .build()
            .mutate()
            .responseTimeout(Duration.ofSeconds(5)) // 设置响应超时时间
            .build();
}

2. 使用拦截器

拦截器可以用于日志记录或添加全局参数：

kotlin 复制代码

@Bean
public WebClient.Builder webClientBuilder() {
    return WebClient.builder()
            .filter((request, next) -> {
                System.out.println("Request: " + request.url());
                return next.exchange(request);
            });
}

3. 异步调用

WebClient 原生支持异步编程，适合处理高并发请求场景：

arduino 复制代码

public Mono<String> fetchDataAsync() {
    return webClient.get()
            .uri("/data")
            .retrieve()
            .bodyToMono(String.class);
}public Mono<String> fetchDataAsync() {
    return webClient.get()
            .uri("/data")
            .retrieve()
            .bodyToMono(String.class);
}

六、错误处理

1. 使用 onStatus 处理 HTTP 错误

WebClient 提供了灵活的错误处理机制：

vbnet 复制代码

import org.springframework.web.reactive.function.client.WebClientResponseException;

public String fetchWithErrorHandling() {
    return webClient.get()
            .uri("/data")
            .retrieve()
            .onStatus(status -> status.is4xxClientError(),
                response -> Mono.error(new RuntimeException("Client error!")))
            .onStatus(status -> status.is5xxServerError(),
                response -> Mono.error(new RuntimeException("Server error!")))
            .bodyToMono(String.class)

2. 捕获异常

可以通过 doOnError 捕获并处理异常：

scss 复制代码

public Mono<String> fetchWithExceptionHandling() {
    return webClient.get()
            .uri("/data")
            .retrieve()
            .bodyToMono(String.class)
            .doOnError(e -> {
                if (e instanceof WebClientResponseException) {
                    WebClientResponseException ex = (WebClientResponseException) e;
                    System.err.println("Error response: " + ex.getResponseBodyAsString());
                }
            });
}

结语

WebClient 是一个功能强大且灵活的 HTTP 客户端，适合在高并发场景下替代 RestTemplate 使用。在实际项目中，通过合理的配置和优化，可以显著提高服务间通信的效率和可靠性。