Java.SpringBoot InputStream流复制.

本文档产生场景.

在项目中，由于需要一个服务做为文件中转，先定义这个中转服务为C。C服务需要将内部文件服务器（这里定义为A）的文件转移到新的文件服务器（定义为B中）。由于编程经验不足，在C中直接将A的文件读成inputStream流，然后将此inputStream流交给B。在测试环境，由于网络正常，没有出现过任何问题，或者说问题不明显，但在生产环境中，由于网络速度的不对称，造成上传文件时好时坏。以下，将些问题进行记录。

这个问题出现，会报以下错误:

Premature end of Content-Length delimited message body (expected: 11,791,936; received: 3,011,273)

org.springframework.web.util.NestedServletException: Request processing failed; nested exception is org.springframework.web.multipart.MultipartException: Failed to parse multipart servlet request; nested exception is java.io.IOException: org.apache.tomcat.util.http.fileupload.FileUploadBase$IOFileUploadException: Processing of multipart/form-data request failed. Unexpected EOF read on the socket

Caused by: org.apache.tomcat.util.http.fileupload.FileUploadBase$IOFileUploadException: Processing of multipart/form-data request failed. Unexpected EOF read on the socketat org.apache.tomcat.util.http.fileupload.FileUploadBase.parseRequest(FileUploadBase.java:297)at org.apache.catalina.connector.Request.parseParts(Request.java:2884)... 33 more

1. 问题代码

由于不能将所有代码都复制到这里，因此，这里的代码可以认为是半伪代码.

CloseableHttpClient httpClient = HttpClients.createDefault();
// 设置CloseableHttpClient的超时时间.
RequestConfig requestConfig = RequestConfig.custom().setConnectionRequestTimeout(100000).setSocketTimeout(100000).setStaleConnectionCheckEnabled(true).setConnectTimeout(100000).build();
httpPost.setConfig(requestConfig);
// uploadUrl  这个是上传服务器，也是在前面问题说到的B服务地址.
HttpPost httpPost = new HttpPost(uploadUrl);
try {// 此部分只是一个下载地址.// 可以认为是执行下载.InputStream inputStreamTemp = 从A服务器获取的InputStream流// 将获取到的inputStreamTemp流放入BufferedInputStream.// 这一步如果只针对下载文件，可以很好的提高效率，但是如果做为一个中转，是没有办法实现的.BufferedInputStream inputStreamBuffer = new BufferedInputStream(inputStreamTemp,1024*1024*100);// 将inputStreamBuffer放入HttpEntity, 这里的代码未写全，如果需要，可根据实际情况进行写.HttpEntity reqEntity = MultipartEntityBuilder.create().addBinaryBody("file", inputStreamBuffer,ContentType.MULTIPART_FORM_DATA).build();httpPost.setEntity(reqEntity);//执行上传CloseableHttpResponse response = httpClient.execute(httpPost);
}

上面的代码，如果在测试环境，不会出任何问题，因在测试环境的网络非常好，下载和上载都有很高的速度，但是放在生产环境，A和B不在一个网段，因此，这样操作就会时不时的出现问题。

以上就是这段代码的问题所在。接下来，我们需要对这段代码进行更改，让其可以在生产环境中使用。

当然，有人会问到，为什么不先把文件下载到本地，然后再把文件上传到服务器呢？
答：不想在中转服务器生成任何文件或临时文件，不喜欢。

2. 正常代码


CloseableHttpClient httpClient = HttpClients.createDefault();
RequestConfig requestConfig = RequestConfig.custom().setConnectionRequestTimeout(100000).setSocketTimeout(100000).setStaleConnectionCheckEnabled(true).setConnectTimeout(100000).build();
httpPost.setConfig(requestConfig);
HttpPost httpPost = new HttpPost(uploadUrl);
try {InputStream inputStreamTemp = 从A服务器获取的InputStream流**以上都没有作改变，所以把注释就删除了**// 创建一个ByteArrayOutputStream，用于存放复制流.ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();// 将从A服务器获取的InputStream流inputStreamTemp 复制到 byteArrayOutputStream流中.byte[] buffer = new byte[1024];int len;while ((len = inputStreamTemp.read(buffer)) > -1) {byteArrayOutputStream.write(buffer,0,len);}byteArrayOutputStream.flush();// 接下来，将复制的流byteArrayOutputStream转换成上传流uploadInputStreamInputStream uploadInputStream = new ByteArrayInputStream(byteArrayOutputStream.toByteArray());// 将inputStreamBuffer放入HttpEntity, 这里的代码未写全，如果需要，可根据实际情况进行写.HttpEntity reqEntity = MultipartEntityBuilder.create().addBinaryBody("file", inputStreamBuffer,ContentType.MULTIPART_FORM_DATA).build();httpPost.setEntity(reqEntity);//执行上传CloseableHttpResponse response = httpClient.execute(httpPost);
}

以上就是流复制的简单方法，以及代码.

欢迎关注，及提问.

如果有什么问题，可以加入QQ群进行讨论。QQ群：839421316

谢谢支持。