SpringBoot实现文件的上传与下载

配置文件

# 文件保存路径，结尾必须有，不然目录不存在时生成的文件目录少一级
file.profile=F:/ruoyi/uploadPath/

上传单个文件

// 从配置文件获取要上传的目标文件路径
@Value("${file.profile}")
private String filePath;
​
private static final Logger log = LoggerFactory.getLogger(FileController.class);
​
// 单个文件上传
@RequestMapping(value = "/upload")
public String upload(@RequestParam("file") MultipartFile file) {
    try {
        if (file.isEmpty()) {
            return "上传失败，文件为空";
        }
        // 获取文件名
        String fileName = file.getOriginalFilename();
        log.info("上传的文件名为：" + fileName);//写日志
        // 获取文件的后缀名
        String suffixName = fileName.substring(fileName.lastIndexOf("."));
        log.info("文件的后缀名为：" + suffixName);//写日志
        // 设置文件存储路径         *************************************************
        String path = filePath + fileName;
        File dist = new File(new File(path).getAbsolutePath());// dist为文件，有多级目录的文件
        // 检测是否存在目录
        if (!dist.getParentFile().exists()) {//因此这里使用.getParentFile()，目的就是取文件前面目录的路径
            dist.getParentFile().mkdirs();// 新建文件夹
        }
        file.transferTo(dist);// 文件写入
        return "上传成功";
    } catch (IllegalStateException | IOException e) {
        e.printStackTrace();
    }
    return "上传失败";
}

上传多个文件

@PostMapping("/batch")
public String handleFileUpload(HttpServletRequest request) {
    // 获取多个上传文件的列表
    List<MultipartFile> files = ((MultipartHttpServletRequest) request).getFiles("file");
    MultipartFile file = null;
    BufferedOutputStream stream = null;
    for (int i = 0; i < files.size(); ++i) {
        file = files.get(i);
        File dest = new File(filePath);
        // 检测是否存在目录
        if (!dest.exists()) {
            dest.mkdirs();// 新建文件夹
        }
        if (!file.isEmpty()) {
            try {
                byte[] bytes = file.getBytes();
                stream = new BufferedOutputStream(new FileOutputStream(
                    filePath + file.getOriginalFilename()));//设置文件路径及名字
                stream.write(bytes);// 写入
                stream.close();
            } catch (Exception e) {
                stream = null;
                return "第 " + i + " 个文件上传失败 ==> " + e.getMessage();
            }
        } else {
            return "第 " + i + " 个文件上传失败因为文件为空";
        }
    }
    return "上传成功";
}

文件下载

只能是单个文件下载

@GetMapping("/download")
public String downloadFile(HttpServletRequest request, HttpServletResponse response) {
    String fileName = "dump.rdb";// 要被下载的文件名
    //设置文件路径
    File file = new File(filePath + 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);
            }
            return "下载成功";
        } 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 "下载失败";
}