RestTemplate上傳檔案 - tw511教學網

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形式獲取到，然後向企業微信伺服器端上傳該輸入流即可。

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;
}

在需要輸入流進行上傳檔案時，需要使用InputStreamResource構建資原始檔，注意要重寫contentLength() 和 getFilename()方法，否則不成功。至於引數中的fileName和contentLength要提前通過檔案的URL獲取到，因為我是從阿里雲OSS讀取的檔案，所以直接能夠獲取到檔案的大小和檔名。

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()方法，不然不成功。

以上就是常用的RestTemplate上傳不同形式檔案的方式。有用點贊~感謝~