|
|
@@ -2,15 +2,16 @@ package org.dromara.common.excel.core;
|
|
|
|
|
|
import cn.hutool.core.collection.CollUtil;
|
|
|
import cn.idev.excel.metadata.Head;
|
|
|
-import cn.idev.excel.write.handler.WorkbookWriteHandler;
|
|
|
-import cn.idev.excel.write.handler.context.WorkbookWriteHandlerContext;
|
|
|
+import cn.idev.excel.write.handler.SheetWriteHandler;
|
|
|
import cn.idev.excel.write.merge.AbstractMergeStrategy;
|
|
|
+import cn.idev.excel.write.metadata.holder.WriteSheetHolder;
|
|
|
+import cn.idev.excel.write.metadata.holder.WriteWorkbookHolder;
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
import org.apache.poi.ss.usermodel.Cell;
|
|
|
import org.apache.poi.ss.usermodel.Sheet;
|
|
|
import org.apache.poi.ss.util.CellRangeAddress;
|
|
|
|
|
|
-import java.util.*;
|
|
|
+import java.util.List;
|
|
|
|
|
|
/**
|
|
|
* 列值重复合并策略
|
|
|
@@ -18,7 +19,7 @@ import java.util.*;
|
|
|
* @author Lion Li
|
|
|
*/
|
|
|
@Slf4j
|
|
|
-public class CellMergeStrategy extends AbstractMergeStrategy implements WorkbookWriteHandler {
|
|
|
+public class CellMergeStrategy extends AbstractMergeStrategy implements SheetWriteHandler {
|
|
|
|
|
|
private final List<CellRangeAddress> cellList;
|
|
|
|
|
|
@@ -32,27 +33,28 @@ public class CellMergeStrategy extends AbstractMergeStrategy implements Workbook
|
|
|
|
|
|
@Override
|
|
|
protected void merge(Sheet sheet, Cell cell, Head head, Integer relativeRowIndex) {
|
|
|
- if (CollUtil.isEmpty(cellList)){
|
|
|
+ if (CollUtil.isEmpty(cellList)) {
|
|
|
return;
|
|
|
}
|
|
|
- //单元格写入了,遍历合并区域,如果该Cell在区域内,但非首行,则清空
|
|
|
+ // 单元格写入了,遍历合并区域,如果该Cell在区域内,但非首行,则清空
|
|
|
final int rowIndex = cell.getRowIndex();
|
|
|
for (CellRangeAddress cellAddresses : cellList) {
|
|
|
final int firstRow = cellAddresses.getFirstRow();
|
|
|
- if (cellAddresses.isInRange(cell) && rowIndex != firstRow){
|
|
|
+ if (cellAddresses.isInRange(cell) && rowIndex != firstRow) {
|
|
|
cell.setBlank();
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public void afterWorkbookDispose(final WorkbookWriteHandlerContext context) {
|
|
|
- if (CollUtil.isEmpty(cellList)){
|
|
|
+ public void afterSheetCreate(final WriteWorkbookHolder writeWorkbookHolder, final WriteSheetHolder writeSheetHolder) {
|
|
|
+ if (CollUtil.isEmpty(cellList)) {
|
|
|
return;
|
|
|
}
|
|
|
- //当前表格写完后,统一写入
|
|
|
+ // 在 Sheet 创建时提前写入合并区域;后续写入只会影响首格,不会移除合并
|
|
|
+ final Sheet sheet = writeSheetHolder.getSheet();
|
|
|
for (CellRangeAddress item : cellList) {
|
|
|
- context.getWriteContext().writeSheetHolder().getSheet().addMergedRegion(item);
|
|
|
+ sheet.addMergedRegion(item);
|
|
|
}
|
|
|
}
|
|
|
|