rest方式处理文件服务

小灰灰 2022-03-20 12:30 298阅读 0赞

### 文章目录 ###

*  文件上传
 *   *  创建FileInfo返回对象
     *  创建文件上传controller
     *  测试
 *  下载
 *   *  加入处理流的依赖
     *  下载方法 get请求
     *  测试

# 文件上传 #

## 创建FileInfo返回对象 ##

![在这里插入图片描述][20190212225249525.png]

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

## 创建文件上传controller ##

![在这里插入图片描述][20190212225353862.png]

@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());
    
            String folder = "D:\\workspace\\whale-security\\whale-security-demo\\src\\main\\java\\com\\whale\\web\\controller";
            File localFile = new File(folder,new Date().getTime()+".txt");
    
            //file.getInputStream()  也可以获取到文件流用阿里so上传
    
            file.transferTo(localFile);//写到本地文件
    
            return  new FileInfo(localFile.getAbsolutePath());
        }
    }

## 测试 ##

@Test
        public void  whenUploadSuccess() throws Exception {
            String result = mockMvc.perform(MockMvcRequestBuilders.fileUpload("/file")  //是post请求
                    .file(new MockMultipartFile("file","test.txt","multipart/form-data","hello upload".getBytes("utf-8"))))
                    .andExpect(MockMvcResultMatchers.status().isOk())
                    .andReturn().getResponse().getContentAsString();
            System.out.println(result);
        }

运行控制台打印：

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3VvdGFpbA_size_16_color_FFFFFF_t_70]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3VvdGFpbA_size_16_color_FFFFFF_t_70 1]

# 下载 #

## 加入处理流的依赖 ##

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

## 下载方法 get请求 ##

![在这里插入图片描述][20190212231118669.png]

@GetMapping("{id}")
        public void download(@PathVariable String id, HttpServletRequest request, HttpServletResponse response){
            //jdk1.7语法 将流声明在try() 可以自动关闭
            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);//输入流copy到输出流
    
                outputStream.flush();
    
    
            }catch (Exception e){
                e.printStackTrace();
            }
        }

## 测试 ##

访问 [http://127.0.0.1:8080/file/1549983068479][http_127.0.0.1_8080_file_1549983068479]  
![在这里插入图片描述][20190212231311139.png]

[20190212225249525.png]: /images/20220320/708ef9be58c040d3af2f5bab7a89e4f8.png
[20190212225353862.png]: /images/20220320/d8e221b045884777a3f5b432b31d0092.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3VvdGFpbA_size_16_color_FFFFFF_t_70]: /images/20220320/ef67afb5b4bf448a808bf70870d91e60.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3VvdGFpbA_size_16_color_FFFFFF_t_70 1]: /images/20220320/1ac210a62bcf449285c514d50aeeb27b.png
[20190212231118669.png]: /images/20220320/83b9d9fb7b1242ae9b10b133ae537c05.png
[http_127.0.0.1_8080_file_1549983068479]: http://127.0.0.1:8080/file/1549983068479
[20190212231311139.png]: /images/20220320/99c1eab629df4a1e916da213229e1fa7.png