导出

sa-admin/src/main/java/net/lab1024/sa/admin/module/business/caseplatformcase/converter/CustomExcelWriteHandler.java

@@ -7,7 +7,9 @@ import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
 import com.alibaba.excel.write.metadata.holder.WriteWorkbookHolder;
 import org.apache.poi.ss.usermodel.*;
 import org.apache.poi.ss.util.CellRangeAddress;
+import org.apache.poi.xssf.streaming.SXSSFSheet;
 
+import java.io.IOException;
 import java.util.List;
 
 public class CustomExcelWriteHandler implements WriteHandler {
@@ -15,6 +17,9 @@ public class CustomExcelWriteHandler implements WriteHandler {
     private final double totalTax;
     private final double totalTotal;
     private final int dataSize;
+    private boolean totalRowCreated = false;
+    // 缓存 Workbook（在 afterSheetCreate 中初始化）
+    private Workbook workbook;
 
     public CustomExcelWriteHandler(double totalActual, double totalTax, double totalTotal, int dataSize) {
         this.totalActual = totalActual;
@@ -24,8 +29,8 @@ public class CustomExcelWriteHandler implements WriteHandler {
     }
 
     public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
-        // 工作表创建后，先处理大标题、表头样式、列宽（此时还未写入数据，行未刷盘）
+        // 初始化 Workbook 缓存（关键：兼容低版本）
-        Workbook workbook = writeWorkbookHolder.getWorkbook();
+        this.workbook = writeWorkbookHolder.getWorkbook();
         Sheet sheet = writeSheetHolder.getSheet();
 
         // 1. 处理大标题（第0行）
@@ -33,7 +38,6 @@ public class CustomExcelWriteHandler implements WriteHandler {
         titleRow.setHeightInPoints(30);
         Cell titleCell = titleRow.createCell(0);
         titleCell.setCellValue("人工肝诊疗病例征集项目专家劳务费表");
-        // 合并大标题单元格（0行0列到0行最后一列）
         sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, 13));
         // 大标题样式
         CellStyle titleStyle = workbook.createCellStyle();
@@ -60,21 +64,14 @@ public class CustomExcelWriteHandler implements WriteHandler {
         headRow.setHeightInPoints(25);
     }
 
-    public void afterWorkbookDispose(WriteWorkbookHolder writeWorkbookHolder) {
+    // 关键：数据行写入完成后创建合计行（兼容低版本）
-        // 数据写入完成后，处理合计行和数据行高（此时所有数据行已写入，但合计行是新创建的，未刷盘）
+    public void afterRowDispose(WriteSheetHolder writeSheetHolder, Row row, Integer relativeRowIndex, Boolean isHead) throws IOException {
-        Workbook workbook = writeWorkbookHolder.getWorkbook();
+        // 仅在最后一行数据写入后创建合计行，且未创建过
-        Sheet sheet = workbook.getSheetAt(0);
+        if (!totalRowCreated && relativeRowIndex != null && relativeRowIndex == dataSize - 1) {
+            Sheet sheet = writeSheetHolder.getSheet();
 
-        // 1. 设置数据行行高（第2行到数据最后一行）
+            // 修正合计行索引：0=大标题，1=表头，2~1+dataSize=数据，2+dataSize=合计
-        for (int i = 2; i <= dataSize + 1; i++) { // 数据从第2行开始，共dataSize行
+            int totalRowIndex = 2 + dataSize;
-            Row row = sheet.getRow(i);
-            if (row != null) {
-                row.setHeightInPoints(22);
-            }
-        }
-
-        // 2. 处理合计行
-        int totalRowIndex = dataSize + 2; // 合计行在数据行最后一行的下一行
             Row totalRow = sheet.createRow(totalRowIndex);
             totalRow.setHeightInPoints(22);
 
@@ -93,13 +90,13 @@ public class CustomExcelWriteHandler implements WriteHandler {
             totalStyle.setBorderLeft(BorderStyle.THIN);
             totalStyle.setBorderRight(BorderStyle.THIN);
 
-        // 合并前6列（0-5）写“合计”
+            // 合并前6列写“合计”
             sheet.addMergedRegion(new CellRangeAddress(totalRowIndex, totalRowIndex, 0, 5));
             Cell totalCell = totalRow.createCell(0);
             totalCell.setCellValue("合计");
             totalCell.setCellStyle(totalStyle);
 
-        // 实发（6列）、个税（8列）、应发（9列）设置值和样式
+            // 实发、个税、应发赋值+样式
             Cell actualCell = totalRow.createCell(6);
             actualCell.setCellValue(totalActual);
             actualCell.setCellStyle(totalStyle);
@@ -112,7 +109,7 @@ public class CustomExcelWriteHandler implements WriteHandler {
             totalAmtCell.setCellValue(totalTotal);
             totalAmtCell.setCellStyle(totalStyle);
 
-        // 其他列补充边框样式
+            // 其他列补充边框
             CellStyle borderStyle = workbook.createCellStyle();
             borderStyle.setBorderBottom(BorderStyle.THIN);
             borderStyle.setBorderTop(BorderStyle.THIN);
@@ -122,7 +119,7 @@ public class CustomExcelWriteHandler implements WriteHandler {
 
             for (int i = 0; i < 14; i++) {
                 if (i == 0 || i == 6 || i == 8 || i == 9) {
-                continue; // 已设置样式的列跳过
+                    continue;
                 }
                 Cell cell = totalRow.getCell(i);
                 if (cell == null) {
@@ -130,14 +127,28 @@ public class CustomExcelWriteHandler implements WriteHandler {
                 }
                 cell.setCellStyle(borderStyle);
             }
+
+            // 标记合计行已创建
+            totalRowCreated = true;
+
+            // 强制刷新（兼容SXSSF）
+            if (sheet instanceof SXSSFSheet) {
+                ((SXSSFSheet) sheet).flushRows();
+            }
         }
 
-    // 其他默认方法实现
+        // 设置数据行行高
+        if (row.getRowNum() >= 2 && row.getRowNum() <= 1 + dataSize) {
+            row.setHeightInPoints(22);
+        }
+    }
+
+    // 其他默认方法（无需修改）
     public void beforeSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {}
     public void beforeWorkbookCreate(WriteWorkbookHolder writeWorkbookHolder) {}
     public void beforeWorkbookDispose(WriteWorkbookHolder writeWorkbookHolder) {}
+    public void afterWorkbookDispose(WriteWorkbookHolder writeWorkbookHolder) {}
     public void afterRowCreate(WriteSheetHolder writeSheetHolder, Row row, Integer relativeRowIndex, Boolean isHead) {}
-    public void afterRowDispose(WriteSheetHolder writeSheetHolder, Row row, Integer relativeRowIndex, Boolean isHead) {}
     public void beforeRowCreate(WriteSheetHolder writeSheetHolder, Integer rowIndex, Integer relativeRowIndex, Boolean isHead) {}
     public void beforeRowDispose(WriteSheetHolder writeSheetHolder, Row row, Integer relativeRowIndex, Boolean isHead) {}
     public void afterCellCreate(WriteSheetHolder writeSheetHolder, Cell cell, Head head, Integer relativeRowIndex, Boolean isHead) {}