在Spring boot中下载AWS S3文件作为流

Question

我想公开一个API来将S3 bucket文件内容作为流下载给它的使用者。API URL类似于/downloadfile/**，即GET request。

1. 现在返回类型应该是什么？我尝试使用accept header=application/octet-stream，但不起作用。
2. 我不想把文件的内容写到任何文件中并发送它。它应该作为流返回。

这里是控制器伪代码，我写到现在，它给我406错误所有的时间。

 @GetMapping(value = "/downloadfile/**", produces = { MediaType.APPLICATION_OCTET_STREAM_VALUE })
    public ResponseEntity<Object> downloadFile(HttpServletRequest request) {
       //reads the content from S3 bucket and returns a S3ObjectInputStream
       S3ObjectInputStream object = null;
       object = publishAmazonS3.getObject("12345bucket", "/logs/file1.log").getObjectContent();
       return object
    }

这里对这样做的方式和我做错了什么有什么建议吗？

Answer 1

通过使用Spring中的StreamingResponseBody类，我可以将该文件作为流下载。

下面是我使用的代码：

    @GetMapping(value = "/downloadfile/**", produces = { MediaType.APPLICATION_OCTET_STREAM_VALUE })
    public ResponseEntity<S3ObjectInputStream> downloadFile(HttpServletRequest request) {
       //reads the content from S3 bucket and returns a S3ObjectInputStream
       S3Object object = publishAmazonS3.getObject("12345bucket", "/logs/file1.log");
       S3ObjectInputStream finalObject = object.getObjectContent();

        final StreamingResponseBody body = outputStream -> {
            int numberOfBytesToWrite = 0;
            byte[] data = new byte[1024];
            while ((numberOfBytesToWrite = finalObject.read(data, 0, data.length)) != -1) {
                System.out.println("Writing some bytes..");
                outputStream.write(data, 0, numberOfBytesToWrite);
            }
            finalObject.close();
        };
        return new ResponseEntity<>(body, HttpStatus.OK);
    }

测试流是否正确的方法是下载一个400MB左右的文件。通过传入vm选项将Xmx减少到256MB。现在，比较使用StreamingResponseBody和不使用StreamingResponseBody的下载功能，当使用传统的OutputStreams编写内容时，您将获得OutofMemoryError