如何通过 Spring Rest Api 检查文件是否已完全下载

Question

我创建了简单的 rest api 来提供来自 hdfs 的文件（文件很大，我不想在本地复制它们）。

我想记录文件下载成功完成的信息，即读取整个流，但我不知道如何。我只能记录文件下载开始的信息。

我将不胜感激。

    @Autowired
    private FileDownloadService fds;

    @RequestMapping(value = GET_FILE_PATH, method = RequestMethod.GET)
    @Produces(MediaType.APPLICATION_OCTET_STREAM_VALUE)
    public ResponseEntity getFileStream(@RequestParam("name") String name) {
        LOG.info("Processing for filename: " + name);
        try {
            Path p = fds.getFilePath(name);
            org.apache.hadoop.fs.FSDataInputStream is = fds.getFileStream(p);

            return ResponseEntity.ok().header("Content-Disposition", "attachment; filename=\"" + p.getName() + "\"'").contentLength(fds.getFileLength(p))
                    .contentType(MediaType.APPLICATION_OCTET_STREAM).body(new InputStreamResource(is));

        } catch (Exception e) {
            LOG.error(e.getLocalizedMessage(), e);
            return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("Internal Server Error");
        } finally {
           LOG.info("File: " + name + " download started");
        }
    }

Answer 1

您可以尝试在 InputStream 上创建一个包装器并在流关闭时触发一些标志 (close())。

比如你可以以ProxyInputStream为基础：

 ProxyInputStreamis = new ProxyInputStream(fds.getFileStream(p)) {
    @Override
    public void close() throws IOException {
            super.close();
            // some trigger
    }
 };