springboot相应文件流文件给欣赏器+前端下载

springboot相应文件流文件给欣赏器+前端下载

1.controller:

1. @Api(tags = {"【样本提取系统】-api"})
2. @RestController("YbtqYstbtqController")
3. @RequiredArgsConstructor
4. @RequestMapping("/ybtq-ystbtq")
5. @Slf4j
6. public class YbtqYstbtqController {
8. private final YbService ybService;
10. @ApiOperation(value = "【样本集管理】-样本集导出", notes = "export", produces = "application/octet-stream")
11.     @PostMapping("/ybgl-set-cloud-download")
12.     @OpLog("【样本集管理】-样本集导出")
13.     public void ybGlYbSetDownload(@RequestBody List<String> pkIds, HttpServletResponse response) throws Exception {
14.         FileDownloadVo fileDownloadVo = tileSetService.ybGlYbSetDownload(pkIds);
15.         if (null != fileDownloadVo && fileDownloadVo.getFileStream() != null) {
16.             try (InputStream inputStream = fileDownloadVo.getFileStream();
17.                  OutputStream outputStream = response.getOutputStream()) {
18.                 //先将字符串以UTF-8转换成字节数组,再将字节数组以ISO-8859-1转换字符串(标准)
19.                 String fileName = new String(fileDownloadVo.getFileName().getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1);
20.                 response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName,"utf-8"));
21.                 response.setContentType(fileDownloadVo.getContentType());
22.                 response.setCharacterEncoding("UTF-8");
23.                 // 响应文件给浏览器
24.                 IOUtils.copy(inputStream, outputStream);
25.             } catch (Exception e) {
26.                 e.printStackTrace();
27.             }
28.         } else{
29.             response.setContentType("application/json;charset=utf-8");
30.             response.getWriter().write(JSONUtil.toJsonStr(Result.error("所选样本集无有效文件！")));
31.         }
32.     }
34. }

复制代码

文件下载对象FileDownloadVo：

1. /**
2. * 文件下载对象
3. */
4. @Data
5. public class FileDownloadVo {
6.     /**
7.      * 文件名称
8.      */
9.     @ApiModelProperty("文件名称")
10.     private String fileName;
11.     /**
12.      * 文件content-type
13.      */
14.     @ApiModelProperty("content-type")
15.     private String contentType;
17.     /**
18.      * 文件流
19.      */
20.     @ApiModelProperty("文件流")
21.     private InputStream fileStream;
22. }

复制代码

2.serviceImpl:

