一、文件上传
@ApiOperation(value = "文件上传", notes= "文件上传")
@PostMapping("/upload")
public Result upLoadFile(@ApiParam(value = "选择需要上传的文件",required = true) MultipartFile file){
if (file.isEmpty()){
return Result.error(CodeMsg.ERROR);
}
String fileName = file.getOriginalFilename();
long size = file.getSize();
System.out.println(fileName+"-->"+size);
int index = fileName.indexOf(".");
String suffix = fileName.substring(index);
String name = Date.valueOf(LocalDate.now())+suffix;
File dest = new File(location, name);
if(!dest.getParentFile().exists()){
dest.getParentFile().mkdir();
}
try {
file.transferTo(dest);
return Result.success();
} catch (IOException e) {
e.printStackTrace();
return Result.error(CodeMsg.ERROR);
}
}
二、文件下载
@ApiOperation(value = "文件下载", notes = "文件下载")
@GetMapping("/download")
public Result downLoadFile(@ApiParam(value = "填写需要下载的文件") String fileName, HttpServletResponse response){
File file = new File(location, fileName);
if(file.exists()){
response.setContentType("application/force-download");
response.addHeader("Content-Disposition", "attachment;fileName=" + fileName);
FileInputStream input = null;
try{
input = new FileInputStream(file);
ServletOutputStream out = response.getOutputStream();
int real;
byte[] bytes = new byte[1024];
while((real=input.read(bytes))!=-1){
out.write(bytes, 0, real);
}
return Result.success();
}catch (Exception e){
e.printStackTrace();
return Result.error(CodeMsg.ERROR);
}
}
return Result.error(CodeMsg.ERROR);
}
三、配置文件
spring:
servlet:
multipart:
max-file-size: 100MB
max-request-size: 100MB