文件的上传与下载 （springboot---终极版）

上传与下载

package com.zkw.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.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.zkw.Mapper.FileMapper;         //mapper文件
import com.zkw.entity.FileEntity;        //实体类
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

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

@RestController
@RequestMapping("/file")
public class TestFileController {

    //引入yml配置文件中的配置
    @Value("${file.upload.path}")
    private String fileUploadPath;

    @Resource
    private FileMapper fileMapper;


    /**
     * 文件上传
     * @param file
     * @param request
     * @return
     */
    @PostMapping("/upload")
    public String upload (@RequestParam MultipartFile file, HttpServletRequest request){
        String filename = file.getOriginalFilename();
        String type = FileUtil.extName(filename);
        long size = file.getSize();

        //先存储到磁盘上
        File uploadFile = new File(fileUploadPath);

        //判断配置文件目录是否存在，不存在就创建
        if (!uploadFile.exists()){
            uploadFile.mkdirs();
        }
        //定义一个文件唯一的标识码
        String uuid = IdUtil.fastSimpleUUID();
        //在uuid后边加上文件的类型 StrUtil.DOT 表示的是一个 .
        String fileUUID = uuid + StrUtil.DOT + type;
        //上传的文件的最终路径
        File uploadFilePath = new File(fileUploadPath + fileUUID);

        try {
            //把获取到的文件存储到磁盘目录上
            file.transferTo(uploadFilePath);
        } catch (IOException e) {
            e.printStackTrace();
        }

        //对文件尽心md5
        String md5 = SecureUtil.md5(uploadFilePath);
        String url = null;

        //通过文件的md5查询 此时上传的文件是都存在
        QueryWrapper<FileEntity> wrapper = new QueryWrapper<>();
        wrapper.eq("md5",md5);
        List<FileEntity> list = fileMapper.selectList(wrapper);
        FileEntity fileEntity = list.size() == 0 ? null : list.get(0);

        //得到浏览器中的url
        StringBuffer oldUrl = request.getRequestURL();

        //把浏览器的url进行替换，得到图片地址
        StringBuffer newUrl = oldUrl.replace(oldUrl.indexOf("upload"),oldUrl.length(), fileUUID);

        //判断文件是否 是第一次上传
        if (fileEntity == null){
            url = newUrl + "";
        }else{
            //文件存在就删除
            uploadFilePath.delete();
            url = fileEntity.getUrl();
        }

        //存储数据库
        FileEntity saveFiles = new FileEntity();
        saveFiles.setName(filename);
        saveFiles.setType(type);
        saveFiles.setSize(size / 1024);
        saveFiles.setUrl(url);
        saveFiles.setMd5(md5);
        fileMapper.insert(saveFiles);
        return saveFiles.getUrl();
    }

    /**
     * 文件下载
     * @param fileUUID //下载时标识这个文件的参数
     * @param response
     */
    @GetMapping("/{fileUUID}")
    public void download(@PathVariable String fileUUID, HttpServletResponse response){
        //根据文件的唯一标识码获取文件
        File uploadFile = new File(fileUploadPath + fileUUID);

        //设置输出流的格式
        ServletOutputStream os = null;
        try {
            os = response.getOutputStream();
            response.addHeader("Content-Disposition","attachment;filename" + URLEncoder.encode(fileUUID, "UTF-8"));
            response.setContentType("application/octet-stream");

            //使用了hutool 工具类 里边的一个方法，字节流的方式读出文件
            os.write(FileUtil.readBytes(uploadFile));

        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            try {
                os.flush();
                os.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

    }

}

yaml 文件配置

spring:
  datasource:
    url: jdbc:mysql://localhost:3306/test?useSSL=true&useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC
    username: root
    password: 123456
    driver-class-name: com.mysql.cj.jdbc.Driver

file:
  upload:
    path: D:\files\ # 问价上传到磁盘上的位置

mybatis-plus 配置

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;


@Configuration
@MapperScan("com.zkw.mapper") // 扫描mapper文件
public class MybatisPlusConfig {

    /**
     * 新的分页插件,一缓和二缓遵循mybatis的规则,
     * 需要设置 MybatisConfiguration#useDeprecatedExecutor = false
     * 避免缓存出现问题(该属性会在旧插件移除后一同移除)
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
        return interceptor;
    }
}

FileMapper.java 的写法

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.zkw.entity.FileEntity;


public interface FileMapper extends BaseMapper<FileEntity> {
}

所使用到的依赖

 <!--mybatis依赖-->
<dependency>
    <groupId>org.mybatis.spring.boot</groupId>
    <artifactId>mybatis-spring-boot-starter</artifactId>
    <version>2.3.1</version>
</dependency>

<!--sql 驱动-->
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>8.0.22</version>
</dependency>

 
<!--hutool工具-->
<dependency>
   <groupId>cn.hutool</groupId>
   <artifactId>hutool-all</artifactId>
   <version>5.7.20</version>
</dependency>
<dependency>
   <groupId>org.apache.poi</groupId>
   <artifactId>poi-ooxml</artifactId>
   <version>4.1.2</version>
</dependency>


<!--mybatis-plus依赖-->
<dependency>
   <groupId>com.baomidou</groupId>
   <artifactId>mybatis-plus-boot-starter</artifactId>
   <version>3.5.1</version>
</dependency>

sql 文件

SET FOREIGN_KEY_CHECKS=0;

-- ----------------------------
-- Table structure for t_file
-- ----------------------------
DROP TABLE IF EXISTS `t_file`;
CREATE TABLE `t_file` (
  `id` int NOT NULL AUTO_INCREMENT,
  `name` varchar(255) DEFAULT NULL,
  `type` varchar(255) DEFAULT NULL,
  `size` bigint DEFAULT NULL,
  `url` varchar(255) DEFAULT NULL,
  `md5` varchar(255) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=17 DEFAULT CHARSET=utf8;