使用RestTemplate发送多部分文件

Question

我有两台服务器，其中服务器A向服务器B发送文件。在服务器B上，我有一个接收给定类别文件的endpoint。这是我需要将文件发送到的endpoint的签名：

@PostMapping("/uploadMultipleFiles/{projectId}")
public List<UploadFileResponseDts> uploadMultipleFiles(@RequestParam("files") MultipartFile[] files, @RequestParam("categoryId") Long categoryId, @PathVariable("projectId") Long projectId) {
    return uploadMulitpleFiles(files, categoryId, projectId);
}

现在我在用RESTTemplateExchange创建这样的请求时遇到了麻烦。在服务器A上，我试图逐个发送文件（服务器B上的endpoint需要接受多部分数组，因为它也在其他地方使用）。这就是我所尝试的：

public Optional<String> uploadFile(File file, Long projectId) throws Exception {

    String authToken = getAccessToken();
    String projectFileUploadEndpoint = fileUploadEndpoint + SEPARATOR + projectId;

    FileInputStream input = new FileInputStream(file);
    byte[] bytes = IOUtils.toByteArray(input);

    MultiValueMap<String, Object> parts =
            new LinkedMultiValueMap<>();
            parts.add("files", new ByteArrayResource(bytes));
    parts.add("categoryId", 0L);

    RestTemplate restTemplate = new RestTemplate();
    HttpHeaders headers = new HttpHeaders();
    headers.setContentType(MediaType.MULTIPART_FORM_DATA);
    headers.set("Authorization", authToken);

    HttpEntity<MultiValueMap<String, Object>> requestEntity =
            new HttpEntity<>(parts, headers);

    ResponseEntity<String> response =
            restTemplate.exchange(projectFileUploadEndpoint ,
                    HttpMethod.POST, requestEntity, String.class);

    return Optional.empty();
}

这几乎工作，唯一的问题是，在服务器B文件从@recestParam（文件）MultipartFile[]文件总是空。我假设问题是我在服务器A上组装部件的方式，但是我找不到我的错误。你能指出来吗？

Answer 1

在您的控制器中一切都很好，问题在于您的客户端代码。

基本上，您应该使用FileSystemResources对象而不是ByteArrayResources。基本动机是关于文件元数据。使用FileSystemResourcesSpring能够重建所需的MultipartFile信息，因此您的代码可能如下所示

  @Test
  public void contextLoads()
  {
    File file = new File("test.txt");
    FileSystemResource fileSystemResource = new FileSystemResource(file);

    MultiValueMap<String, Object> parts = new LinkedMultiValueMap<>();
    parts.add("categoryId", 0L);
    parts.add("files", fileSystemResource);

    String serviceUrl = "http://localhost:8080/uploadMultipleFiles/0";

    RequestEntity<MultiValueMap<String, Object>> requestEntity = post(fromPath(serviceUrl).build().toUri())
      .contentType(MediaType.MULTIPART_FORM_DATA)
      .body(parts);

    RestTemplate restTemplate = new RestTemplate();

    restTemplate.setErrorHandler(new ResponseErrorHandler()
    {
      @Override
      public boolean hasError(ClientHttpResponse response) throws IOException
      {
        return false;
      }

      @Override
      public void handleError(ClientHttpResponse response) throws IOException
      {
        System.out.println(response);
      }
    });
    ResponseEntity<String> response = restTemplate.exchange(serviceUrl, HttpMethod.POST, requestEntity, String.class);

    assertThat(200, Is.is(response.getStatusCode()));

  }

我希望这个测试用例能够帮助您解决这个问题