如何在Spring Boot中上传下载文件
在日常开发工作中,基本上每个项目都会有各种文件上载和下载。 大多数文件是excel文件。 我使用apache POI操作的excel Java API。 稍后我将专门研究POI。 这次我们将不讨论如何操作文件,而仅讨论如何上载和下载文件。
1. 文件的上传
在现在的JavaWeb项目中,与前端进行交互的基本都是SpringMVC框架。Spring专门提供了一个接口MultipartFile,这个接口帮我们简化了从页面到服务端的文件操作,以前我们只能从request里面拿,现在有了这个接口,我们操作起来就会很简单。话不多说,直接上代码。
我们只是模拟文件上传的操作,所以我们从页面上传文件,然后存到电脑桌面。
@RequestMapping("/upload")
public void upload(MultipartFile file) throws IOException {
init(file, 1);
}
@RequestMapping("/uploads")
public void uploads(MultipartFile[] files, DTO dto) throws IOException {
for (int i = 0; i < files.length; i++) {
init(files[i], i);
}
System.out.println(dto.getParam());
}
/**
* 下载文件初始化
* @param file 文件
* @param i 为了区别文件,防止被覆盖
* @throws IOException
*/
private void init(MultipartFile file, int i) throws IOException {
String filename = file.getOriginalFilename();
OutputStream os = new FileOutputStream(new File("C:\\Users\\WJF\\Desktop" + File.separator + i + filename));
InputStream is = file.getInputStream();
StreamUtils.copy(is, os);
is.close();
os.close();
}
上面的代码中,不管是单个文件还是多文件都是很简单的,还可以带其他参数DTO。
2. 文件的下载
文件的下载也是很简单的,但是Response头如果没有设置好,就可能下载失败,下载下来也可能是乱码的。但是这些头设置其实很简单,但是每次下载文件的时候都要这么设置一下,其实是很繁琐的,所以在这里我使用了已经封装好的下载API,直接上代码。
首先我们引入jar包:
<!--springboot父工程-->
<parent>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>2.2.2.RELEASE</version>
<relativePath/> <!-- lookup parent from repository -->
</parent>
<dependencies>
<!--springboot框架web组件-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<!--wjf-download下载小框架,用来简化下载的无味繁琐的操作-->
<dependency>
<groupId>red.wjf</groupId>
<artifactId>wjf-download</artifactId>
<version>0.1.0.RELEASE</version>
</dependency>
</dependencies>
在这里我引入了一个下载的jar包,这个包可以在maven中央仓库直接找到的。这个下载文件jar包的官网地址:传送门
在下载文件之前我们需要配置一下,创建一个下载器,将下载器交给Spring管理:
DownloadConfig.java:
package com.butterflytri.config;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import red.wjf.download.downloader.ImageDownloader;
import red.wjf.download.downloader.WebDownloader;
/**
* @author: WJF
* @date: 2020/7/29
* @description: DownloadConfig
*/
@Configuration
public class DownloadConfig {
@Bean
public WebDownloader webDownloader() {
return new WebDownloader();
}
@Bean
public ImageDownloader imageDownloader() {
return new ImageDownloader();
}
}
在这里我们创建了两个下载器,都交给了Spring框架管理。
@RequestMapping("/downloadTest")
public void downloadTest(HttpServletRequest request, HttpServletResponse response, String fileName) {
InputStream is = this.getClass().getClassLoader().getResourceAsStream("test.docx");
try {
DataWriter writer = new DataWriter(is);
webDownloader.defaultDownload(request, response, fileName, writer);
} catch (IOException e) {
e.printStackTrace();
}
}
@RequestMapping("/downloadImg")
public void downloadImg(HttpServletRequest request, HttpServletResponse response) {
InputStream is = this.getClass().getClassLoader().getResourceAsStream("img.jpg");
ImageWriter writer = new ImageWriter(is, ImgSuffix.JPG);
imageDownloader.defaultDownload(request, response, writer);
}
这两个文件(test.docx和img.jpg)都在类路径下,从浏览器页面调用方法,就会下载,不会乱码也不会下载失败,除非文件不存在。jar包还提供了图片的Base64格式下载。在国内我们只需要调用这个默认的下载方法就可以了。下载的演示我就不演示了,大家可以拉下我的代码看看,觉得我写的东西对你有用,也可以给我点个赞哦。