RestTemplate直接发送内存文件对象multipart

一、 RestTemplate直接发送内存文件对象multipart

1. 创建一个字节数组输出流对象,并使用 Word 文档对象的 write 方法将其内容写入该流

2. 使用Spring的 RestTemplate 来发送含有附件的POST请求。

使用 HttpHeaders 设置 multipart/form-data 头:
然后,构建一个由字节数组和其他信息(如原始文件名和媒体类型)组成的资源对象,用于作为 HTTP 请求的一部分。然后,构造一个 MultiValueMap 对象来保存文件和其他表单数据:
最后,使用 RestTemplate 的exchange或postForEntity 方法发送请求:

public void uploadDoc(XWPFDocument document, ReportMessage reportMessage) {
    ByteArrayOutputStream out = new ByteArrayOutputStream();
    try {
      document.write(out);
    } catch (IOException e) {
      throw new RuntimeException(e);
    }

    HttpHeaders headers = new HttpHeaders();
    headers.setContentType(MediaType.MULTIPART_FORM_DATA);
    String fileName = "report-" + reportMessage.getReportId() + ".docx";
    ByteArrayResource resource =
        new ByteArrayResource(out.toByteArray()) {
          @Override
          public String getFilename() {
            return fileName;
          }
        };

    MultiValueMap<String, Object> body = new LinkedMultiValueMap<>();
    body.add("file", new HttpEntity<>(resource, headers));
    body.add("reportId", reportMessage.getReportId());

    ResponseEntity<String> response = restTemplate.postForEntity(upLoadUrl, body, String.class);

    log.info(response.getBody());
  }

以下引用https://www.yixuebiancheng.com/article/101586.html

二、 上传的文件是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资源对象。然后将该资源放入请求体中。

三、上传的文件是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读取的文件,所以直接能够获取到文件的大小和文件名。

四、上传的是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()方法,不然不成功。

  • 7
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值