如何在 Project Reactor 中将 DataBuffer 的通量转换为字节数组的 Mono？

Question

有一个Flux<DataBuffer>。将其转换为Mono<byte[]> 的自然方式是什么？

Mono<byte[]> mergeDataBuffers(Flux<DataBuffer> flux){
  // ?
}

Answer 1

使用org.springframework.core.io.buffer.DataBufferUtils 将Flux<DataBuffer> 中的DataBuffers 连接成一个DataBuffer，然后将该缓冲区读入一个字节数组。

    Mono<byte[]> mergeDataBuffers(Flux<DataBuffer> dataBufferFlux) {
        return DataBufferUtils.join(dataBufferFlux)
                .map(dataBuffer -> {
                    byte[] bytes = new byte[dataBuffer.readableByteCount()];
                    dataBuffer.read(bytes);
                    DataBufferUtils.release(dataBuffer);
                    return bytes;
                });
    }

这样做时请务必考虑内存使用情况，因为这种方法会将所有数据加载到内存中两次（一次在 DataBuffers 中，然后在复制到 byte[] 时再次加载）。