乐趣区

关于spring:精讲RestTemplate第6篇文件上传下载与大文件流式下载

jiezi

本文是精讲 RestTemplate 第 6 篇，前篇的 blog 拜访地址如下：

  • 精讲 RestTemplate 第 1 篇 - 在 Spring 或非 Spring 环境下如何应用
  • 精讲 RestTemplate 第 2 篇 - 多种底层 HTTP 客户端类库的切换
  • 精讲 RestTemplate 第 3 篇 -GET 申请应用办法详解
  • 精讲 RestTemplate 第 4 篇 -POST 申请办法应用详解
  • 精讲 RestTemplate 第 5 篇 -DELETE、PUT 等申请办法应用详解

RestTemplate 是 HTTP 客户端库，所以为了应用 RestTemplate 进行文件上传和下载，须要咱们先编写服务端的反对文件上传和下载的程序。请参考我之前写的一篇文章：SpringBoot 实现本地存储文件上传及提供 HTTP 拜访服务。依照此文实现学习之后，能够取得

  • 一个以拜访服务 URI 为 ”/upload”的文件上传服务端点
  • 服务端点上传文件胜利后会返回一个 HTTP 连贯, 能够用来下载文件。

上面咱们就开始学习应用 RestTemplate 是 HTTP 客户端库，进行文件的上传与下载。

一、文件上传

写一个单元测试类，来实现 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);
   }

}

输入后果如下：

--- 开始上传文件 ---
--- 拜访地址：http://localhost:8888/2020/08/12/028b38f1-3f9b-4088-9bea-1af8c18cd619.png

文件上传之后，能够通过下面的拜访地址，在浏览器拜访。或者通过 RestTemplate 客户端进行下载。

二、文件下载

执行下列代码之后，被下载文件 url，会被正确的保留到本地磁盘目录 targetPath。

@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(),
               "未获取到下载文件"));
}

这种下载办法实际上是将下载文件一次性加载到客户端本地内存，而后从内存将文件写入磁盘。这种形式对于小文件的下载还比拟适宜，如果文件比拟大或者文件下载并发量比拟大，容易造成内存的大量占用，从而升高利用的运行效率。

三、大文件下载

这种下载方式的区别在于

  • 设置了申请头 APPLICATION_OCTET_STREAM，示意以流的模式进行数据加载
  • RequestCallback 联合 File.copy 保障了接管到一部分文件内容，就向磁盘写入一部分内容。而不是全副加载到内存，最初再写入磁盘文件。
@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;
   });
}

欢送关注我的博客，外面有很多精品合集

  • 本文转载注明出处（必须带连贯，不能只转文字）：字母哥博客。

感觉对您有帮忙的话，帮我点赞、分享！您的反对是我不竭的创作能源！。另外，笔者最近一段时间输入了如下的精品内容，期待您的关注。

  • 《手摸手教你学 Spring Boot2.0》
  • 《Spring Security-JWT-OAuth2 一本通》
  • 《实战前后端拆散 RBAC 权限管理系统》
  • 《实战 SpringCloud 微服务从青铜到王者》
  • 《VUE 深入浅出系列》
退出移动版