EasyExcel为单个Cell设置样式

EasyExcel是阿里巴巴对POI封装的一个库，号称解决了POI的OOM问题，而且在使用上也更方便一些

Github：

然而我在使用的时候发现仍是有不少坑，其中一个比较头疼的是对单个单元格样式的设置。EasyExcel提供了一个BaseRowModel做为每行数据的一个模型，而且其中有一个属性cellStyleMap表明每列样式的集合，原本我觉得这个只要在本身定义模型的时候，也把CellStyle定义进去就好了，然而，仍是我想多了……定义了CellStyle并无什么卵用，这是第一个蛋疼的地方

/** * Excel基础模型 * @author jipengfei */
public class BaseRowModel {

    /** * 每列样式 */
    private Map<Integer,CellStyle> cellStyleMap = new HashMap<Integer,CellStyle>();

    public void addStyle(Integer row, CellStyle cellStyle){
        cellStyleMap.put(row,cellStyle);
    }

    public CellStyle getStyle(Integer row){
        return cellStyleMap.get(row);
    }

    public Map<Integer, CellStyle> getCellStyleMap() {
        return cellStyleMap;
    }

    public void setCellStyleMap(Map<Integer, CellStyle> cellStyleMap) {
        this.cellStyleMap = cellStyleMap;
    }
}
复制代码

后来测试半天，才发现建立CellStyle时必须经过一个Workbook对象来建立，而这个Workbook不能随便新建一个对象完事儿，得用你当前写入的Workbook来建立对应的CellStyle样式才能起做用。然而……事情并无那么简单，通过我对源码的反复查看，EasyExcel生成excel表的步骤是用一个ExcelWriter来写入数据，并无提供获取Workbook的方法，不知道什么缘由让阿里巴巴不提供这样一个接口……这是第二个蛋疼的地方

既然没有提供接口，那就只能用反射来硬刚了，下面就直接上代码了

我这里是在开始写数据以前就将每张表的CellStyle与每张表关联起来，再在后面的handler中获取到这个CellStyle进行设置

package edu.uddp.util;

import com.alibaba.excel.EasyExcelFactory;
import com.alibaba.excel.ExcelReader;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.context.WriteContext;
import com.alibaba.excel.event.WriteHandler;
import com.alibaba.excel.metadata.BaseRowModel;
import com.alibaba.excel.metadata.Sheet;
import com.alibaba.excel.support.ExcelTypeEnum;
import com.alibaba.excel.write.ExcelBuilderImpl;
import com.sun.corba.se.spi.orbutil.threadpool.Work;
import edu.uddp.enums.CellStyleEnum;
import edu.uddp.model.SignExcelRow;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.streaming.SXSSFSheet;

import java.io.*;
import java.lang.reflect.Field;
import java.util.*;

/** * 生成Excel表 * * @author Juzi * @date 2018/12/23 12:37 * Blog https://juzibiji.top */
public class ExcelUtil {

    private static Map<Workbook, Map<String, CellStyle>> cellStyles = new HashMap<>();

    /** * 使用java对象模型建立excel，并使用handler * 生成Excel格式为xlsx * * @param path Excel生成路径 * @param headLineMun 表头占几行 * @param data 传入的键值对数据（key为sheet名，value为sheet数据） * @param handler 自定义的EasyExcel Handler，不使用传入null便可 * @param columnWidthMap 每列宽度 * @throws IOException */
    public static void createExcelWithModelAndHandler( String path, int headLineMun, Map<String, List<? extends BaseRowModel>> data, WriteHandler handler, Map<Integer, Integer> columnWidthMap, List<CellStyleEnum> cellStyleEnums) throws IOException {

        File file = new File(path);
        if (!file.getParentFile().exists()) {
            file.getParentFile().mkdirs();
        }

        OutputStream out = new FileOutputStream(path);
        // ExcelWriter用于导出Excel
        ExcelWriter writer = EasyExcelFactory.getWriterWithTempAndHandler(null, out,
                ExcelTypeEnum.XLSX, true, handler);

        // 构造单元格样式
        Workbook workbook = getWorkbook(writer);
        cellStyles.put(workbook, createCellStyle(workbook, cellStyleEnums));

        // sheet的序号，从1开始
        int i = 1;
        // 遍历传入的sheet名和sheet数据来建立sheet
        for (Map.Entry<String, List<? extends BaseRowModel>> entry : data.entrySet()) {
            Sheet sheet = new Sheet(i, headLineMun, entry.getValue().get(0).getClass(), entry.getKey(), null);
            sheet.setColumnWidthMap(columnWidthMap);
            writer.write(entry.getValue(), sheet);
            i++;
        }
        // 必需要调用finish()，不然数据不会写入文件
        writer.finish();
        out.close();
    }


