SpringBoot文件上传下载

文件上传

使用到hutool工具，需要导入pom依赖。

1、在yml中定义文件上传的位置

files:
  upload:
    path: E:/workspaces/workspace-Javacode/IDEA/idea/vuepro/files/

2、在controller层先拿到文件位置

    @Value(value = "${files.upload.path}")
    private String fileUploadPath;

3、在方法中通过@RequestParam，创建一个MultipartFile作为接收文件类型的参数

参考来源

 4、拿到文件的名称和类型等

因为不知道原来文件名称是否含有一些特殊字符，就自己再定义一个名字。这里用到hutool的IdUtil拿到一个uuid,再使用UUID + type类型拿到文件的标识码，即文件名称，使用Strutil.DOT作为名称和类型的分隔符‘.’号

        //获取原始名称
        String originalFilename = file.getOriginalFilename();
        //获得文件的类型
        String type = FileUtil.extName(originalFilename);
        long size = file.getSize();

        //定义文件唯一的标识码 StrUtil.DOT 点
        String uuid = IdUtil.fastSimpleUUID();
        String fileUuid =  uuid + StrUtil.DOT + type;

5、判断工程是否存在files目录，没有存在的话创建一个

        //判断配置的文件的父目录是否存在，不存在创建新的文件目录
        if(!uploadFile.getParentFile().exists()){
            uploadFile.getParentFile().mkdirs();
        }

6、在 E:/workspaces/workspace-Javacode/IDEA/idea/vuepro/files/目录下new一个文件，将前端上传的文件写入磁盘

              File uploadFile = new File(fileUploadPath  + fileUuid);

              file.transferTo(uploadFile);

7、因为可能会上传到相同的文件，所有先拿到每个文件的md5，再在数据库根据md5查找是否有相同的文件，如果有，就将写入磁盘的文件删除

        //文件存在
        if(dbFile !=null){
            url = dbFile.getUrl();
            //删除刚才上传已存在的文件
            uploadFile.delete();
        }else {
            //不存在重复文件，就不删除
            url = "http://localhost:8081/file/" + fileUuid;
        }

 8、再将文件名称类型等写入数据库中

即使存在相同的文件也要作为一条上传数据写入数据库。

下载

1、通过@PathVariable拿到文件唯一标识符，使用HttpServletResponse作为服务端响应。

HttpServletResponse这个对象中封装了向客户端发送数据、发送响应头，发送响应状态码的方法。getOutputStream方法用于返回Servlet引擎创建的字节输出流对象，Servlet程序可以按字节形式输出响应正文。

2、根据文件的唯一标识码获取文件

File uploadFile = new File(fileUploadPath + fileUuid);

3、设置文件输出流格式 写出流

ServletOutputStream os = response.getOutputStream();
response.addHeader("Content-Disposition","attachment;filename=" + URLEncoder.encode(fileUuid,"UTF-8"));
response.setContentType("application/octet-stream");

 4、读取文件的字节流再关闭流

os.write(FileUtil.readBytes(uploadFile));
os.flush();
os.close();

完整的代码如下

package com.aqya.controller;

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.SecureUtil;
import com.aqya.common.Result;
import com.aqya.entity.Files;
import com.aqya.service.IFilesService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.List;

/**  文件操作接口
 * @author 阿青呀
 * @date 2022/05/15 13:20
 **/
@RestController
@RequestMapping("/file")
public class FileController {

    @Value(value = "${files.upload.path}")
    private String fileUploadPath;


    @Autowired
    private IFilesService filesService;



    /**
     * 文件上传接口
     * @param file
     * @return  file 前端传过来的文件
     * @throws IOException
     */
    @PostMapping("/upload")
    public String upload(@RequestParam MultipartFile file) throws IOException {
        //获取原始名称
        String originalFilename = file.getOriginalFilename();
        //获得文件的类型
        String type = FileUtil.extName(originalFilename);
        long size = file.getSize();



        //定义文件唯一的标识码 StrUtil.DOT 点
        String uuid = IdUtil.fastSimpleUUID();
        String fileUuid =  uuid + StrUtil.DOT + type;
        File uploadFile = new File(fileUploadPath  + fileUuid);

        String md5 = null;
        String url = null;
        //判断配置的文件的父目录是否存在，不存在创建新的文件目录
        if(!uploadFile.getParentFile().exists()){
            uploadFile.getParentFile().mkdirs();
        }

        //把前端传过来的文件传到磁盘目录
        file.transferTo(uploadFile);
        //获取文件md5
        md5 = SecureUtil.md5(uploadFile);
        //查数据库   通过md5文件避免传入相同的文件
        Files dbFile = getFileByMd5(md5);


        //文件存在
        if(dbFile !=null){
            url = dbFile.getUrl();
            //删除刚才上传已存在的文件
            uploadFile.delete();
        }else {
            //不存在重复文件，就不删除
            url = "http://localhost:8081/file/" + fileUuid;
        }



        //存储数据库
        Files saveFile = new Files();
        saveFile.setName(originalFilename);
        saveFile.setSize(size/1024);
        saveFile.setType(type);
        saveFile.setUrl(url);
        saveFile.setMd5(md5);
        filesService.save(saveFile);

        return  url;
    }

    /**
     * 文件下载接口  http：//localhost:8081/file/{fileUuid}
     * @param fileUuid
     * @param response
     * @throws IOException
     */
    @GetMapping("/{fileUuid}")
    public void downLoad(@PathVariable String fileUuid, HttpServletResponse response) throws IOException {
        //根据文件的唯一标识码获取文件
        File uploadFile = new File(fileUploadPath + fileUuid);
        //设置文件输出流格式  写出流
        ServletOutputStream os = response.getOutputStream();
        response.addHeader("Content-Disposition","attachment;filename=" + URLEncoder.encode(fileUuid,"UTF-8"));
        response.setContentType("application/octet-stream");

        //读取文件的字节流
        os.write(FileUtil.readBytes(uploadFile));
        os.flush();
        os.close();
    }

    /**
     * 通过文件 md5查询
     * @param md5
     * @return
     */
    private Files getFileByMd5 (String md5){
        //查询文件的md5是否存在
        QueryWrapper<Files> wrapper = new QueryWrapper<>();
        wrapper.eq("md5",md5);
        List<Files> filesList = filesService.list(wrapper);
        return filesList.size() == 0 ? null  : filesList.get(0);
    }