Spring WebClient:如何将大型byte[]流式传输到文件？

基础概念

Spring WebClient 是 Spring 5 引入的一个用于构建非阻塞的、响应式的 HTTP 客户端的工具。它基于 Reactor 项目，提供了强大的流式处理能力，非常适合处理大型数据传输。

类型

WebClient 支持多种类型的请求和响应处理，包括：

GET 请求
POST 请求
PUT 请求
DELETE 请求
其他 HTTP 方法

应用场景

适用于需要处理大量数据传输的场景，如文件下载、大数据处理、实时数据流等。

流式传输大型 byte[] 到文件

以下是一个示例代码，展示如何使用 WebClient 将大型 byte[] 流式传输到文件：

import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.http.HttpStatus;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import reactor.core.scheduler.Schedulers;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.channels.FileChannel;
import java.nio.file.Path;
import java.nio.file.Paths;

public class WebClientFileDownload {

    public static void main(String[] args) {
        String url = "https://example.com/large-file.bin";
        Path filePath = Paths.get("downloaded-file.bin");

        WebClient webClient = WebClient.create();

        Mono<Void> download = webClient.get()
                .uri(url)
                .exchangeToFlux(response -> {
                    if (response.statusCode().equals(HttpStatus.OK)) {
                        return response.bodyToFlux(DataBuffer.class);
                    } else {
                        return Flux.error(new RuntimeException("Failed to download file"));
                    }
                })
                .doOnNext(dataBuffer -> {
                    try (FileOutputStream fos = new FileOutputStream(filePath.toFile());
                         FileChannel channel = fos.getChannel()) {
                        channel.write(dataBuffer.asByteBuffer());
                    } catch (IOException e) {
                        throw new RuntimeException("Failed to write file", e);
                    }
                })
                .doOnError(throwable -> System.err.println("Error during download: " + throwable.getMessage()))
                .then();

        download.block();
    }
}

解释

创建 WebClient 实例：使用 WebClient.create() 创建一个 WebClient 实例。
发起 GET 请求：使用 webClient.get().uri(url).exchangeToFlux() 发起 GET 请求，并将响应转换为 Flux<DataBuffer>。
处理响应：检查响应状态码，如果是 200 OK，则继续处理响应体。
流式写入文件：使用 doOnNext 操作符将每个 DataBuffer 写入文件。这里使用了 FileOutputStream 和 FileChannel 来高效地写入文件。
错误处理：使用 doOnError 操作符处理下载过程中可能出现的错误。