easyExcel生成excel并导出自定义样式(三)指定列字体样式

public class ExcelRowColorHandler implements CellWriteHandler {

    private final Integer columnIndex;
    private final Set<Integer> redRowIndex;
    private final Set<Integer> yellowRowIndex;
    private final Set<Integer> greenRowIndex;

    public ExcelRowColorHandler(Integer columnIndex, Set<Integer> redRowIndex, Set<Integer> yellowRowIndex, Set<Integer> greenRowIndex) {
        this.columnIndex = columnIndex;
        this.redRowIndex = redRowIndex;
        this.yellowRowIndex = yellowRowIndex;
        this.greenRowIndex = greenRowIndex;
    }

    @Override
    public void beforeCellCreate(CellWriteHandlerContext context) {
        // Row in source code context to be optimized: supports parameter input to obtain the number of
        // occurrences of the required field and set the background color
        // to do：（row.getCell(columnIndex)lose efficacy）
    }

    @Override
    public void afterCellDispose(CellWriteHandlerContext context) {
        if (BooleanUtils.isNotTrue(context.getHead())) {
            Integer currentRowIndex = context.getRowIndex();
            Integer currentColumnIndex = context.getColumnIndex();
            //最后一列，（备注信息）设置颜色
            if (currentColumnIndex.equals(columnIndex - 1)) {
                WriteCellData<?> cellData = context.getFirstCellData();
                WriteCellStyle writeCellStyle = cellData.getOrCreateStyle();
                //根据数据内容，设置不同字体颜色
                if (!Objects.isNull(redRowIndex) && redRowIndex.contains(currentRowIndex)) {
                    WriteFont font = new WriteFont();
                    font.setColor(IndexedColors.RED.getIndex());
                    writeCellStyle.setWriteFont(font);
                } else if (!Objects.isNull(yellowRowIndex) && yellowRowIndex.contains(currentRowIndex)) {
                    WriteFont font = new WriteFont();
                    font.setColor(IndexedColors.YELLOW.getIndex());
                    writeCellStyle.setWriteFont(font);
                } else if (!Objects.isNull(greenRowIndex) && greenRowIndex.contains(currentRowIndex)) {
                    WriteFont font = new WriteFont();
                    font.setColor(IndexedColors.GREEN.getIndex());
                    writeCellStyle.setWriteFont(font);
                }
            }
        }
    }

}