参考了其他博主的代码,留做收藏学习
@RestController
public class FileController {
private static final Logger log = LoggerFactory.getLogger(FileController.class);
// 单个文件上传
@RequestMapping(value = "/upload")
public String upload(@RequestParam("file") MultipartFile file) { //注意参数
try {
if (file.isEmpty()) {
return "文件为空";
}
// 获取文件名
String fileName = file.getOriginalFilename();
log.info("上传的文件名为:" + fileName);//写日志
// 获取文件的后缀名
String suffixName = fileName.substring(fileName.lastIndexOf("."));
log.info("文件的后缀名为:" + suffixName);//写日志
// 设置文件存储路径 *************************************************
String filePath = "./FILE/KING/";
String path = filePath + fileName;
File dest = new File(new File(path).getAbsolutePath());// dist为文件,有多级目录的文件
// 检测是否存在目录
if (!dest.getParentFile().exists()) {//因此这里使用.getParentFile(),目的就是取文件前面目录的路径
dest.getParentFile().mkdirs();// 新建文件夹
}
file.transferTo(dest);// 文件写入
return "上传成功";
} catch (IllegalStateException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return "上传失败";
}
// 多个文件一起上传
@PostMapping("/batch")
public String handleFileUpload(HttpServletRequest request) { //注意参数
List<MultipartFile> files = ((MultipartHttpServletRequest) request).getFiles("file");
MultipartFile file = null;
BufferedOutputStream stream = null;
for (int i = 0; i < files.size(); ++i) {
file = files.get(i);
String filePath = "./FILE/KING/"; // 文件路径
File dest = new File(filePath);
// 检测是否存在目录
if (!dest.exists()) {
dest.mkdirs();// 新建文件夹
}
if (!file.isEmpty()) {
try {
byte[] bytes = file.getBytes();
stream = new BufferedOutputStream(new FileOutputStream(
new File(filePath + file.getOriginalFilename())));//设置文件路径及名字
stream.write(bytes);// 写入
stream.close();
} catch (Exception e) {
stream = null;
return "第 " + i + " 个文件上传失败 ==> " + e.getMessage();
}
} else {
return "第 " + i + " 个文件上传失败因为文件为空";
}
}
return "上传成功";
}
@GetMapping("/download")
public String downloadFile(HttpServletRequest request, HttpServletResponse response) {
String fileName = "png.png";// 文件名
if (fileName != null) {
//设置文件路径
File file = new File("./FILE/KING/png.png");
//File file = new File(realPath , fileName);
if (file.exists()) {
response.setContentType("application/force-download");// 设置强制下载不打开
response.addHeader("Content-Disposition", "attachment;fileName=" + fileName);// 设置文件名
byte[] buffer = new byte[1024];
FileInputStream fis = null;
BufferedInputStream bis = null;
try {
fis = new FileInputStream(file);
bis = new BufferedInputStream(fis);
OutputStream os = response.getOutputStream();
int i = bis.read(buffer);
while (i != -1) {
os.write(buffer, 0, i);
i = bis.read(buffer);
}
return "下载成功";
} catch (Exception e) {
e.printStackTrace();
} finally { // 做关闭操作
if (bis != null) {
try {
bis.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (fis != null) {
try {
fis.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
}
return "下载失败";
}
}``