1. @Override
2.     public FileDownloadVo ybGlYbSetDownload(List<String> pkIds) {
3.         // 文件根目录
4.         FileDownloadVo ret = new FileDownloadVo();
5.         // 获取样本集信息
6.         List<TileSet> ybSetList = tileSetMapper.selectList(new LambdaQueryWrapper<TileSet>().in(TileSet::getPkId, pkIds));
7.         // 查询到的样本集，影像压缩成一个zip
8.         if (ybSetList.size() > 0) {
9.             try {
10.                 // 获取当前日期并格式化为 yyyyMMdd
11.                 String currentDate = new SimpleDateFormat("yyyyMMdd").format(new Date());
12.                 // 创建临时 ZIP 文件
13.                 File tempZipFile = Files.createTempFile("样本集_" + currentDate + "_", ".zip").toFile();
15.                 // 创建临时根文件夹用于存放样本文件夹
16.                 File tempRootDir = Files.createTempDirectory("样本集_" + currentDate + "_").toFile();
18.                 for (TileSet tileSet : ybSetList) {
19.                     // 为每个样本集创建一个文件夹
20.                     File sampleDir = new File(tempRootDir, tileSet.getTileName() + "_" + tileSet.getPkId());
21.                     sampleDir.mkdirs();
23.                     // 获取文件复制到临时目录下
24.                     setTitleSetFile(tileSet, sampleDir);
25.                 }
26.                 try (ZipOutputStream zipOut = new ZipOutputStream(new FileOutputStream(tempZipFile))) {
27.                     // 遍历 tempRootDir 下的所有子文件夹并添加到 ZIP 文件中
28.                     File[] subDirs = tempRootDir.listFiles();
29.                     if (subDirs != null) {
30.                         for (File subDir : subDirs) {
31.                             UnZipUtils.zipFiles(subDir, subDir.getName(), zipOut); // 将每个子文件夹递归添加到 ZIP
32.                         }
33.                     }
34.                 } finally {
35.                     // 删除临时文件夹及其内容
36.                     UnZipUtils.deleteDirectory(tempRootDir);
37.                 }
39.                 try {
40.                     // 将 ZIP 文件转为 FileInputStream
41.                     ret.setFileStream(new FileInputStream(tempZipFile));
42.                 } catch (IOException e) {
43.                     e.printStackTrace();
44.                 }
46.                 // 异步任务：300秒后删除文件
47.                 UnZipUtils.deleteAsyncFiles(tempZipFile);
48.             } catch (IOException e) {
49.                 e.printStackTrace();
50.             }
51.         }
52.         ret.setContentType("application/zip");
53.         ret.setFileName(String.valueOf(new Date().getTime()));
54.         return ret;
55.     }
57. // 获取文件复制到临时目录下
58.     private void setTitleSetFile(TileSet tileSet, File sampleDir) {
59.         String tileSetPath = tileSet.getTilePath();
61.         boolean isTileSetPath = tileSetPath != null && !Objects.equals(tileSetPath, "");
62.         if (isTileSetPath) {
63.             // 获取样本切片file
64.             List<File> tileSetFileList = UnZipUtils.getFileByImagePaths(tileSetPath);
65.             for (File file : tileSetFileList) {
66.                 try {
67.                     Files.copy(file.toPath(), new File(sampleDir, file.getName()).toPath(), StandardCopyOption.REPLACE_EXISTING);
68.                 } catch (Exception e) {
69.                     System.out.print("前时项影像文件路径无效：" + file.getPath());
70.                 }
71.             }
72.         }
73.     }
75. /**
76.      * 辅助方法，用于从ZIP输入流中提取文件
77.      *
78.      * @param zis ZIP输入流
79.      * @param filePath 文件的完整路径
80.      * @throws IOException 如果发生I/O错误
81.      */
82.     public static void extractFile(ZipInputStream zis, String filePath) throws IOException {
83.         BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(filePath));
84.         byte[] bytesIn = new byte[4096];
85.         int read = 0;
86.         while ((read = zis.read(bytesIn)) != -1) {
87.             bos.write(bytesIn, 0, read);
88.         }
89.         bos.close();
90.     }
92. public static void zipFiles(File fileToZip, String fileName, ZipOutputStream zipOut) throws IOException {
93.         if (fileToZip.isHidden()) return;
95.         if (fileToZip.isDirectory()) {
96.             if (fileName.endsWith("/")) {
97.                 zipOut.putNextEntry(new ZipEntry(fileName));
98.                 zipOut.closeEntry();
99.             } else {
100.                 zipOut.putNextEntry(new ZipEntry(fileName + "/"));
101.                 zipOut.closeEntry();
102.             }
104.             File[] children = fileToZip.listFiles();
105.             for (File childFile : children) {
106.                 zipFiles(childFile, fileName + "/" + childFile.getName(), zipOut);
107.             }
108.             return;
109.         }
111.         try (FileInputStream fis = new FileInputStream(fileToZip)) {
112.             ZipEntry zipEntry = new ZipEntry(fileName);
113.             zipOut.putNextEntry(zipEntry);
114.             byte[] bytes = new byte[1024];
115.             int length;
116.             while ((length = fis.read(bytes)) >= 0) {
117.                 zipOut.write(bytes, 0, length);
118.             }
119.         }
120.     }
122. public static void deleteDirectory(File file) {
123.         File[] contents = file.listFiles();
124.         if (contents != null) {
125.             for (File f : contents) {
126.                 deleteDirectory(f);
127.             }
128.         }
129.         file.delete();
130.     }
132. // 异步任务-删除临时文件
133.     public static void deleteAsyncFiles(File tempZipFile) {
134.         executorService.submit(() -> {
135.             try {
136.                 TimeUnit.SECONDS.sleep(300);
137.                 Files.deleteIfExists(tempZipFile.toPath());
138.                 System.out.println("临时压缩包已删除：" + tempZipFile.toPath());
139.             } catch (Exception e) {
140.                 System.err.println("临时压缩包删除失败：" + e.getMessage());
141.             }
142.         });
143.     }
145. public static List<File> getFileByImagePaths(String imagePaths) {
146.         List<File> ret = new ArrayList<>();
147.         if (imagePaths != null && !imagePaths.equals("")) {
148.             List<String> preImagePathList = Arrays.asList(imagePaths.split(","));
149.             if (preImagePathList.size() > 0) {
150.                 for (String preImagePath : preImagePathList) {
151.                     // 根据path获取文件
152.                     File file = new File(preImagePath);
153.                     ret.add(file);
154.                 }
155.             }
156.         }
157.         return ret;
158.     }

复制代码

3.前端下载：

