xinyin025/RuoYi-Vue-Plus
1
0
Fork 0
mirror of https://github.com/dromara/RuoYi-Vue-Plus.git synced 2026-04-30 23:14:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update 优化 exportExcelZip 方法

Browse Source
This commit is contained in:
疯狂的狮子Li
2026-04-24 16:35:39 +08:00
parent c75f549ea3
commit c4d0157531
1 changed files with 73 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

143
ruoyi-common/ruoyi-common-excel/src/main/java/org/dromara/common/excel/utils/ExcelUtil.java
View File

@@ -2,9 +2,7 @@ package org.dromara.common.excel.utils;
import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.collection.ListUtil; import cn.hutool.core.collection.ListUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.io.resource.ClassPathResource; import cn.hutool.core.io.resource.ClassPathResource;
import cn.hutool.core.thread.ThreadUtil;
import cn.hutool.core.util.CharsetUtil; import cn.hutool.core.util.CharsetUtil;
import cn.hutool.core.util.IdUtil; import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.ZipUtil; import cn.hutool.core.util.ZipUtil;
@@ -26,13 +24,13 @@ import org.dromara.common.excel.core.*;
import org.dromara.common.excel.handler.DataWriteHandler; import org.dromara.common.excel.handler.DataWriteHandler;
import java.io.*; import java.io.*;
import java.util.Collection; import java.net.URLEncoder;
import java.util.List; import java.nio.charset.StandardCharsets;
import java.util.Map; import java.util.*;
import java.util.concurrent.ConcurrentSkipListMap; import java.util.concurrent.ExecutionException;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService; import java.util.concurrent.ExecutorService;
import java.util.concurrent.TimeUnit; import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.zip.ZipEntry; import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream; import java.util.zip.ZipOutputStream;
@@ -109,84 +107,89 @@ public class ExcelUtil {
* @param clazz 实体类 * @param clazz 实体类
*/ */
public static <T> void exportExcelZip(List<T> list, String sheetName, Class<T> clazz, HttpServletResponse response) { public static <T> void exportExcelZip(List<T> list, String sheetName, Class<T> clazz, HttpServletResponse response) {
exportExcelZip(list, sheetName, clazz, response, 999, 5, 10); exportExcelZip(list, sheetName, clazz, response, 999);
} }
/** /**
* 大数据量Excel导出 * 大数据量Excel导出
* *
* @param list 导出数据集合 * @param list 导出数据集合
* @param sheetName 工作表的名称 * @param sheetName 工作表的名称
* @param clazz 实体类 * @param clazz 实体类
* @param pageSize 每页条数 * @param pageSize 每页条数
* @param coreThread 核心线程
* @param maxThread 最大线程
*/ */
public static <T> void exportExcelZip(List<T> list, String sheetName, Class<T> clazz, HttpServletResponse response, int pageSize, int coreThread, int maxThread) { public static <T> void exportExcelZip(List<T> list, String sheetName, Class<T> clazz, HttpServletResponse response, int pageSize) {
List<List<T>> pageList = ListUtil.partition(list, pageSize);
if (pageList.size() <= 1) {
exportSingleExcel(list, sheetName, clazz, response);
return;
}
Map<String, byte[]> excelMap = buildExcelZipData(pageList, sheetName, clazz);
writeExcelZipResponse(sheetName, response, excelMap);
}
private static <T> void exportSingleExcel(List<T> list, String sheetName, Class<T> clazz, HttpServletResponse response) {
try { try {
List<List<T>> pageList = ListUtil.partition(list, pageSize); resetResponse(sheetName, response);
int totalPage = pageList.size(); ServletOutputStream os = response.getOutputStream();
exportExcel(list, sheetName, clazz, false, os, null);
} catch (IOException e) {
throw new RuntimeException("导出Excel异常", e);
}
}
// 数据量不足1页，直接导出普通Excel private static <T> Map<String, byte[]> buildExcelZipData(List<List<T>> pageList, String sheetName, Class<T> clazz) {
if (totalPage == 1) { Map<String, byte[]> excelMap = new LinkedHashMap<>(pageList.size());
resetResponse(sheetName, response); List<Future<Map.Entry<String, byte[]>>> futures = new ArrayList<>(pageList.size());
ServletOutputStream os = response.getOutputStream(); try (ExecutorService executor = Executors.newVirtualThreadPerTaskExecutor()) {
exportExcel(list, sheetName, clazz, false, os, null); for (int i = 0; i < pageList.size(); i++) {
return;
}
// ====================== 初始化线程池 ======================
ExecutorService executor = ThreadUtil.newExecutor(coreThread, maxThread);
// 存储结果：key=文件名，value=文件字节码（线程安全Map）
Map<String, byte[]> excelMap = new ConcurrentSkipListMap<>();
// 计数器：等待所有线程完成
CountDownLatch latch = new CountDownLatch(totalPage);
// ====================== 多线程生成Excel ======================
for (int i = 0; i < totalPage; i++) {
int pageNum = i + 1; int pageNum = i + 1;
List<T> pageData = pageList.get(i); List<T> pageData = pageList.get(i);
futures.add(executor.submit(() -> buildExcelZipEntry(pageData, sheetName, clazz, pageNum)));
executor.execute(() -> {
try {
ByteArrayOutputStream bos = new ByteArrayOutputStream();
String fileName = sheetName + "_第" + pageNum + "页.xlsx";
// 生成Excel
exportExcel(pageData, sheetName + "_第" + pageNum + "页", clazz, false, bos, null);
excelMap.put(fileName, bos.toByteArray());
} catch (Exception e) {
throw new RuntimeException("第" + pageNum + "页Excel生成失败", e);
} finally {
latch.countDown();
}
});
} }
for (Future<Map.Entry<String, byte[]>> future : futures) {
Map.Entry<String, byte[]> excel = getExcelZipEntry(future);
excelMap.put(excel.getKey(), excel.getValue());
}
}
return excelMap;
}
// 等待所有线程执行完毕（最多等待10分钟） private static <T> Map.Entry<String, byte[]> buildExcelZipEntry(List<T> pageData, String sheetName, Class<T> clazz, int pageNum) {
latch.await(10, TimeUnit.MINUTES); try (ByteArrayOutputStream bos = new ByteArrayOutputStream()) {
String exportSheetName = sheetName + "_第" + pageNum + "页";
exportExcel(pageData, exportSheetName, clazz, false, bos, null);
return Map.entry(exportSheetName + ".xlsx", bos.toByteArray());
} catch (Exception e) {
throw new RuntimeException("第" + pageNum + "页Excel生成失败", e);
}
}
// ====================== 打包ZIP并下载 ====================== private static Map.Entry<String, byte[]> getExcelZipEntry(Future<Map.Entry<String, byte[]>> future) {
try {
return future.get();
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
throw new RuntimeException("Excel导出线程被中断", e);
} catch (ExecutionException e) {
throw new RuntimeException("Excel导出失败", e.getCause());
}
}
private static void writeExcelZipResponse(String sheetName, HttpServletResponse response, Map<String, byte[]> excelMap) {
try {
response.setContentType("application/zip"); response.setContentType("application/zip");
response.setHeader("Content-Disposition", response.setHeader("Content-Disposition",
"attachment;filename*=UTF-8''" + java.net.URLEncoder.encode(sheetName, "UTF-8") + ".zip"); "attachment;filename*=UTF-8''" + URLEncoder.encode(sheetName, StandardCharsets.UTF_8) + ".zip");
try (ZipOutputStream zipOut = ZipUtil.getZipOutputStream(response.getOutputStream(), CharsetUtil.CHARSET_UTF_8)) {
ZipOutputStream zipOut = ZipUtil.getZipOutputStream(response.getOutputStream(), CharsetUtil.CHARSET_UTF_8); for (Map.Entry<String, byte[]> entry : excelMap.entrySet()) {
zipOut.putNextEntry(new ZipEntry(entry.getKey()));
// 写入ZIP zipOut.write(entry.getValue());
for (Map.Entry<String, byte[]> entry : excelMap.entrySet()) { zipOut.closeEntry();
zipOut.putNextEntry(new ZipEntry(entry.getKey())); }
zipOut.write(entry.getValue());
zipOut.closeEntry();
} }
} catch (IOException e) {
// 关闭资源 throw new RuntimeException("导出Zip异常", e);
IoUtil.close(zipOut);
executor.shutdown();
} catch (Exception e) {
throw new RuntimeException("Excel导出失败", e);
} }
} }