7、springBoot 简单优雅是实现文件上传和下载

前言

很久没有更新spring Boot 这个项目了。最近看了一下docker 的知识,后期打算将spring boot 和docker 结合起来。恰好最近有一个上传文件的工做呢,恰好就想起这个脚手架,将文件上传和下载整理进来。java

配置

在application.properties 中增长上传文件存放的路径配置linux

#文件上传目录
file.upload.url=E:/test

controller 层

上传文件和下载文件都比较简单,咱们就直接在controller层来编写。也不用在pom.xml 中增长什么依赖。因此直接上代码。 在controller 包下建立一个file包,在file 包下建立一个FileController 类。git

@RestController
@RequestMapping("file")
@Slf4j
public class FileController {
    @Value("${file.upload.url}")
    private String uploadFilePath;
		
		@RequestMapping("/upload")
    public String httpUpload(@RequestParam("files") MultipartFile files[]){
        JSONObject object=new JSONObject();
        for(int i=0;i<files.length;i++){
            String fileName = files[i].getOriginalFilename();  // 文件名
            File dest = new File(uploadFilePath +'/'+ fileName);
            if (!dest.getParentFile().exists()) {
                dest.getParentFile().mkdirs();
            }
            try {
                files[i].transferTo(dest);
            } catch (Exception e) {
                log.error("{}",e);
                object.put("success",2);
                object.put("result","程序错误,请从新上传");
                return object.toString();
            }
        }
        object.put("success",1);
        object.put("result","文件上传成功");
        return object.toString();
    }

}

上面的代码看起来有点多,其实就是一个上传的方法,首先经过 MultipartFile 接收文件。这里我用的是file[] 数组接收文件,这是为了兼容多文件上传的状况,若是只用file 接收,而后在接口上传多个文件的话,只会接收最后一个文件。这里你们注意一下。看本身的需求,我这里兼容多文件因此用数组接收。程序员

而后遍历files 获取文件,下面这段代码是判断文件在所在目录是否存在,若是不存在就建立对应的目录。github

File dest = new File(uploadFilePath +'/'+ fileName);
            if (!dest.getParentFile().exists()) {
                dest.getParentFile().mkdirs();
            }
files[i].transferTo(dest);

就是将文件存放到对应的服务器,这里有一点须要说明一下,若是咱们上传重复的文件会怎么样么?上传重复的文件不会报错,后上传的文件会直接覆盖已经上传的文件。spring

总体代码就是这样。如今就能够实现文件的上传操做。docker

测试

咱们写好以后,基本上传功能就已经实现了,咱们如今来测试一下。启动项目后咱们用postman 请求,由于咱们须要上传文件,用get 方式请求不了。 file file数组

能够看到文件上传成功了,因而可知,springboot文件上传一个方法就搞定了。springboot

文件下载

其实文件下载,不太建议用接口作,由于文件下载通常都是下载一些静态文件,咱们能够先将文件处理好,而后经过Nginx 服务下载静态文件,这样速度会快不少。可是这里咱们仍是写一下。代码也很简单,就一个方法,也写在fileController 类中服务器

@RequestMapping("/download")
    public String fileDownLoad(HttpServletResponse response, @RequestParam("fileName") String fileName){
        File file = new File(downloadFilePath +'/'+ fileName);
        if(!file.exists()){
            return "下载文件不存在";
        }
        response.reset();
        response.setContentType("application/octet-stream");
        response.setCharacterEncoding("utf-8");
        response.setContentLength((int) file.length());
        response.setHeader("Content-Disposition", "attachment;filename=" + fileName );

        try(BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));) {
            byte[] buff = new byte[1024];
            OutputStream os  = response.getOutputStream();
            int i = 0;
            while ((i = bis.read(buff)) != -1) {
                os.write(buff, 0, i);
                os.flush();
            }
        } catch (IOException e) {
            log.error("{}",e);
            return "下载失败";
        }
        return "下载成功";
    }

代码也很简单,就是根据文件名判断是否存在文件,不存在就提示没有文件,存在就将文件下载下来。response设置返回文件的格式,以文件流的方式返回,采用utf-8 字符集,设置下载后的文件名。而后就是以文件流的方式下载文件了。

测试的话也简单,咱们启动项目,访问接口

http://localhost:9090/zlflovemm/file/download?fileName=11
http://localhost:9090/zlflovemm/file/download?fileName=1.rar

file file 能够看到若是文件存在,会直接下载,不会提示下载成功或者失败。

删除文件

删除文件是很简单的,我这里讲一下删除文件下全部文件夹和文件。并作一个定时任务,天天清理一次。

@Scheduled(cron="0 0 3 * * ?")
   private void deleteFiles(){
       deleteFile(new File(deleteFilePath));
   }

   public void deleteFile(File file){
       //判断文件不为null或文件目录存在
       if (file == null || !file.exists()){
           log.info("暂无文件");
           return;
       }
       //取得这个目录下的全部子文件对象
       File[] files = file.listFiles();
       //遍历该目录下的文件对象
       for (File f: files){
           //打印文件名
           String name = f.getName();
           log.info(name);
           //判断子目录是否存在子目录,若是是文件则删除
           if (f.isDirectory()){
               deleteFile(f);
           }else {
               f.delete();
           }
       }
       //删除空文件夹  for循环已经把上一层节点的目录清空。
       file.delete();
   }

番外

到此为止,咱们经常使用的镜像和容器的操做就会使用啦。都是一些命令。忘记的能够--help 查看一下。

好了,就说这么多啦 代码上传到github: https://github.com/QuellanAn/zlflovemm

后续加油♡

欢迎你们关注我的公众号 "程序员爱酸奶"

分享各类学习资料,包含java,linux,大数据等。资料包含视频文档以及源码,同时分享本人及投递的优质技术博文。

若是你们喜欢记得关注和分享哟❤

file

相关文章
相关标签/搜索