SpringBoot整合 文件上传下载功能

工程代码

package zhw.example.zhw.fileModule.fileController;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.UUID;
@Controller
@RequestMapping(value = "file")
public class fileController {
    private final Logger log = LoggerFactory.getLogger(this.getClass());

    @ResponseBody
    @RequestMapping(value = "upload")
    public void upload(@RequestParam("file")MultipartFile file){
        // 获取原始名字
        String fileName = file.getOriginalFilename();
        // 文件保存路径
        String filePath = "D:/upload/";
        // 文件重命名，防止重复
        fileName = filePath + UUID.randomUUID() + "_" + fileName;
        log.info("fileName:"+fileName);

        // 文件对象
        File dest = new File(fileName);
        // 判断路径是否存在，如果不存在则创建
        if(!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs();
        }
        try {
            // 保存到服务器中
            file.transferTo(dest);
            log.info("file save succ");
        } catch (Exception e) {
            e.printStackTrace();
            log.info("fileName save fail");
        }
    }

    @GetMapping(value = "download")
    public void download(@RequestParam("filename") String fileName, HttpServletResponse response) throws IOException {
        // 文件地址，真实环境是存放在数据库中的
        String filename = "D:\\upload\\" + fileName;
        File file = new File(filename);
        // 穿件输入对象
        FileInputStream fis = new FileInputStream(file);
        // 设置相关格式
        response.setContentType("application/force-download");
        // 设置下载后的文件名以及header
        response.addHeader("Content-disposition", "attachment;fileName=" + filename);
        // 创建输出对象
        OutputStream os = response.getOutputStream();
        // 常规操作
        byte[] buf = new byte[1024];
        int len = 0;
        while((len = fis.read(buf)) != -1) {
            os.write(buf, 0, len);
        }
        fis.close();
    }
}

测试

上传测试

下载测试