easyexcel 自定义合并单元格策略

EasyExcel 是一个基于Java的简单、省内存的读写Excel的开源项目，它提供了丰富的API，可以方便地读取和写入Excel文件。为了实现自定义合并单元格策略，我们可以使用 EasyExcel 提供的 SheetWriteHandler 接口，它可以在写入 Excel 文件时对每个 sheet 进行自定义操作。

具体实现步骤如下：

1. 创建一个实现 SheetWriteHandler 接口的类，实现合并单元格的逻辑，例如：

public class MergeStrategy implements SheetWriteHandler {
    private Map<Integer, List<MergeModel>> mergeMap = new HashMap<>();

    @Override
    public void beforeSheetCreate(String sheetName, int index) {
        // do nothing
    }

    @Override
    public void afterSheetCreate(String sheetName, int index) {
        // do nothing
    }

    @Override
    public void afterRowCreate(Row row, int rowIndex) {
        // do nothing
    }

    @Override
    public void beforeRowDispose(Row row, int rowIndex) {
        // do nothing
    }

    @Override
    public void afterRowDispose(Row row, int rowIndex) {
        // do nothing
    }

    @Override
    public void beforeCellCreate(CellData cellData, Cell cell, int rowIndex, int columnIndex) {
        // do nothing
    }

    @Override
    public void afterCellCreate(CellData cellData, Cell cell, int rowIndex, int columnIndex) {
        // do nothing
    }

    @Override
    public void afterCellDispose(CellData cellData, Cell cell, int rowIndex, int columnIndex) {
        // do nothing
    }

    @Override
    public void beforeCellDispose(CellData cellData, Cell cell, int rowIndex, int columnIndex) {
        // do nothing
    }

    @Override
    public void afterSheetDispose(String sheetName, int index) {
        // merge cells
        Sheet sheet = EasyExcel.writerSheet(index, sheetName).build().getSheet();
        mergeMap.forEach((rowIndex, mergeList) -> {
            for (MergeModel mergeModel : mergeList) {
                if (mergeModel.getEndRow() > rowIndex) {
                    sheet.merge(rowIndex, mergeModel.getEndRow(), mergeModel.getStartCol(), mergeModel.getEndCol());
                }
            }
        });
    }

    public void addMergedRegion(int startRow, int endRow, int startCol, int endCol) {
        List<MergeModel> mergeList = mergeMap.getOrDefault(startRow, new ArrayList<>());
        mergeList.add(new MergeModel(startCol, endCol, endRow));
        mergeMap.put(startRow, mergeList);
    }

    private static class MergeModel {
        private int startCol;
        private int endCol;
        private int endRow;

        public MergeModel(int startCol, int endCol, int endRow) {
            this.startCol = startCol;
            this.endCol = endCol;
            this.endRow = endRow;
        }

        public int getStartCol() {
            return startCol;
        }

        public int getEndCol() {
            return endCol;
        }

        public int getEndRow() {
            return endRow;
        }
    }
}

2. 在需要写入 Excel 文件的地方，创建一个 ExcelWriter 对象，并设置自定义的 SheetWriteHandler，例如：

public static void main(String[] args) {
    // 创建 ExcelWriter 对象
    String fileName = "test.xlsx";
    ExcelWriter excelWriter = EasyExcel.write(fileName).build();

    // 创建 MergeStrategy 对象
    MergeStrategy mergeStrategy = new MergeStrategy();

    // 写入数据到 Excel 文件，并调用 MergeStrategy 对象的 addMergedRegion 方法进行单元格合并
    List<List<Object>> data = getData();
    WriteSheet writeSheet = EasyExcel.writerSheet("sheet1").build();
    for (int i = 0; i < data.size(); i++) {
        List<Object> rowData = data.get(i);
        for (int j = 0; j < rowData.size(); j++) {
            Object cellData = rowData.get(j);
            if (cellData instanceof String && ((String) cellData).startsWith("merge")) {
                mergeStrategy.addMergedRegion(i, i + 1, j, j + 1);
            }
        }
    }
    excelWriter.write(data, writeSheet).finish();

    // 关闭 ExcelWriter 对象
    excelWriter.finish();
}

在上述代码中，我们创建了一个 ExcelWriter 对象，并设置了一个 SheetWriteHandler 对象 MergeStrategy，然后调用 addMergedRegion 方法添加需要合并的单元格信息。在写入数据到 Excel 文件时，如果某个单元格需要合并，我们就调用 MergeStrategy 对象的 addMergedRegion 方法进行单元格合并。最后，调用 finish 方法关闭 ExcelWriter 对象。

这样，我们就可以自定义合并单元格策略了。