2. /**
3.    * 【样本集管理】-样本集导出
4.    */
5.   ybGlYbSetDownload(context = undefined, params = undefined, fileName) {
6.     const url = `${this.interfaceUrl}/ybtq-ystbtq/ybgl-set-cloud-download`;
7.     return ajaxHelperInstance.downloadFilePost(url, fileName, {}, params);
8.   }
10. async downloadFilePost(url, fileName, urlParam, data = {}, requestMethod = 'POST', vueContext = undefined, spinName = 'spinning', useStreamSaver = false) {
11.         url = this.prepareUrl(url, {...urlParam, ...data});
12.         data = convertDateParam(data);
13.         setVueContextSpinState(vueContext, spinName);
14.         const requestOptions = this.setupRequestOptions(requestMethod, data);
16.         try {
17.             let response;
18.             if (useStreamSaver) {
19.                 await this.downloadWithStream(url, fileName, requestOptions);
20.             } else {
21.                 response = await axios({url, ...requestOptions});
22.                 await this.handleResponse(response, fileName);
23.             }
24.         } catch (error) {
25.             this.handleError(error);
26.         } finally {
27.             resetVueContextSpinState(vueContext, spinName);
28.         }
29.     }
31. prepareUrl(url, params) {
32.         const reqData = _.map(params, (val, prop) => ({name: prop, value: val}));
33.         const queryString = abp.utils.buildQueryString(reqData);
34.         return url + queryString;
35.     }
37. /**
38. * 转换时间参数,主要处理表单传入参数含有moment对象的问题
39. * @param obj 传入对象
40. * @constructor
41. */
42. function convertDateParam(obj) {
43.     _.each(obj, (val, prop) => {
44.         if (val instanceof moment) {
45.             obj[prop] = (val as any).format('YYYY-MM-DD HH:mm:ss');
46.         }
48.         if (val instanceof Date) {
49.             obj[prop] = moment(val)
50.                 .format('YYYY-MM-DD HH:mm:ss');
51.         }
52.     });
53.     return obj;
54. }
56. function setVueContextSpinState(vueContext, spinName: string) {
57.     if (vueContext && vueContext[spinName] != undefined) {
58.         vueContext[spinName] = true;
59.     }
60. }
62. setupRequestOptions(method, data = null) {
63.         const headers = {
64.             'Content-Type': 'application/json',
65.             'Authorization': 'Bearer ' + window.abp.auth.getToken(),
66.         };
67.         return {
68.             method: method,
69.             headers: headers,
70.             responseType: 'blob',
71.             data: data ? JSON.stringify(data) : undefined,
72.         };
73.     }
75. /**
76.      * 通过fetch Api进行流式下载
77.      * @param url 下载地址
78.      * @param fileName 文件名称
79.      * @param options 下载选项
80.      */
81.     async downloadWithStream(url, fileName, options: any = {}) {
82.         const {timeout = 1000 * 60 * 180} = options;
83.         const response = await this.fetchWithTimeout(url, options, timeout);
84.         if (!response.ok) {
85.             notification.warning({
86.                 message: '提示',
87.                 description: '下载失败',
88.             });
89.             throw new Error('下载失败');
90.         }
91.         const fileStream = StreamSaver.createWriteStream(fileName);
92.         const writer = fileStream.getWriter();
93.         if (response.body.pipeTo) {
94.             writer.releaseLock();
95.             return response.body.pipeTo(fileStream);
96.         }
98.         const reader = response.body.getReader();
99.         const pump = () =>
100.             reader
101.                 .read()
102.                 .then(({value, done}) => (done ? writer.close() : writer.write(value).then(pump)));
104.         return pump();
105.     }
107. async handleResponse(response, fileName) {
108.         const contentType = response.headers['content-type'];
109.         console.log(contentType);
110.         if (contentType && contentType.includes('application/json')) {
111.             const reader = new FileReader();
112.             reader.onload = () => {
113.                 try {
114.                     const jsonData = JSON.parse(reader.result as string);
115.                     if (jsonData && jsonData.message) {
116.                         notification.error({
117.                             message: '提示',
118.                             description: jsonData.message,
119.                         });
120.                     }
121.                 } catch (error) {
122.                     console.error('解析json失败:', error);
123.                 }
124.             };
125.             reader.readAsText(response.data);
126.         } else {
127.             const blob = new Blob([response.data]);
128.             const link = document.createElement('a');
129.             link.href = URL.createObjectURL(blob);
130.             link.download = fileName;
131.             link.click();
132.             URL.revokeObjectURL(link.href);
133.         }
134.     }
136. handleError(error) {
137.         console.error('下载失败:', error);
138.     }
140. function resetVueContextSpinState(vueContext, spinName: string) {
141.     if (vueContext && vueContext[spinName] == true) {
142.         vueContext[spinName] = false;
143.     }
144. }

复制代码