前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >springboot实现下载文件,并且在下载完成删除文件

springboot实现下载文件,并且在下载完成删除文件

原创
作者头像
高久峰
发布2024-05-07 12:00:20
1880
发布2024-05-07 12:00:20
举报

需求是别人请求我的接口,我的接口调用第三方接口下载文件到我本地,我再把文件给别人。由于我的sdk是把文件保存在本地,我需要保证本地不会有太多的临时文件占用,因此需要下载完成删除文件。

首先我们是不清楚用户下载完成的时间的,但是我们只需要保证把所有数据写入缓冲区后删除文件即可。以下是代码,代码是粗略版,仅供参考。

代码语言:javascript
复制
/**
     * 下载文件
     */
    @ApiOperation(value = "下载文件", notes = "下载文件")
    @PostMapping(value = "/contract/download", produces = "application/json;charset=UTF-8")
    public void contractDownload(HttpServletResponse response, @RequestBody X2ContractDownloadRequest req) {
        try {
            // 获取远程文件
            Path filePath = iQiyuesuoSdkService.contractDownload(req);

            // 设置响应头
            response.setContentType("application/octet-stream");
            response.setHeader("Content-Disposition", "attachment;filename=\"" + filePath.getFileName().toString() + "\"");

            // 分段下载
            int BUFFER_SIZE = 1024 * 100; // 100KB
            try (FileInputStream fis = new FileInputStream(filePath.toFile());
                 BufferedInputStream bis = new BufferedInputStream(fis);
                 ServletOutputStream sos = response.getOutputStream()) {
                byte[] buffer = new byte[BUFFER_SIZE];
                int bytesRead;
                while ((bytesRead = bis.read(buffer)) != -1) {
                    sos.write(buffer, 0, bytesRead);
                }
            }

            // 删除文件
            Files.delete(filePath.toFile().toPath());
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

也可以通过springboot的 ResponseEntity<InputStreamResource>+FileInputStream  重写read方法在读取完成删除文件,只是我没有测试成功,时间原因就先临时解决下。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档