文件的上传和下载
单文件上传
铭记:在使用Postman 进行接口测试时 , 不仅要选择File , 更要自己亲自写入file , 并且图片的内存也不能太大 , 这样才能测试成功
代码演示:
@AuthAccess
@PostMapping("/upload")
public Result upload(@RequestParam("file") MultipartFile file) throws IOException {
String originalFilename = file.getOriginalFilename(); // 文件的原始名称
// aaa.png
String mainName = FileUtil.mainName(originalFilename); // aaa
String extName = FileUtil.extName(originalFilename);// png
if (!FileUtil.exist(ROOT_PATH)) {
FileUtil.mkdir(ROOT_PATH); // 如果当前文件的父级目录不存在,就创建
}
if (FileUtil.exist(ROOT_PATH + File.separator + originalFilename)) { // 如果当前上传的文件已经存在了,那么这个时候我就要重名一个文件名称
originalFilename = System.currentTimeMillis() + "_" + mainName + "." + extName;
}
File saveFile = new File(ROOT_PATH + File.separator + originalFilename);
file.transferTo(saveFile); // 存储文件到本地的磁盘里面去
String url = "http://" + ip + ":" + port + "/file/download/" + originalFilename;
return Result.success(url); //返回文件的链接,这个链接就是文件的下载地址,这个下载地址就是我的后台提供出来的
}
文件下载
/**
* 文件下载
* @param response
* @throws IOException
*/
@AuthAccess
@GetMapping("/download/{fileName}")
public void download(@PathVariable String fileName, HttpServletResponse response) throws IOException {
// response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileFullName, "UTF-8")); // 附件下载
String filePath = ROOT_PATH + File.separator + fileName;
if (!FileUtil.exist(filePath)) {
return;
}
byte[] bytes = FileUtil.readBytes(filePath);
ServletOutputStream outputStream = response.getOutputStream();
outputStream.write(bytes); // 数组是一个字节数组,也就是文件的字节流数组
outputStream.flush();
outputStream.close();
}
全部代码演示
package com.north.controller;
import cn.hutool.core.io.FileUtil;
import com.north.common.AuthAccess;
import com.north.common.Result;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
/**
* @author North
* @Date 9/2/2023 10:25 PM
*/
@RestController
@RequestMapping("/file")
public class FileController {
@Value("${ip:localhost}")
String ip;
@Value("${server.port}")
String port;
private static final String ROOT_PATH = System.getProperty("user.dir") + File.separator + "files"; // D:\B站\小白做毕设2024\代码\小白做毕设2024\files
@AuthAccess
@PostMapping("/upload")
public Result upload(@RequestParam("file") MultipartFile file) throws IOException {
String originalFilename = file.getOriginalFilename(); // 文件的原始名称
// aaa.png
String mainName = FileUtil.mainName(originalFilename); // aaa
String extName = FileUtil.extName(originalFilename);// png
if (!FileUtil.exist(ROOT_PATH)) {
FileUtil.mkdir(ROOT_PATH); // 如果当前文件的父级目录不存在,就创建
}
if (FileUtil.exist(ROOT_PATH + File.separator + originalFilename)) { // 如果当前上传的文件已经存在了,那么这个时候我就要重名一个文件名称
originalFilename = System.currentTimeMillis() + "_" + mainName + "." + extName;
}
File saveFile = new File(ROOT_PATH + File.separator + originalFilename);
file.transferTo(saveFile); // 存储文件到本地的磁盘里面去
String url = "http://" + ip + ":" + port + "/file/download/" + originalFilename;
return Result.success(url); //返回文件的链接,这个链接就是文件的下载地址,这个下载地址就是我的后台提供出来的
}
/**
* 文件下载
* @param response
* @throws IOException
*/
@AuthAccess
@GetMapping("/download/{fileName}")
public void download(@PathVariable String fileName, HttpServletResponse response) throws IOException {
// response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileFullName, "UTF-8")); // 附件下载
String filePath = ROOT_PATH + File.separator + fileName;
if (!FileUtil.exist(filePath)) {
return;
}
byte[] bytes = FileUtil.readBytes(filePath);
ServletOutputStream outputStream = response.getOutputStream();
outputStream.write(bytes); // 数组是一个字节数组,也就是文件的字节流数组
outputStream.flush();
outputStream.close();
}
}