SpringBoot学习：文件上传和下载

maven导入依赖

首先创建一个maven项目，然后加入以下配置，就创建好了一个springboot项目

　　<parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.1.5.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
    </dependencies>

项目

主要是实现文件的上传和下载，所以项目结构很简单

先看一下配置文件，SpringBoot提供了文件解析类，我们需要去配置一些参数

主要是限制上传文件的大小

然后是上传文件的工具类，这里是自己写的文件操作，（使用commons-fileupload和commons-io这些写好的工具更好）

package com.david.learn.utlis;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;

public class FileUtils {

    /**
     *
     * @param input 输入的文件流
     * @param filePath  文件要保存的路径
     * @throws IOException
     */
  public static void upload(InputStream input, File filePath) throws IOException {
      FileOutputStream output = new FileOutputStream(filePath);

      byte[] bytes = new byte[4096];
      int n ;
      while (-1  != (n = input.read(bytes)) ) {
         output.write(bytes,0,n);
      }
  }



    /**
     * 将文件下载到客户端
     *
     * @param filePath
     * @param fileName
     * @param response
     * @param isOnLine 是否在线打开
     * @throws Exception
     */
    public static void download(String filePath, String fileName, HttpServletRequest request,
                                HttpServletResponse response, boolean isOnLine) throws Exception {
        File f = new File(filePath);

        BufferedInputStream br = new BufferedInputStream(new FileInputStream(f));
        byte[] buf = new byte[1024];
        int len = 0;

        response.reset(); // 非常重要
        // 解决文件名中文乱码问题
        String fileNameDisplay = URLEncoder.encode(fileName, "UTF-8");
        // 针对火狐浏览器需要特殊处理
        if ("FF".equals(getBrowser(request))) {
            fileNameDisplay = new String(fileName.getBytes("UTF-8"), "iso-8859-1");
        }
        // 在线打开方式
        if (isOnLine) {
            response.setHeader("Content-Disposition", "inline; filename=" + fileNameDisplay);
        }
        // 纯下载方式
        else {
            response.setContentType("application/x-msdownload");
            response.setHeader("Content-Disposition", "attachment; filename=" + fileNameDisplay);
        }

        OutputStream out = response.getOutputStream();
        while ((len = br.read(buf)) > 0) {
            out.write(buf, 0, len);
        }
        out.flush();
        br.close();
        out.close();
    }

    /**
     * 判断客户端浏览器类型
     *
     * @param request
     * @return
     */
    private static String getBrowser(HttpServletRequest request) {
        String UserAgent = request.getHeader("USER-AGENT").toLowerCase();
        if (UserAgent != null) {
            if (UserAgent.indexOf("msie") >= 0) {
                return "IE";
            }
            if (UserAgent.indexOf("firefox") >= 0) {
                return "FF";
            }
            if (UserAgent.indexOf("safari") >= 0) {
                return "SF";
            }
        }
        return null;
    }
}

在写controller代码之前，还需要写一个接口调用后返回的响应类

package com.david.learn.dto;

import lombok.AllArgsConstructor;
import lombok.Data;

@Data
@AllArgsConstructor
public class BaseReturnDto {
    // 返回码
    private int code;
    // 结果说明
    private String msg;
}

最后是controller的代码，总共写了三个接口，分别是上传文件、下载文件、在线查看文件

查看文件还得看浏览器是否支持，一般查看pdf文件是可以的

package com.david.learn.controller;

import com.david.learn.dto.BaseReturnDto;
import com.david.learn.utlis.FileUtils;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;

@RestController
public class FileOperateController {
    // 当前项目绝对路径
    private static final String PROJECT_PATH = new File("").getAbsolutePath();
    // 文件上传的路径
    private static final String UPLOAD_PATH = PROJECT_PATH + "/upload/";

    // 最好手动在项目里创建文件夹，如果不想手动创建，加上这段代码
    static {
        // 项目里没有upload这个文件夹时，会自动创建
        File file = new File(UPLOAD_PATH);
        if (!file.exists()){
            if (!file.mkdirs()) {
                throw new RuntimeException("创建文件夹失败");
            }
        }
    }

    /**
     * 文件上传
     */
    @PostMapping("/uploadFile")
    // 使用数组可以实现多文件上传，单文件的话不用数组
    public BaseReturnDto uploadFile(MultipartFile[] multipartFile) throws IOException {

        for (MultipartFile file : multipartFile) {
            String fileLocation = UPLOAD_PATH + file.getOriginalFilename();

            FileUtils.upload(file.getInputStream(), new File(fileLocation));
        }

        return new BaseReturnDto(200, "文件上传成功");
    }

    /**
     * 文件下载
     */
    @GetMapping("/downloadFile/fileName/{fileName}")
    public BaseReturnDto download(HttpServletRequest request, HttpServletResponse response, @PathVariable String fileName) {

        try {
            FileUtils.download(UPLOAD_PATH + fileName, fileName, request, response, false);
            return new BaseReturnDto(200, "文件下载成功");
        } catch (Exception e) {
            e.printStackTrace();
            return new BaseReturnDto(500, "文件下载失败");
        }
    }

    /**
     * 在线查看文件
     */
    @GetMapping("/watchFile/fileName/{fileName}")
    public BaseReturnDto watch(HttpServletRequest request, HttpServletResponse response, @PathVariable String fileName) {

        try {
            FileUtils.download(UPLOAD_PATH + fileName, fileName, request, response, true);
            return new BaseReturnDto(200, "文件查看成功");
        } catch (Exception e) {
            e.printStackTrace();
            return new BaseReturnDto(500, "文件查看失败");
        }
    }

}