forked from luyan/epmet-cloud-lingshan
jianjun
4 years ago
9 changed files with 8 additions and 223 deletions
@ -1,213 +0,0 @@ |
|||
package com.epmet.commons.tools.utils.poi.excel.handler; |
|||
|
|||
/** |
|||
* desc: |
|||
* |
|||
* @author: LiuJanJun |
|||
* @date: 2022/4/26 1:36 下午 |
|||
* @version: 1.0 |
|||
*/ |
|||
|
|||
import com.alibaba.excel.enums.CellDataTypeEnum; |
|||
import com.alibaba.excel.metadata.Head; |
|||
import com.alibaba.excel.metadata.data.CellData; |
|||
import com.alibaba.excel.metadata.data.WriteCellData; |
|||
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; |
|||
import java.util.Objects; |
|||
import java.util.stream.IntStream; |
|||
|
|||
/** |
|||
* 相同值合并策略 |
|||
*/ |
|||
|
|||
public class ExcelMergeStrategy implements CellWriteHandler { |
|||
|
|||
/** |
|||
* 合并起始行 |
|||
*/ |
|||
|
|||
private int mergeRowIndex; |
|||
/** |
|||
* 多少行合并一次 |
|||
*/ |
|||
|
|||
private int eachRow; |
|||
|
|||
/** |
|||
* 合并字段的下标 |
|||
*/ |
|||
|
|||
private int[] mergeColumnIndex; |
|||
|
|||
public ExcelMergeStrategy(int mergeRowIndex, int[] mergeColumnIndex, int eachRow) { |
|||
|
|||
if (mergeRowIndex < 0) { |
|||
|
|||
throw new IllegalArgumentException("mergeRowIndex must be greater than 0"); |
|||
|
|||
} |
|||
|
|||
if (eachRow < 0) { |
|||
|
|||
throw new IllegalArgumentException("eachRow must be greater than 0"); |
|||
|
|||
} |
|||
|
|||
this.mergeRowIndex = mergeRowIndex; |
|||
|
|||
this.mergeColumnIndex = mergeColumnIndex; |
|||
|
|||
this.eachRow = eachRow; |
|||
|
|||
} |
|||
|
|||
@Override |
|||
|
|||
public void beforeCellCreate(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, Row row, Head head, Integer columnIndex, Integer relativeRowIndex, Boolean isHead) { |
|||
} |
|||
|
|||
@Override |
|||
|
|||
public void afterCellCreate(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, Cell cell, Head head, Integer relativeRowIndex, Boolean isHead) { |
|||
} |
|||
|
|||
// @Override
|
|||
public void afterCellDataConverted(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, CellData cellData, Cell cell, Head head, Integer relativeRowIndex, Boolean isHead) { |
|||
|
|||
int curRowIndex = cell.getRowIndex(); |
|||
|
|||
//当前列
|
|||
|
|||
int curColIndex = cell.getColumnIndex(); |
|||
|
|||
//合并条件：
|
|||
|
|||
//1.当前行>合并起始行，默认标题行（0）不参加合并
|
|||
|
|||
//2.间隔行（eachRow）的上下两条不参加合并
|
|||
|
|||
//2.1间隔行（eachRow）==0时，不设置间隔
|
|||
|
|||
if (isMerge(curRowIndex)) { |
|||
|
|||
IntStream.range(0, mergeColumnIndex.length).anyMatch(i -> { |
|||
|
|||
if (curColIndex == mergeColumnIndex[i]) { |
|||
|
|||
mergeWithPrevRow(writeSheetHolder, cellData, cell, curRowIndex, curColIndex); |
|||
|
|||
return true; |
|||
|
|||
} |
|||
|
|||
return false; |
|||
|
|||
}); |
|||
|
|||
} |
|||
|
|||
} |
|||
|
|||
@Override |
|||
public void afterCellDispose(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, List<WriteCellData<?>> cellDataList, Cell cell, Head head, Integer relativeRowIndex, Boolean isHead) { |
|||
} |
|||
|
|||
/** |
|||
* 判断是否合并 |
|||
* <p> |
|||
* 1.当前位置必须大于开始位置：curRowIndex > mergeRowIndex |
|||
* <p> |
|||
* 2.根据eachRow 判断数据分割的间隔 |
|||
* <p> |
|||
* 2.1如果根据eachRow=0，默认不合并 |
|||
* <p> |
|||
* 2.2如果1如果根据eachRow>0,分割后的第一条数据不会与之前的合并：（curRowIndex-mergeRowIndex）%eachRow==0 |
|||
* |
|||
* @return |
|||
*/ |
|||
|
|||
private boolean isMerge(Integer curRowIndex) { |
|||
if ((curRowIndex > mergeRowIndex) && eachRow > 0) { |
|||
|
|||
if ((curRowIndex - mergeRowIndex) % eachRow == 0) { |
|||
|
|||
return false; |
|||
|
|||
} |
|||
|
|||
return true; |
|||
|
|||
} |
|||
return false; |
|||
|
|||
} |
|||
|
|||
private void mergeWithPrevRow(WriteSheetHolder writeSheetHolder, CellData cellData, Cell cell, int curRowIndex, int curColIndex) { |
|||
|
|||
//获取当前行的当前列的数据和上一行的当前列列数据，通过上一行数据是否相同进行合并
|
|||
|
|||
Object curData = cellData.getType() == CellDataTypeEnum.STRING ? cellData.getStringValue() : cellData.getNumberValue(); |
|||
|
|||
Cell preCell = cell.getSheet().getRow(curRowIndex - 1).getCell(curColIndex); |
|||
|
|||
Object preData = preCell.getCellTypeEnum() == CellType.STRING ? preCell.getStringCellValue() : |
|||
|
|||
preCell.getNumericCellValue(); |
|||
|
|||
// 比较当前行的第一列的单元格与上一行是否相同，相同合并当前单元格与上一行
|
|||
|
|||
if (Objects.equals(curData, preData)) { |
|||
|
|||
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); |
|||
|
|||
} |
|||
|
|||
} |
|||
|
|||
} |
|||
|
|||
} |
|||
|
|||
@ -1,4 +1,4 @@ |
|||
package com.epmet.commons.tools.utils.poi.excel; |
|||
package com.epmet.commons.tools.utils.poi.excel.handler; |
|||
|
|||
/** |
|||
* desc:easyExcel 冻结标题 |
|||
Loading…
Reference in new issue