RestTemplate文件上传下载与大文件流式下载

@SpringBootTest

class UpDownLoadTests {

    @Resource

    private RestTemplate restTemplate;

    @Test

    void testUpload()  {

       // 文件上传服务上传接口

       String url = "http://localhost:8888/upload" ;

       // 待上传的文件（存在客户端本地磁盘）

       String filePath = "D:\\data\\local\\splash.png" ;

       // 封装请求参数

       FileSystemResource resource = new FileSystemResource( new File(filePath));

       MultiValueMap<String, Object> param = new LinkedMultiValueMap<>();

       param.add( "uploadFile" , resource);  //服务端MultipartFile uploadFile

       //param.add("param1", "test");   //服务端如果接受额外参数，可以传递

       // 发送请求并输出结果

       System.out.println( "--- 开始上传文件 ---" );

       String result = restTemplate.postForObject(url, param, String. class );

       System.out.println( "--- 访问地址：" + result);

    }

}

@Test

void testDownLoad() throws IOException {

    // 待下载的文件地址

    String url = "http://localhost:8888/2020/08/12/028b38f1-3f9b-4088-9bea-1af8c18cd619.png" ;

    ResponseEntity< byte []> rsp = restTemplate.getForEntity(url, byte []. class );

    System.out.println( "文件下载请求结果状态码：" + rsp.getStatusCode());

    // 将下载下来的文件内容保存到本地

    String targetPath = "D:\\data\\local\\splash-down.png" ;

    Files.write(Paths.get(targetPath), Objects.requireNonNull(rsp.getBody(),

                "未获取到下载文件" ));

}

@Test

void testDownLoadBigFile() throws IOException {

    // 待下载的文件地址

    String url = "http://localhost:8888/2020/08/12/028b38f1-3f9b-4088-9bea-1af8c18cd619.png" ;

    // 文件保存的本地路径

    String targetPath = "D:\\data\\local\\splash-down-big.png" ;

    //定义请求头的接收类型

    RequestCallback requestCallback = request -> request.getHeaders()

                .setAccept(Arrays.asList(MediaType.APPLICATION_OCTET_STREAM, MediaType.ALL));

    //对响应进行流式处理而不是将其全部加载到内存中

    restTemplate.execute(url, HttpMethod.GET, requestCallback, clientHttpResponse -> {

       Files.copy(clientHttpResponse.getBody(), Paths.get(targetPath));

       return null ;

    });

}