看了很多的博主写的文件的上传下载,然后自己修修补补,写了一个
/**
* @author zyj
* @date 2021年04月日14:00
*/
@RestController
public class UpOrDownLoadController {
private static final Logger logger = LoggerFactory.getLogger(UpOrDownLoadController.class);
@RequestMapping("upload")
@ResponseBody
public String upload(@RequestParam("file") MultipartFile file, HttpServletRequest request) {
if (file.isEmpty()) {
return "文件为空";
}
// 获取文件名
String fileName = file.getOriginalFilename();
logger.info("上传的文件名为:" + fileName);
// 获取文件的后缀名
String suffixName = fileName.substring(fileName.lastIndexOf("."));
logger.info("上传的后缀名为:" + suffixName);
// 文件上传后的路径
String resource = this.getClass().getClassLoader().getResource("").getPath();
System.out.println(request);
File dest = new File(resource + fileName);
// 检测是否存在目录/Users/zhangyongjie/IdeaProjects/cat/src/main/resources/tmp
if (!dest.getParentFile().exists()) {
dest.getParentFile().mkdirs();
}
try {
file.transferTo(dest);
return "上传成功";
} catch (IllegalStateException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return "上传失败";
}
//文件下载相关代码
@RequestMapping("/download")
public File downloadFile(HttpServletRequest request, HttpServletResponse response) {
String fileName = "STRUCT_kkbdb.xls";
if (fileName != null) {
///Users/zhangyongjie/IdeaProjects/cat/target/classes/STRUCT_kkbdb.xls
String realPath = "/Users/zhangyongjie/IdeaProjects/cat/target/classes/";
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);
}
System.out.println("success");
} 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 file;
}
}
}
return null;
}
}