RestTemplate 调用远程接口上传文件
问题描述
第三方写了一个文件上传的接口,该接口的请求方式为Post请求,请求参数全部是以form-data表单形式进行提交,包含三个参数
- 第一个:cookie(字符串类型)
- 第二个:seqNo(字符串类型)
- 第三个:file(文件类型)
解决方法
使用传统的Spring Cloud的Feign组件在调用远程接口实现文件上传时有时会出现异常错误,可考虑使用下述两种方式文件上传
第一种方式
使用RestTemplate进行调用
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
import org.springframework.core.io.InputStreamResource; import java.io.InputStream; public class CommonInputStreamResource extends InputStreamResource { private long length; private String fileName; public CommonInputStreamResource(InputStream inputStream, long length, String fileName) { super (inputStream); this .length = length; this .fileName = fileName; } /** * 覆写父类方法 * 如果不重写这个方法,并且文件有一定大小,那么服务端会出现异常 * {@code The multi-part request contained parameter data (excluding uploaded files) that exceeded} */ @Override public String getFilename() { return fileName; } /** * 覆写父类 contentLength 方法 * 因为 {@link org.springframework.core.io.AbstractResource#contentLength()}方法会重新读取一遍文件, * 而上传文件时,restTemplate 会通过这个方法获取大小。然后当真正需要读取内容的时候,发现已经读完,会报如下错误。 */ @Override public long contentLength() { long estimate = length; return estimate == 0 ? 1 : estimate; } public void setLength( long length) { this .length = length; } public void setFileName(String fileName) { this .fileName = fileName; } } |
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
try { String applySeqNo = "123456" ; String cookie= "654321" ; File file= new File( "E:\\1.rar" ); FileInputStream fileInputStream= new FileInputStream(file); //请求头设置为MediaType.MULTIPART_FORM_DATA类型 HttpHeaders requestHeaders = new HttpHeaders(); requestHeaders.setContentType(MediaType.MULTIPART_FORM_DATA); //构建请求体 MultiValueMap<String, Object> requestBody = new LinkedMultiValueMap<>(); CommonInputStreamResource commonInputStreamResource = null ; try { commonInputStreamResource = new CommonInputStreamResource(fileInputStream,file.length(),file.getName()); } catch (Exception e) { log.error( "文件输入流转换错误" ,e); } requestBody.add( "cookie" , cookie); requestBody.add( "seqNoFile" , applySeqNo); requestBody.add( "file" ,commonInputStreamResource); HttpEntity<MultiValueMap> requestEntity = new HttpEntity<MultiValueMap>(requestBody, requestHeaders); //直接调用远程接口 ResponseEntity<String> responseEntity = restTemplate.postForEntity( "http://xxx.xxx.xxx.xxx:8080/test/upload" ,requestEntity, String. class ); System.out.println( "返回结果:" +responseEntity.getBody()) } catch (Exception e){ log.error( "远程调用出现异常:" , e); } |
第二种方式
Spring Cloud Feign组件 + MultiValueMap + CommonInputStreamResource
CommonInputStreamResource对象的构造在上面已经实现了这里就不再重复构造,沿用上面的那个就行
feign接口
1
2
3
4
5
6
7
8
9
|
@Component @FeignClient (name = "taxRecodes" , url = "${spider.url}" , qualifier = "TaxRecodeFeignClient" ,fallback = TaxRecodeFallBack. class ) public interface TaxRecodeFeignClient { /** * 单证申请-合同信息表附件上传 */ @PostMapping (value = "/attachFile/upload" ,consumes = MediaType.MULTIPART_FORM_DATA_VALUE) String attachFileUpload(MultiValueMap<String, Object> multiValueMap); } |
请求部分
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
@PostMapping ( "/upload" ) public void upload(){ try { File file= new File( "E:\\1.rar" ); FileInputStream fileInputStream= new FileInputStream(file); CommonInputStreamResource commonInputStreamResource = null ; try { commonInputStreamResource = new CommonInputStreamResource(fileInputStream,fileInputStream.available(),file.getName()); } catch (Exception e) { log.error( "文件输入流转换错误:" ,e); } MultiValueMap<String, Object> dto= new LinkedMultiValueMap<String, Object>(); dto.add( "cookie" , "xxx" ); dto.add( "file" ,commonInputStreamResource); dto.add( "seqNoFile" , "xxx" ); String returnInfo = taxRecodeFeignClient.attachFileUpload(dto); JSONObject returnInfoJsonObject = JSONObject.parseObject(returnInfo); } catch (Exception e){ log.error( "异常:" ,e); } } |
RestTemplate调用远程接口添加请求头
项目中我们经常会碰到与第三方系统对接,通过调用第三方系统中的接口来集成服务,为了接口的安全性都为加一些验证,比如:
basic、authority等,通过请求头添加authrization的机制比较容易接入,从第三方系统获取到authorization,然后请求接口时在请求头上带上获取到的authorization,说了怎么多不如直接上代码更容易理解。
1
2
3
4
5
6
7
8
9
|
// 获取第三方的authorization String auth= OAuthContentHelper.getAuthorizationHeader(); HttpHeaders requestHeader= new HttpHeaders(); // 将获取到的authorization添加到请求头 requestHeader.add(AuthConstants.AUTHORIZATION_HEADER,auth); // 构建请求实体 HttpEntity<Object> requestEntity= new HttpEntity(requestParam,requestHeaders); // 使用restTemplate调用第三方接口 restTemplate.exchage(url,HttpMethod.POST,requestEntity,responseClass); |
以上为个人经验,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/y_bccl27/article/details/115088532