java - 如何通过 Spring-Feign 获取 InputStream？

Question

我想通过 Spring-OpenFeign 从服务器下载文件并将其保存在本地目录中，并进行零拷贝。

天真的下载方法如下：

import org.apache.commons.io.FileUtils

@GetMapping("/api/v1/files")
ResponseEntity<byte[]> getFile(@RequestParam(value = "key") String key) {
    ResponseEntity<byte[]> resp = getFile("filename.txt")
    File fs = new File("/opt/test")
    FileUtils.write(file, resp.getBody())
}

在这段代码中，数据流将是这样的feign Internal Stream -> Buffer -> ByteArray -> Buffer -> File

如何以更高效、更快的方式下载和保存文件？

Answer 1

TL;博士。使用ResponseEntity<InputStreamResource>和Java NIO

根据SpringDecoder，Spring 使用 HttpMessageConverters 解码响应

ResourceHttpMessageConverter 是 HttpMesageConverters 之一，它返回InputStreamResource，其中包含 InputStream 和派生自Content-Disposition.

但是，必须初始化 ResourceHttpMessageConvertersupportsReadStreaming = true (default value) 如果您对此实现有进一步的兴趣，请检查此代码。

所以，修改后的代码如下：

@GetMapping("/api/v1/files")
ResponseEntity<InputStreamResource> getFile(@RequestParam(value = "key") String key)

JDK9

try (OutputStream os = new FileOutputStream("filename.txt")) {
    responeEntity.getBody().getInputStream().transferTo(os);
}

JDK8以下

使用 Guava ByteStreams.copy()

Path p = Paths.get(responseEntity.getFilename())
ReadableByteChannel rbc = Channels.newChannel(responeEntity.getBody().getInputStream())
try(FileChannel fc = FileChannel.open(p, StandardOpenOption.WRITE)) {
    ByteStreams.copy(rbc, fc)
}

现在，Feign Internal Stream -> File