Spring Rest将流响应模板化为另一个请求

Question

我正在尝试使用spring的RestTemplate将文件下载结果直接流式传输到另一个帖子中

我目前的做法如下：

   ResponseEntity<InputStreamResource> downloadResponse = restTemplate.getForEntity(fileToDownloadUri, InputStreamResource.class);

   InputStreamResource imageInputStreamResource = downloadResponse.getBody();

   ResponseEntity<String> response = restTemplate.exchange(storageUri, POST, new HttpEntity<>(imageInputStreamResource), String.class);

但是，在运行上述代码时，我遇到以下异常：

org.springframework.web.client.ResourceAccessException: I/O error on POST request for "http://host:port/path/some.jpg": stream is closed; nested exception is java.io.IOException: stream is closed

    at org.springframework.web.client.RestTemplate.doExecute(RestTemplate.java:6
...
Caused by: java.io.IOException: stream is closed
    at sun.net.www.protocol.http.HttpURLConnection$HttpInputStream.ensureOpen(HttpURLConnection.java:3348)
    at sun.net.www.protocol.http.HttpURLConnection$HttpInputStream.read(HttpURLConnection.java:3373)

似乎响应总是作为处理的最后一步而结束。通过响应，HttpURLConnection关闭，流不再可处理。

我希望能够实现这个场景，而不必将文件完全保存在内存中或将其写入文件（如此处所述）。

非常感谢任何提示。

Answer 1

因为您告诉RestTemplate期待InputStreamResource，所以它将尝试使用适当的转换器将您的消息转换为InputStreamResource。(我猜没有一个能像你希望的那样处理这个问题)

您应该能够让它期待一个资源，从那里您可以获得一个输入流并读取它。

 import org.springframework.core.io.Resource;

 ResponseEntity<Resource> exchange = RestTemplate.exchange(url, HttpMethod.GET, new HttpEntity(httpHeaders), Resource.class);
 InputStream inputStream = exchange.getBody().getInputStream();

使用这个，你可以把响应写到其他地方。Files.write（inputStream， new File（“./test.json”））;为我编写了文件，因此我认为输入流也可以在其他地方使用。（ 我用了Spring 4.3.5 ）

编辑：

正如OP所述，这仍然会将文件加载到内存中。在后台，InputStream是一个ByteArrayInputStream。

默认的RestTemplate和MessageConverters根本不是为流内容而设计的。您可以编写自己的< code > org . spring framework . web . client . response extractor 和MessageConverter的实现。在ResponseExtractor中，您可以访问< code > org . spring framework . http . client . clienthttpresponse

对于您的用例，最好使用Apache Httpcomponents HttpClient，在那里可以找到HttpEntity#writeTo（OutputStream）。

Answer 2

如果要直接转发响应而不将其保存在内存中，则必须直接写入响应：

@RequestMapping(value = "/yourEndPoint")
public void processRequest(HttpServletResponse response) {
    RestTemplate restTemplate = new RestTemplate();

    response.setStatus(HttpStatus.OK.value());

    restTemplate.execute(
        fileToDownloadUri,
        HttpMethod.GET,
        (ClientHttpRequest requestCallback) -> {},
        responseExtractor -> {
            IOUtils.copy(responseExtractor.getBody(), response.getOutputStream());
            return null;
        });
}