RestTemplate上傳文件
- 2022 年 8 月 5 日
- 筆記
- JAVA, RestTemplate, RestTemplate上傳, RestTemplate上傳InputStream, RestTemplate上傳MultipartFile, RestTemplate上傳文件
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上傳不同形式文件的方式。有用點贊~感謝~
