需求背景:
公司有一个上传文件的内部接口,但是由于网络路线没有打通,导致外部不能直接访问这个接口,现在需要新增一个外部能访问的接口,然后通过这个外部接口代理到内部的文件上传接口。
功能实现
关于RestTemplate发送数据的案例有很多,但是这次需要使用RestTemplate转发MultipartFile文件,网上很多都说需要将要转发的文件保存到本地作为零时文件,然后通过**new FileSystemResource(localFile)**后来发送文件。但是我觉得应该有更好的方法,通过查看MultipartFile的源码,其中getResource()方法有这样一段话,才解决了我的疑惑。
default Resource getResource() {
return new MultipartFileResource(this);
}
MultipartFile官方提供了getResource()方法返回一个MultipartFile的资源形式,并且可以将返回都Resouce作为RestTemplate或者WebClient的输入。那么RestTemplate转发MultipartFile的实现就可以这样实现:
@RequestMapping(method = RequestMethod.POST, value = "/upload/logFiles", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
public String receiveFile(@RequestParam String vin,
@RequestParam(required = false, value = "Command_session_ID") Object cmdSessionIdStr,
@RequestParam String progress, @RequestParam String sig, @RequestParam String iv,
@RequestHeader("Authorization") String token,
HttpServletRequest request) {
MultiValueMap<String, MultipartFile> multiFileMap = ((MultipartHttpServletRequest) request).getMultiFileMap();
String first = (String) multiFileMap.keySet().iterator().next();
MultipartFile file = (MultipartFile) multiFileMap.getFirst(first);
HttpHeaders headers = new HttpHeaders();
headers.add("Authorization", token);
headers.setContentType(MediaType.MULTIPART_FORM_DATA);
LinkedMultiValueMap<String, Object> params = new LinkedMultiValueMap<>();
params.add("vin", vin);
params.add("Command_session_ID", cmdSessionIdStr);
params.add("progress", progress);
params.add("sig", sig);
params.add("iv", iv);
params.add("file", file.getResource());
HttpEntity<MultiValueMap<String, Object>> requestEntity = new HttpEntity<MultiValueMap<String, Object>>(params, headers);
ResponseEntity<String> responseEntity = restTemplate.exchange(logUpLoadUrl, HttpMethod.POST, requestEntity, String.class);
return responseEntity.getBody();
}
|