SpringBoot文件上传接口分享
在Controller层实现文件上传
FileController
package com.example.stu.controller;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.List;
@RestController
@RequestMapping("/file")
public class FileController {
@PostMapping("/upload")
public String upload(MultipartFile file) {
String filepath = System.getProperty("user.dir") + "/src/main/resource/static/file/";
String filename = file.getOriginalFilename();
String flag = IdUtil.fastSimpleUUID();
int num = filename.lastIndexOf(".");
String type = filename.substring(num);
String files = filepath + flag + filename + "." + type;
try {
FileUtil.writeBytes(file.getBytes(), files);
System.out.println("------上传成功");
Thread.sleep(1L);
} catch (Exception e) {
e.printStackTrace();
}
return files;
}
@GetMapping("/{flag}")
public void download(String flag, HttpServletResponse response) throws Exception {
OutputStream os;
String basepath = System.getProperty("user.dir") + "/src/main/resource/static/file/";
List<String> fileNames = FileUtil.listFileNames(basepath);
String avatar = fileNames.stream().filter(name -> name.contains(flag)).findAny().orElse("");
try {
if (StrUtil.isNotEmpty(avatar)) {
response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(avatar, "UTF-8"));
response.setContentType("application/octet-stream");
byte[] bytes = FileUtil.readBytes(avatar);
os=response.getOutputStream();
os.write(bytes);
os.flush();
os.close();
}
} catch (Exception e) {
e.printStackTrace();
}
}
}