009 spring boot中文件的上传与下载

一：任务

1.任务

　　文件的上传

　　文件的下载

二：文件的上传

1.新建一个对象

　　FileInfo.java

package com.cao.dto;

public class FileInfo {
    private String path;
    public FileInfo(String path) {
        this.path=path;
    }
    public String getPath() {
        return path;
    }
    public void setPath(String path) {
        this.path = path;
    }
    
}

2.新建控制器

package com.cao.web.controller;

import java.io.File;
import java.io.IOException;
import java.util.Date;

import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import com.cao.dto.FileInfo;

@RestController
@RequestMapping("/file")
public class FileController {
    @PostMapping
    public FileInfo uploadFile(MultipartFile fileKey) throws Exception {
        System.out.println("fileName: "+fileKey.getName());
        System.out.println("OriginalFilename: "+fileKey.getOriginalFilename());
        System.out.println("size: "+fileKey.getSize());
        //将要存储在controller的目录下
        String folder="E:\workspace-sts-3.9.5.RELEASE\it-security-demo\src\main\java\com\cao\web\controller";
        File localFile=new File(folder,new Date().getTime()+".txt");
        fileKey.transferTo(localFile);
        //写入流中，这个主要用于写入其他的地方，例如服务器等，这里不写了
        //fileKey.getInputStream()
        return new FileInfo(localFile.getAbsolutePath());
    }
}

3.测试类

/**
     * 测试文件的上传
     * @throws Exception 
     */
    @Test
    public void whenUploadSuccess() throws Exception {
        String result=mockMvc.perform(MockMvcRequestBuilders.fileUpload("/file")
                .file(new MockMultipartFile("fileKey","test.txt","multipart/form-data","hello".getBytes("UTF-8"))))
                .andExpect(MockMvcResultMatchers.status().isOk())
                .andReturn().getResponse().getContentAsString();
        System.out.println("result="+result);
    }

4.控制台

　　

　　存储到的现象

　　

三：文件的下载

1.添加io操作的包

          <dependency>
            <groupId>commons-io</groupId>
            <artifactId>commons-io</artifactId>
          </dependency>

2.文件下载的程序

package com.cao.web.controller;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Date;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;
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.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import com.cao.dto.FileInfo;

@RestController
@RequestMapping("/file")
public class FileController {
    
    String folder="E:\workspace-sts-3.9.5.RELEASE\it-security-demo\src\main\java\com\cao\web\controller";

    /**
     * 文件的上传控制器
     * @param fileKey
     * @return
     * @throws Exception
     */
    @PostMapping
    public FileInfo uploadFile(MultipartFile fileKey) throws Exception {
        System.out.println("fileName: "+fileKey.getName());
        System.out.println("OriginalFilename: "+fileKey.getOriginalFilename());
        System.out.println("size: "+fileKey.getSize());
        //将要存储在controller的目录下
        File localFile=new File(folder,new Date().getTime()+".txt");
        fileKey.transferTo(localFile);
        //写入流中，这个主要用于写入其他的地方，例如服务器等，这里不写了
        //fileKey.getInputStream()
        return new FileInfo(localFile.getAbsolutePath());
    }
    
    /**
     * 文件的下载
     */
    @GetMapping("/{id}")
    public void downFile(@PathVariable String id, HttpServletRequest request,HttpServletResponse response) {
        try(
            InputStream inputStream=new FileInputStream(new File(folder,id+".txt"));
            OutputStream outputStream=response.getOutputStream();    
            ) 
        {
            response.setContentType("application/x-download");
            response.addHeader("Content-Disposition", "attachment;filename=test.txt");
            IOUtils.copy(inputStream, outputStream);
            outputStream.flush();
        } catch (Exception e) {
            // TODO: handle exception
        }
    }
    
    
}

3.在浏览器上访问