当前位置：首页 > news >正文

target目录的文件上传controller，无需做静态映射。以及上传路径配置

news 来源：原创 2024/7/3 13:51:47

target目录的文件上传controller

为什么说target？因为直接上传到target无需做静态资源映射，即可直接访问。而那些上传到别的目录下的，均需要做映射才能访问

import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.bind.annotation.RestController;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.util.UUID;@RestController
public class FileController {// 文件上传接口@RequestMapping("/upFile")public String upFile(@RequestParam("file") MultipartFile file, HttpServletRequest request) throws IOException {// 设置文件保存路径String path = request.getServletContext().getRealPath("/upload");System.out.println("path:" + path);File realPath = new File(path);if (!realPath.exists()) {realPath.mkdir();}// 生成UUID作为文件名String uuid = UUID.randomUUID().toString();String originalFilename = file.getOriginalFilename();assert originalFilename != null;String newFilename = uuid + "_" + originalFilename;System.out.println("上传的文件地址：" + realPath);file.transferTo(new File(realPath + "/" + newFilename));return "localhost:8080/upload/" + newFilename;}// 文件下载接口@GetMapping("/download/{filename:.+}")public void downloadFile(@PathVariable("filename") String filename, HttpServletRequest request, HttpServletResponse response) throws IOException {System.out.println("Requested filename: " + filename);String path = request.getServletContext().getRealPath("/upload");File file = new File(path + "/" + filename);if (file.exists()) {response.setContentType("application/octet-stream");String encodedFilename = URLEncoder.encode(filename, "UTF-8").replaceAll("\\+", "%20");response.addHeader("Content-Disposition", "attachment; filename*=UTF-8''" + encodedFilename);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;while ((i = bis.read(buffer)) != -1) {os.write(buffer, 0, i);}} catch (Exception e) {e.printStackTrace();} finally {if (bis != null) {bis.close();}if (fis != null) {fis.close();}}} else {response.sendError(HttpServletResponse.SC_NOT_FOUND, "File not found");}}
}