SpringBoot使用EasyExcel并行导出多个excel文件并压缩zip下载

admin

SpringBoot使用EasyExcel并行导出多个excel文件并压缩zip下载
SpringBoot的同步导出方式中，服务器会阻塞直到Excel文件生成完毕，这在处理大量数据的导出功能是效率很低，所以我们可以将导出数据拆分后利用CompletableFuture，将导出任务异步化，并行使用easyExcel导出多个excel文件，最后这些文件进一步压缩成ZIP格式以方便下载。
DEMO代码:

1. @RestController
   
2. @RequestMapping("/export")
   
3. public class ExportController {
   
4. 
   
5.     @Autowired
   
6.     private ExcelExportService excelExportService;
   
7. 
   
8.     @GetMapping("/zip")
   
9.     public ResponseEntity<byte[]> exportToZip() throws Exception {
   
10.         List<List<Data>> dataSets = Lists.partition(dataList, 10);
    
11.         List<CompletableFuture<String>> futures = new ArrayList<>();
    
12.         // 异步导出所有Excel文件
    
13.          String outputDir = "path/to/output/dir/";
    
14.         for (List<Data> dataSet : dataSets) {
    
15.             futures.add(excelExportService.exportDataToExcel(dataSet, outputDir));
    
16.         }
    
17. 
    
18.         // 等待所有导出任务完成      
    
19.         CompletableFuture.allOf(futures.toArray(CompletableFuture[]::new)).get(10, TimeUnit.MINUTES);;
    
20. 
    
21.         // 收集Excel文件路径
    
22.         List<String> excelFilePaths = futures.stream()
    
23.                 .map(CompletableFuture::join)
    
24.                 .collect(Collectors.toList());
    
25. 
    
26.         // 压缩文件
    
27.         File zipFile = new File("path/to/output.zip");
    
28.         try (ZipOutputStream zipOut = new ZipOutputStream(new FileOutputStream(zipFile))) {
    
29.             for (String filePath : excelFilePaths) {
    
30.                 zipFile(new File(filePath), zipOut, new File(filePath).getName());
    
31.             }
    
32.         }
    
33. 
    
34.         // 返回ZIP文件
    
35.         byte[] data = Files.readAllBytes(zipFile.toPath());
    
36.         return ResponseEntity.ok()
    
37.                 .header("Content-Disposition", "attachment; filename="" + zipFile.getName() + """)
    
38.                 .contentType(MediaType.parseMediaType("application/zip"))
    
39.                 .body(data);
    
40.     }
    
41. 
    
42.     // 将文件添加到ZIP输出流中  
    
43.     private void zipFile(File file, ZipOutputStream zipOut, String entryName) throws IOException {  
    
44.         try (BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file))) {  
    
45.             ZipEntry zipEntry = new ZipEntry(entryName);  
    
46.             zipOut.putNextEntry(zipEntry);  
    
47.             byte[] bytesIn = new byte[4096];  
    
48.             int read;  
    
49.             while ((read = bis.read(bytesIn)) != -1) {  
    
50.                 zipOut.write(bytesIn, 0, read);  
    
51.             }  
    
52.             zipOut.closeEntry();  
    
53.         }  
    
54.     }
    
55. }

复制代码

异步并行生成excel文件,利用EasyExcel库来简化Excel的生成过程:

1. @Service
   
2. public class ExcelExportService {
   
3. 
   
4.     private static final String TEMPLATE_PATH = "path/to/template.xlsx";
   
5. 
   
6.     @Autowired
   
7.     private TaskExecutor taskExecutor;
   
8. 
   
9.     public CompletableFuture<Void> exportDataToExcel(List<Data> dataList, String outputDir) {
   
10.         Path temproaryFilePath = Files.createTempFile(outputDir, "excelFilePre",".xlsx");
    
11.         return CompletableFuture.runAsync(() -> {
    
12.             try (OutputStream outputStream = new FileOutputStream(temproaryFilePath )) {
    
13.                 EasyExcel.write(outputStream, Data.class)
    
14.                         .withTemplate(TEMPLATE_PATH)
    
15.                         .sheet()
    
16.                         .doFill(dataList)
    
17.                         .finish();
    
18.                return temproaryFilePath.toString();
    
19.             } catch (IOException e) {
    
20.                 throw new RuntimeException("Failed to export Excel file", e);
    
21.             }
    
22.         }, taskExecutor);
    
23.     }
    
24. }

复制代码