parent
48431670fc
commit
f6f120d9e6
@ -0,0 +1,107 @@ |
|||||||
|
package com.hfkj.common.utils; |
||||||
|
|
||||||
|
import com.alibaba.excel.metadata.CellData; |
||||||
|
import com.alibaba.excel.metadata.Head; |
||||||
|
import com.alibaba.excel.write.handler.CellWriteHandler; |
||||||
|
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder; |
||||||
|
import com.alibaba.excel.write.metadata.holder.WriteTableHolder; |
||||||
|
import org.apache.poi.ss.usermodel.Cell; |
||||||
|
import org.apache.poi.ss.usermodel.CellType; |
||||||
|
import org.apache.poi.ss.usermodel.Row; |
||||||
|
import org.apache.poi.ss.usermodel.Sheet; |
||||||
|
import org.apache.poi.ss.util.CellRangeAddress; |
||||||
|
|
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
public class ExcelFillCellMergeStrategyUtils implements CellWriteHandler { |
||||||
|
|
||||||
|
/** |
||||||
|
* 需要合并列的下标，从0开始 |
||||||
|
*/ |
||||||
|
private int[] mergeColumnIndex; |
||||||
|
/** |
||||||
|
* 从第几行开始合并，表头下标为0 |
||||||
|
*/ |
||||||
|
private int mergeRowIndex; |
||||||
|
|
||||||
|
public ExcelFillCellMergeStrategyUtils(int mergeRowIndex, int[] mergeColumnIndex) { |
||||||
|
this.mergeRowIndex = mergeRowIndex; |
||||||
|
this.mergeColumnIndex = mergeColumnIndex; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void beforeCellCreate(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, Row row, Head head, Integer integer, Integer integer1, Boolean aBoolean) { |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void afterCellCreate(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, Cell cell, Head head, Integer integer, Boolean aBoolean) { |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void afterCellDataConverted(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, CellData cellData, Cell cell, Head head, Integer integer, Boolean aBoolean) { |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void afterCellDispose(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, List<CellData> list, Cell cell, Head head, Integer integer, Boolean aBoolean) { |
||||||
|
//当前行
|
||||||
|
int curRowIndex = cell.getRowIndex(); |
||||||
|
//当前列
|
||||||
|
int curColIndex = cell.getColumnIndex(); |
||||||
|
|
||||||
|
if (curColIndex == 0) { |
||||||
|
Row preRow = cell.getSheet().getRow(curRowIndex - 1); |
||||||
|
if (preRow == null) { |
||||||
|
// 当获取不到上一行数据时，使用缓存sheet中数据
|
||||||
|
preRow = writeSheetHolder.getCachedSheet().getRow(curRowIndex - 1); |
||||||
|
} |
||||||
|
if (preRow != null) { |
||||||
|
if (cell.getStringCellValue().equals(preRow.getCell(0).getStringCellValue())) { |
||||||
|
for (int i = 0; i < mergeColumnIndex.length; i++) { |
||||||
|
if (curColIndex == mergeColumnIndex[i]) { |
||||||
|
mergeWithPrevRow(writeSheetHolder, cell, i, i); |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
// preRow.getCell(curColIndex);
|
||||||
|
|
||||||
|
/* if (curRowIndex > mergeRowIndex) { |
||||||
|
for (int i = 0; i < mergeColumnIndex.length; i++) { |
||||||
|
if (curColIndex == mergeColumnIndex[i]) { |
||||||
|
mergeWithPrevRow(writeSheetHolder, cell, curRowIndex, curColIndex); |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
}*/ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
private void mergeWithPrevRow(WriteSheetHolder writeSheetHolder, Cell cell, int curRowIndex, int curColIndex) { |
||||||
|
Sheet sheet = writeSheetHolder.getSheet(); |
||||||
|
List<CellRangeAddress> mergeRegions = sheet.getMergedRegions(); |
||||||
|
boolean isMerged = false; |
||||||
|
for (int i = 0; i < mergeRegions.size() && !isMerged; i++) { |
||||||
|
CellRangeAddress cellRangeAddr = mergeRegions.get(i); |
||||||
|
// 若上一个单元格已经被合并，则先移出原有的合并单元，再重新添加合并单元
|
||||||
|
if (cellRangeAddr.isInRange(curRowIndex - 1, curColIndex)) { |
||||||
|
sheet.removeMergedRegion(i); |
||||||
|
cellRangeAddr.setLastRow(curRowIndex); |
||||||
|
sheet.addMergedRegion(cellRangeAddr); |
||||||
|
isMerged = true; |
||||||
|
} |
||||||
|
} |
||||||
|
// 若上一个单元格未被合并，则新增合并单元
|
||||||
|
if (!isMerged) { |
||||||
|
CellRangeAddress cellRangeAddress = new CellRangeAddress(curRowIndex - 1, curRowIndex, curColIndex, |
||||||
|
curColIndex); |
||||||
|
sheet.addMergedRegion(cellRangeAddress); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,186 @@ |
|||||||
|
package com.hfkj.common.utils; |
||||||
|
|
||||||
|
import com.alibaba.excel.metadata.Head; |
||||||
|
import com.alibaba.excel.write.merge.AbstractMergeStrategy; |
||||||
|
import org.apache.commons.lang3.StringUtils; |
||||||
|
import org.apache.poi.ss.usermodel.Cell; |
||||||
|
import org.apache.poi.ss.usermodel.Sheet; |
||||||
|
import org.apache.poi.ss.util.CellRangeAddress; |
||||||
|
|
||||||
|
import java.util.*; |
||||||
|
|
||||||
|
/** |
||||||
|
* @className: MergeCellStrategyHandler |
||||||
|
* @author: HuRui |
||||||
|
* @date: 2024/8/2 |
||||||
|
**/ |
||||||
|
public class MergeCellStrategyHandler extends AbstractMergeStrategy { |
||||||
|
/** |
||||||
|
* 相同列合并 |
||||||
|
*/ |
||||||
|
private boolean alikeColumn; |
||||||
|
|
||||||
|
/** |
||||||
|
* 相同行合并 |
||||||
|
*/ |
||||||
|
private boolean alikeRow; |
||||||
|
|
||||||
|
/** 开始进行合并的行index */ |
||||||
|
private int rowIndex; |
||||||
|
|
||||||
|
/** 跨行合并的列index */ |
||||||
|
private Set<Integer> columns; |
||||||
|
|
||||||
|
private int currentRowIndex = 0; |
||||||
|
|
||||||
|
public MergeCellStrategyHandler(boolean alikeColumn, boolean alikeRow, int rowIndex, Set<Integer> columns){ |
||||||
|
this.alikeColumn = alikeColumn; |
||||||
|
this.alikeRow = alikeRow; |
||||||
|
this.rowIndex = rowIndex; |
||||||
|
this.columns = columns; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 指定是否进行跨列合并单元格 |
||||||
|
* @param alikeColumn |
||||||
|
* @param rowIndex |
||||||
|
*/ |
||||||
|
public MergeCellStrategyHandler(boolean alikeColumn, int rowIndex){ |
||||||
|
this(alikeColumn, false, rowIndex, new HashSet<>()); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 指定是否进行跨行合并单元格 |
||||||
|
* @param alikeRow |
||||||
|
* @param rowIndex |
||||||
|
* @param columns |
||||||
|
*/ |
||||||
|
public MergeCellStrategyHandler(boolean alikeRow, int rowIndex, Set<Integer> columns){ |
||||||
|
this(false, alikeRow, rowIndex, columns); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
@Override |
||||||
|
protected void merge(Sheet sheet, Cell cell, Head head, Integer integer) { |
||||||
|
int rowId = cell.getRowIndex(); |
||||||
|
currentRowIndex = rowId == currentRowIndex ? currentRowIndex : rowId; |
||||||
|
if (rowIndex > rowId) { |
||||||
|
return; |
||||||
|
} |
||||||
|
int columnId = cell.getColumnIndex(); |
||||||
|
// 列合并
|
||||||
|
if (alikeColumn && columnId > 0) { |
||||||
|
String currentCellVal = this.getCellVal(cell); |
||||||
|
Cell preCell = cell.getRow().getCell(columnId - 1); |
||||||
|
String preCellVal = this.getCellVal(preCell); |
||||||
|
if (null != currentCellVal && null != preCellVal && StringUtils.isNotBlank(preCellVal) && StringUtils.isNotBlank(currentCellVal)) { |
||||||
|
// 当前单元格内容与上一个单元格内容相等,进行合并处理
|
||||||
|
if (preCellVal.equals(currentCellVal)) { |
||||||
|
CellRangeAddress rangeAddress = new CellRangeAddress(currentRowIndex, currentRowIndex, columnId - 1, columnId); |
||||||
|
rangeAddress = this.findExistAddress(sheet, rangeAddress, currentCellVal); |
||||||
|
if (null != rangeAddress) { |
||||||
|
sheet.addMergedRegion(rangeAddress); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
// 行合并
|
||||||
|
if (alikeRow && rowIndex < rowId && columns.contains(columnId)) { |
||||||
|
String currentCellVal = this.getCellVal(cell); |
||||||
|
Cell preCell = sheet.getRow(rowId - 1).getCell(columnId); |
||||||
|
String preCellVal = this.getCellVal(preCell); |
||||||
|
if (null != currentCellVal && null != preCellVal && !preCellVal.isBlank() && !currentCellVal.isBlank()) { |
||||||
|
// 当前单元格内容与上一行单元格内容相等,进行合并处理
|
||||||
|
if (preCellVal.equals(currentCellVal)) { |
||||||
|
//sheet.validateMergedRegions();
|
||||||
|
CellRangeAddress rangeAddress = new CellRangeAddress(currentRowIndex - 1, currentRowIndex, columnId, columnId); |
||||||
|
rangeAddress = this.findExistAddress(sheet, rangeAddress, currentCellVal); |
||||||
|
if (null != rangeAddress) { |
||||||
|
sheet.addMergedRegion(rangeAddress); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 合并单元格地址范围,发现存在相同的地址则进行扩容合并 |
||||||
|
* |
||||||
|
* @param isColumn true:跨列合并 false:跨行合并 |
||||||
|
* @param sheet |
||||||
|
* @param rangeAddress : |
||||||
|
* @param currentVal 当前单元格中的值 |
||||||
|
* @return |
||||||
|
*/ |
||||||
|
private CellRangeAddress findExistAddress(Sheet sheet, CellRangeAddress rangeAddress, String currentVal) { |
||||||
|
List<CellRangeAddress> mergedRegions = sheet.getMergedRegions(); |
||||||
|
int existIndex = 0; |
||||||
|
Map<Integer, CellRangeAddress> existIdexMap = new LinkedHashMap<>(); |
||||||
|
if (null != mergedRegions && !mergedRegions.isEmpty()) { |
||||||
|
//验证当前合并的单元格是否存在重复(fixme 重复问题待处理!!!)
|
||||||
|
for (CellRangeAddress mergedRegion : mergedRegions) { |
||||||
|
if (mergedRegion.intersects(rangeAddress)) { |
||||||
|
existIdexMap.put(existIndex, mergedRegion); |
||||||
|
} |
||||||
|
existIndex++; |
||||||
|
} |
||||||
|
} |
||||||
|
if (existIdexMap.isEmpty()) { |
||||||
|
return rangeAddress; |
||||||
|
} |
||||||
|
List<Integer> existIndexList = new ArrayList<>(existIdexMap.size()); |
||||||
|
for (Map.Entry<Integer, CellRangeAddress> addressEntry : existIdexMap.entrySet()) { |
||||||
|
CellRangeAddress exist = addressEntry.getValue(); |
||||||
|
// 自动进行单元格合并处理
|
||||||
|
int firstRow = rangeAddress.getFirstRow(); |
||||||
|
int lastRow = rangeAddress.getLastRow(); |
||||||
|
int firstColumn = rangeAddress.getFirstColumn(); |
||||||
|
int lastColumn = rangeAddress.getLastColumn(); |
||||||
|
|
||||||
|
int firstRow1 = exist.getFirstRow(); |
||||||
|
int lastRow1 = exist.getLastRow(); |
||||||
|
int firstColumn1 = exist.getFirstColumn(); |
||||||
|
int lastColumn1 = exist.getLastColumn(); |
||||||
|
// 跨行合并 最后一列相等, 行不相等
|
||||||
|
if (lastRow > lastRow1 && lastColumn == lastColumn1) { |
||||||
|
// 检查进行跨行合并的单元格是否已经存在跨列合并
|
||||||
|
if (lastColumn > 0 && firstColumn1 != lastColumn1) { |
||||||
|
// 获取当前单元格的前一列单元格
|
||||||
|
String cellVal = this.getCellVal(sheet.getRow(lastRow).getCell(lastColumn - 1)); |
||||||
|
if (null != cellVal && cellVal.equals(currentVal)) { |
||||||
|
exist.setLastRow(lastRow); |
||||||
|
} |
||||||
|
} else { |
||||||
|
exist.setLastRow(lastRow); |
||||||
|
} |
||||||
|
rangeAddress = exist; |
||||||
|
existIndexList.add(addressEntry.getKey()); |
||||||
|
} |
||||||
|
|
||||||
|
// 跨列合并 行相等,列不相等
|
||||||
|
if (lastColumn > lastColumn1 && firstRow == firstRow1 ) { |
||||||
|
exist.setLastColumn(lastColumn); |
||||||
|
rangeAddress = exist; |
||||||
|
existIndexList.add(addressEntry.getKey()); |
||||||
|
} |
||||||
|
} |
||||||
|
// 移除已经存在且冲突的合并数据
|
||||||
|
if (existIndexList.isEmpty()) { |
||||||
|
rangeAddress = null; |
||||||
|
}else { |
||||||
|
sheet.removeMergedRegions(existIdexMap.keySet()); |
||||||
|
} |
||||||
|
return rangeAddress; |
||||||
|
} |
||||||
|
|
||||||
|
private String getCellVal(Cell cell) { |
||||||
|
String val = null; |
||||||
|
try { |
||||||
|
val = cell.getStringCellValue(); |
||||||
|
}catch (Exception e){ |
||||||
|
System.out.printf("读取单元格内容失败:行%d 列%d %n", (cell.getRowIndex() + 1), (cell.getColumnIndex() + 1)); |
||||||
|
} |
||||||
|
return val; |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue