接口间传输文件流

1、业务背景

　　外部服务请求文件信息，通过对外暴露的前置服务，请求到应用服务器上，应用服务器从文件存储平台下载文件，并返回文件流给外部服务。请求流程如下：

  

 　　这里我们做个简化，假设文件存储在了应用服务器中，前置服务、应用服务在同一台服务器上。

2、代码实现

2.1、pom依赖

 <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.1.6.RELEASE</version>
    </parent>
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
        </dependency>
        <dependency>
            <groupId>org.apache.httpcomponents</groupId>
            <artifactId>httpclient</artifactId>
            <version>4.5.13</version>
        </dependency>

        <!--    Java集合增强    -->
        <dependency>
            <groupId>commons-collections</groupId>
            <artifactId>commons-collections</artifactId>
            <version>3.2.2</version>
        </dependency>
        <!-- 通用工具包  -->
        <dependency>
            <groupId>org.apache.commons</groupId>
            <artifactId>commons-lang3</artifactId>
            <version>3.8.1</version>
        </dependency>
    </dependencies>

2.2、汇总实现

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLConnection;


@Controller
public class FileStreamController {

    /**  前置服务的实现
     *   前置服务接收获取文件请求
     * @param fileName  文件名称
     * @param resp      响应
     * @throws Exception
     */
    @RequestMapping("/front/file/{fileName}")
    public void getFile(@PathVariable("fileName") String fileName, HttpServletResponse resp) throws Exception {
        // 应用服务地址
        String appurl = "http://localhost:8085/download/" + fileName;

        int connectTimeOut = Integer.valueOf(30000);
        int readTimeOut = Integer.valueOf(100) * 1000;

        HttpURLConnection httpUrlConnection = null;
        URL url = null;
        InputStream inStrm = null;
        try{
            url = new URL(appurl);
            URLConnection rulConnection = url.openConnection();
            // 设置连接超时时间
            rulConnection.setConnectTimeout(connectTimeOut);
            // 设置读超时时间
            rulConnection.setReadTimeout(readTimeOut);
            httpUrlConnection = (HttpURLConnection) rulConnection;
            // 开启向服务器发送信息标志
            httpUrlConnection.setDoOutput(true);
            // 开启从服务器获取信息标志
            httpUrlConnection.setDoInput(true);
            // 关闭关村
            httpUrlConnection.setUseCaches(false);
            // 请求类型
            httpUrlConnection.setRequestProperty("Content-Type", "text/xml;charset=utf-8");
            // 请求方式
            httpUrlConnection.setRequestMethod("POST".toUpperCase());

            // 获取应用服务的输入流
            inStrm = httpUrlConnection.getInputStream();

            //设置响应头类型
            resp.setContentType("multipart/form-data");
            //设置响应头打开方式，并设置下载时的文件名称
            resp.setHeader("content-disposition","attachment;filename=" + fileName);
            // 获取响应的输出流，输出到浏览器
            ServletOutputStream sos = resp.getOutputStream();
            byte[] buffer = new byte[1024];
            int length = 0;
            while((length = inStrm.read(buffer)) != -1){
                sos.write(buffer,0,length);
                sos.flush();
            }
            sos.close();
            inStrm.close();
        }catch (Exception ex) {
            ex.printStackTrace();
        }
    }

    /**  应用服务的实现
     *   应用服务获取文件
     * @param fileName 文件名称
     * @param response 响应
     * @return
     * @throws Exception
     */
    @RequestMapping("/download/{fileName}")
    public HttpServletResponse downloadFile(@PathVariable("fileName") String fileName, HttpServletResponse response) throws Exception {
        // 指定文件地址，这里假设目标文件存储在D盘
        String filePath = "D:\\" + fileName;
        // 获取文件输入流
        FileInputStream inputStream = new FileInputStream(new File(filePath));
        // 将文件流传输到前置服务
        byte[] buffer = new byte[1024];
        int len;
        ServletOutputStream outputStream = response.getOutputStream();
        while ((len = inputStream.read(buffer)) != -1) {
            outputStream.write(buffer, 0 , len);
            outputStream.flush();
        }
        return response;

    }

}

2.3、配置文件及启动类

2.3.1、配置文件

server:
  port: 8085

2.3.2、启动类

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication
public class FileStreamApplication {
    public static void main(String[] args) {
        SpringApplication.run(FileStreamApplication.class, args);
    }
}

3、测试

3.1、在D盘上准备好要下载的文件

　　这里我在D盘上准备了要下载的Test压缩包Test.tar.gz。

    

3.2、启动服务浏览器请求前置服务下载文件

    

　　文件下载完成。