由于之前我的网站改版，导致一些小伙伴找不到RestTemplate这一个专栏的文章（共10篇），所以我一次性全部发布在公众号上。大家可以在文末点击“阅读下一篇”，同时期待新的朋友关注我！
RestTemplate是HTTP客户端库，所以为了使用RestTemplate进行文件上传和下载，需要我们先编写服务端的支持文件上传和下载的程序。该程序需要满足如下的需求点（这个程序我就不带着大家写了，很普通）：
-
一个以访问服务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;
});
}
本文分享自微信公众号 - 字母哥课堂（zimug_blog）。
如有侵权，请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”，欢迎正在阅读的你也加入，一起分享。
