文件上传接口详解

SpringBoot文件上传接口分享

在Controller层实现文件上传

FileController

/**
 * 先写上传接口在写下载接口
 * 涉及的知识点有IO流的简单应用
 * 首先获取文件的根路径，然后获取文件的唯一标识码（使用hutool工具中的IdUtils）,然后获取文件的文件名，最后获取文件的文件类型。
 */

/**
 * 上传接口
 */
package com.example.stu.controller;

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.List;

@RestController
@RequestMapping("/file")
public class FileController {
    @PostMapping("/upload")
    public String upload(MultipartFile file) {
        //首先要定义好上传的路劲
        //文件的根路径
        String filepath = System.getProperty("user.dir") + "/src/main/resource/static/file/";
        //获取文件的文件名
        String filename = file.getOriginalFilename();
        //获取文件的唯一标识
        String flag = IdUtil.fastSimpleUUID();
        int num = filename.lastIndexOf(".");
        String type = filename.substring(num);
        String files = filepath + flag + filename + "." + type;
        try {
            FileUtil.writeBytes(file.getBytes(), files);//相当于InputStream输入流
            System.out.println("------上传成功");
            Thread.sleep(1L);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return files;
    }

    /**
     * 下载接口
     */
    @GetMapping("/{flag}")
    public void download(String flag, HttpServletResponse response) throws Exception {
        //获取文件的根路径
        OutputStream os;
        String basepath = System.getProperty("user.dir") + "/src/main/resource/static/file/";
        List<String> fileNames = FileUtil.listFileNames(basepath);
        String avatar = fileNames.stream().filter(name -> name.contains(flag)).findAny().orElse("");
        try {
            if (StrUtil.isNotEmpty(avatar)) {
                response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(avatar, "UTF-8"));
                response.setContentType("application/octet-stream");
                byte[] bytes = FileUtil.readBytes(avatar);//将我的文件读写成字节流
                os=response.getOutputStream();
                os.write(bytes);//将字节流写入输出流
                os.flush();
                os.close();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}