StreamingResponseBody正在返回空文件

Question

我正在尝试创建一个rest服务，使用Springboot从存储库中下载文件。

我试图返回一个带有StreamingResponseBody的ResponseEntity，将我从存储库中获得的文件作为InputStream返回。

这是我目前拥有的代码：

@GetMapping(path = "/downloadFile")
    public ResponseEntity<StreamingResponseBody> downloadFile(@RequestParam(value = "documentId") String documentId,
            HttpServletRequest request, HttpServletResponse response) throws InterruptedException, IOException {

        InputStream is = downloadService.getDocument(documentId);

        StreamingResponseBody out = outputStream -> {

            outputStream.write(IOUtils.toByteArray(is));
        };

        HttpHeaders headers = new HttpHeaders();
        headers.add("Content-Type", "text/csv");
        headers.add("Content-Disposition", "attachment; filename=" + documentId);
        headers.add("Pragma", "no-cache");
        headers.add("Cache-Control", "no-cache");

        return (new ResponseEntity<>(out, headers, HttpStatus.OK));

    }

当我直接使用浏览器或邮递员使用此端点时，下载的文件为空。我知道OutputStream是异步写入的(异步是在config类中启用的)。

如何使用此服务并将文件完全写入我正在使用的存储库？(如果可能，使用Postman，仅用于测试目的)

我是否正确地构建了服务？

Answer 1

我稍微修改了一下代码，在我的documentId中是要下载的文件的名称。我测试过了，它工作得很好。检查下面的代码。

@GetMapping(path = "/downloadFile")
public ResponseEntity<StreamingResponseBody> downloadFile(
      @RequestParam(value = "documentId") String documentId,
      HttpServletRequest request,
      HttpServletResponse response)
      throws InterruptedException, IOException {
    String dirPath = "E:/sure-delete/"; //Directory having the files
    InputStream inputStream = new FileInputStream(new File(dirPath + documentId));
    final StreamingResponseBody out =
        outputStream -> {
          int nRead;
          byte[] data = new byte[1024];
          while ((nRead = inputStream.read(data, 0, data.length)) != -1) {
            System.out.println("Writing some bytes of file...");
            outputStream.write(data, 0, nRead);
          }
        };
    HttpHeaders headers = new HttpHeaders();
    headers.add("Content-Type", "text/csv");
    headers.add("Content-Disposition", "attachment; filename=" + documentId);
    headers.add("Pragma", "no-cache");
    headers.add("Cache-Control", "no-cache");
    return ResponseEntity.ok().headers(headers).body(out);
  }