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 socket
        at 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转换成上传流uploadInputStream
    InputStream 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

谢谢支持。