通过RestTemplate上传文件

落日映苍穹つ 2024-03-27 13:36 85阅读 0赞

### 二、客户端 ###

### 1、上传的文件是File类型 ###

> 如果文件保存在本地，即可以通过File file = new File(path) 或者 文件路径地址获取到指定文件

public String uploadFile(File file) {
        // 1、封装请求头
        HttpHeaders headers = new HttpHeaders();
        MediaType type = MediaType.parseMediaType("multipart/form-data");
        headers.setContentType(type);
        headers.setContentLength(file.length());
        headers.setContentDispositionFormData("media", file.getName());
        // 2、封装请求体
        MultiValueMap<String, Object> param = new LinkedMultiValueMap<>();
        FileSystemResource resource = new FileSystemResource(file);
        param.add("file", resource);
        // 3、封装整个请求报文
        HttpEntity<MultiValueMap<String, Object>> formEntity = new HttpEntity<>(param, headers);
        // 4、发送请求
        ResponseEntity<String> data = restTemplate.postForEntity(tempMaterialUploadUrl, formEntity, String.class);
        // 5、请求结果处理
        JSONObject weChatResult = JSONObject.parseObject(data.getBody());
        return weChatResult;
    }

这种方式可直接将File文件或者文件路径传递给FileSystemResource资源对象。然后将该资源放入请求体中。

### 2、上传的文件是InputStream流 ###

> 如果文件不存在本地，而是在阿里云OSS或者其他只能够通过URL获取文件流，并且不想将文件存到本地而直接通过restTemplate发送，则可以使用下面方式。
> 
> 例子：在工作中用到这个例子：我将系统上传的附件通过阿里云API上传至阿里云OSS，紧接着要讲文件存到企业微信服务器，而且系统服务器不存储任何附件，就用到了下面的方式，通过阿里云API将上传的附件A通过InputStream形式获取到，然后向企业微信服务端上传该输入流即可。

在需要输入流进行上传文件时，需要使用**InputStreamResource**构建资源文件，注意要重写**contentLength(**) 和 **getFilename()**方法，否则不成功。至于参数中的**fileName**和**contentLength**要提前通过文件的URL获取到，因为我是从阿里云OSS读取的文件，所以直接能够获取到文件的大小和文件名。

public String uploadInputStream(InputStream inputStream,String fileName,long cententLength) {
        // 1、封装请求头
        HttpHeaders headers = new HttpHeaders();
        MediaType type = MediaType.parseMediaType("multipart/form-data");
        headers.setContentType(type);
        headers.setContentDispositionFormData("media", fileName);
        // 2、封装请求体
        MultiValueMap<String, Object> param = new LinkedMultiValueMap<>();
        InputStreamResource resource = new InputStreamResource(inputStream){
            @Override
            public long contentLength(){
                return cententLength;
            }
            @Override
            public String getFilename(){
                return fileName;
            }
        };
        param.add("file", resource);
        // 3、封装整个请求报文
        HttpEntity<MultiValueMap<String, Object>> formEntity = new HttpEntity<>(param, headers);
        // 4、发送请求
        ResponseEntity<String> data = restTemplate.postForEntity(tempMaterialUploadUrl, formEntity, String.class);
        // 5、请求结果处理
        JSONObject weChatResult = JSONObject.parseObject(data.getBody());
        // 6、返回结果
        return weChatResult;
    }

### 3、上传的是MultipartFile类型文件 ###

> 有时候我们需要直接将系统上传SpringMVC通过MultipartFile类型接收的文件，此时就可使用下面方式解决。

public String uploadFileWithInputStream(MultipartFile file) throws IOException {
        // 1、封装请求头
        HttpHeaders headers = new HttpHeaders();
        MediaType type = MediaType.parseMediaType("multipart/form-data");
        headers.setContentType(type);
        headers.setContentLength(file.getSize());
        headers.setContentDispositionFormData("media", file.getOriginalFilename());
        // 2、封装请求体
        MultiValueMap<String, Object> param = new LinkedMultiValueMap<>();
        // 将multipartFile转换成byte资源进行传输
        ByteArrayResource resource = new ByteArrayResource(file.getBytes()) {
            @Override
            public String getFilename() {
                return file.getOriginalFilename();
            }
        };
        param.add("file", resource);
        // 3、封装整个请求报文
        HttpEntity<MultiValueMap<String, Object>> formEntity = new HttpEntity<>(param, headers);
        // 4、发送请求
        ResponseEntity<String> data = restTemplate.postForEntity(tempMaterialUploadUrl, formEntity, String.class);
        // 5、请求结果处理
        JSONObject weChatResult = JSONObject.parseObject(data.getBody());
        // 6、返回结果
        return weChatResult;
    }

注意点：使用**ByteArrayResource**构建资源时，应重写**ByteArrayResource**的**getFilename()**方法，不然不成功。

###  ###

## 二、服务端 ##

@PostMapping("/CloudUploadFile")
        @ResponseBody
        public String UploadFile(@RequestParam("file") MultipartFile file,@RequestParam("module") String module){
            return managementService.ExecuteCloudUploadFile(file,module);
        }

三、参考

[RestTemplate发送带文件的请求\_君之虞的博客-CSDN博客\_resttemplate 发送文件][RestTemplate_-CSDN_resttemplate]

[通过RestTemplate上传文件 - 简书][RestTemplate_ -]

[RestTemplate_-CSDN_resttemplate]: https://blog.csdn.net/weixin_43851064/article/details/109670484?spm=1001.2101.3001.6661.1&utm_medium=distribute.pc_relevant_t0.none-task-blog-2~default~CTRLIST~Rate-1-109670484-blog-119602052.pc_relevant_aa&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-2~default~CTRLIST~Rate-1-109670484-blog-119602052.pc_relevant_aa&utm_relevant_index=1
[RestTemplate_ -]: http://events.jianshu.io/p/93616d280e29