从其余模板中获取打开的输入流,以进行大文件处理

时间:2016-01-21 23:06:52

标签: java spring rest spring-mvc resttemplate

我正在寻找一种从rest模板中获取打开的输入流的方法 - 我试图使用ResponseExtractor,但是在返回之前流已经关闭,如下所示:

https://jira.spring.io/browse/SPR-7357

“请注意,您不能简单地从提取器返回InputStream,因为在execute方法返回时,底层连接和流已经关闭”

我希望有一种方法,我不必直接在其余模板中写入输出流。

1 个答案:

答案 0 :(得分:1)

我没有办法做到这一点,流总是关闭。作为一种解决方法,我创建了以下代码:

public interface ResourceReader {
    void read(InputStream content);
}

具有以下实现:

public class StreamResourceReader implements ResourceReader {

private HttpServletResponse response;

public StreamResourceReader(HttpServletResponse response) {
    this.response = response;
}

@Override
public void read(InputStream content) {
    try {
        IOUtils.copy(content, response.getOutputStream());
    } catch (IOException e) {
        throw new IllegalStateException(e);
    }
}
}

然后在控制器中:

@RequestMapping(value = "document/{objectId}")
public void getDocumentContent(@PathVariable String objectId, HttpServletResponse response) {
    ResourceReader reader = new StreamResourceReader(response);
    service.readDocumentContent(objectId, reader);
}

致电休息模板:

restTemplate.execute(uri, HttpMethod.GET, null,
            new StreamResponseExtractor(reader));

和字符串响应提取器:

@Override
public ResponseEntity extractData(ClientHttpResponse response) throws IOException {
    reader.read(response.getBody());
    return null;
}

它就像一个魅力! :)