【Spring Security OAuth2笔记系列】- 【使用Spring MVC开发RESTful API】 rest服处理文件上传

rest服处理文件上传

现在的文件上传服务基本上都是先上传，后提交路径

测试用例

@Test
public void whenFileUploadSuccess() throws Exception {
    // v5.0+ fileUpLoad方法已经过时了
    String file = mockMvc.perform(multipart("/file")
                                          .file(new MockMultipartFile("file", "test.txt", "multipart/form-data", "hello upload".getBytes())))
            .andExpect(status().isOk())
            .andReturn().getResponse().getContentAsString();
    System.out.println(file);
}

输出
{"absolutePath":"G:\\dev\\project\\mrcode\\example\\imooc\\spring-security\\security-demo\\fileUploadTest.txt"}

文件上传服务

package com.example.demo.web.controller;

import com.example.demo.dto.FileInfo;
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 java.io.File;
import java.io.IOException;

@RestController
@RequestMapping("/file")
public class FileController {
    @PostMapping
    public FileInfo upload(MultipartFile file) throws IOException {
        System.out.println(file.getName());
        System.out.println(file.getOriginalFilename());
        System.out.println(file.getSize());

        File localFile = new File("fileUploadTest.txt");
        file.transferTo(localFile);
        return new FileInfo(localFile.getAbsolutePath());
    }
}

文件下载服务

下载服务好像在测试用例里面没有看到怎么写的；浏览器访问该路径，文件会被下载

@GetMapping("/{id}")
public void download(@PathVariable String id, HttpServletRequest request, HttpServletResponse response) {
    try (FileInputStream inputStream = new FileInputStream(new File("G:\\dev\\project\\mrcode\\example\\imooc\\spring-security\\security-demo", id));
         ServletOutputStream outputStream = response.getOutputStream();
    ) {
        // 声明响应类型
        response.setContentType("application/x-download");
        // 下载的文件名称
        response.addHeader("Content-Disposition", "attachment;filename-test.txt");
        IOUtils.copy(inputStream, outputStream);
    } catch (FileNotFoundException e) {
        e.printStackTrace();
    } catch (IOException e) {
        e.printStackTrace();
    }
}