    /** * **获取workbook** * 由于EasyExcel这个库设计的缘由 * 只能使用反射获取workbook * * @param writer * @return */
    private static Workbook getWorkbook(ExcelWriter writer) {
        Workbook workbook = null;
        try {
            Class<?> clazz1 = Class.forName("com.alibaba.excel.ExcelWriter");
            Field[] fs = clazz1.getDeclaredFields();
            for (Field field : fs) {
                // 要设置属性可达，否则会抛出IllegalAccessException异常
                field.setAccessible(true);
                if ("excelBuilder".equals(field.getName())) {
                    ExcelBuilderImpl excelBuilder = (ExcelBuilderImpl) field.get(writer);
                    Class<?> clazz2 = Class.forName("com.alibaba.excel.write.ExcelBuilderImpl");
                    Field[] fs2 = clazz2.getDeclaredFields();
                    for (Field field2 : fs2) {
                        field2.setAccessible(true);
                        if ("context".equals(field2.getName())) {
                            WriteContext context = (WriteContext) field2.get(excelBuilder);
                            workbook = context.getWorkbook();
                        }
                    }
                }
            }
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
        return workbook;
    }

    public static Map createCellStyle(Workbook workbook, List<CellStyleEnum> cellStyleEnums) {
        Map<String, CellStyle> map = new HashMap<>();
        for (CellStyleEnum cellStyleEnum : cellStyleEnums) {
            if (cellStyleEnum.getNo() == 1) {
                CellStyle cellStyle = workbook.createCellStyle();
                cellStyle.setBorderBottom(BorderStyle.THIN); //下边框
                cellStyle.setBorderLeft(BorderStyle.THIN);//左边框
                cellStyle.setBorderTop(BorderStyle.THIN);//上边框
                cellStyle.setBorderRight(BorderStyle.THIN);//右边框
                cellStyle.setAlignment(HorizontalAlignment.CENTER);
                map.put(cellStyleEnum.getName(), cellStyle);
            } else if (cellStyleEnum.getNo() == 2) {
                CellStyle cellStyle = workbook.createCellStyle();
                cellStyle.setBorderBottom(BorderStyle.THIN); //下边框
                cellStyle.setBorderLeft(BorderStyle.THIN);//左边框
                cellStyle.setBorderTop(BorderStyle.THIN);//上边框
                cellStyle.setBorderRight(BorderStyle.THIN);//右边框
                cellStyle.setAlignment(HorizontalAlignment.CENTER);
                cellStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
                cellStyle.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex());
                map.put(cellStyleEnum.getName(), cellStyle);
            }
        }
        return map;
    }

    public static Map<Workbook, Map<String, CellStyle>> getCellStyles() {
        return cellStyles;
    }
}
复制代码

EasyExcel提供了一个WriteHandler，咱们实现这个接口，就能够在每一个单元格写入以后或者每行写入以前来进行拦截（这个handler设计的也很蛋疼），并注入咱们本身的业务逻辑（设置单元格样式）。

package edu.uddp.handler;

import com.alibaba.excel.event.WriteHandler;
import edu.uddp.util.ExcelUtil;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFSheet;

import java.util.Map;

/** * 第三方库EasyExcel的Handler * 教师端生成签到历史表Excel时 * 将未签到学生进行特殊标识 * * @author Juzi * @since 2018/12/22 22:07 * Blog https://juzibiji.top */
public class SignRecordExcelHandler implements WriteHandler {
    @Override
    public void sheet(int i, Sheet sheet) {

    }

    @Override
    public void row(int i, Row row) {

    }

    @Override
    public void cell(int i, Cell cell) {
        // 获取当前workbook对应的CellStyle集合
        Map<String, CellStyle> cellStyleMap = ExcelUtil.getCellStyles().get(cell.getSheet().getWorkbook());

        // 从第二行开始设置格式，第一行是表头
        if (cell.getRowIndex() > 0) {
            if (i == 7 && "未签到".equals(cell.getStringCellValue())) {
                // 该生未签到
                for (int j = 0; j < 8; j++) {
                    cell.getRow().getCell(j).setCellStyle(cellStyleMap.get("未签到"));
                }
            } else if (i == 8 && "已签到".equals(cell.getRow().getCell(7).getStringCellValue())) {
                // 该生已签到
                for (int j = 0; j < 9; j++) {
                    cell.getRow().getCell(j).setCellStyle(cellStyleMap.get("已签到"));
                }
            }else if(i == 8 && "未签到".equals(cell.getRow().getCell(7).getStringCellValue())){
                cell.setCellStyle(cellStyleMap.get("已签到"));
            }
        }
    }
}

复制代码

上面有一些简单的逻辑处理，就不一一介绍了。

若文章有任何问题，欢迎留言指出——做者博客：桔子笔记