EasyExcel导出工具类

目录

工具类

头部实体类(要和工具类在同一个module或项目下)

日期转换器


工具类

java 复制代码
/**
 * 导出Excel工具类
 */
public class EasyExcelUtil<T> {

    /**
     * 单sheet(Map写入)
     * @param response 响应对象
     * @param headList 头部集合
     * @param dataList 数据集合
     */
    public static void write(HttpServletResponse response, List<ExcelHead> headList, List<Map<String, Object>> dataList) throws IOException {
        ExcelWriterBuilder writerBuilder = EasyExcel.write();
        writerBuilder.file(response.getOutputStream());
        writerBuilder.excelType(ExcelTypeEnum.XLSX);
        //日期转换器
        TimestampStringConverter converter = new TimestampStringConverter();
        writerBuilder.registerConverter(converter).registerWriteHandler(new ColumnWidthStyleStrategy())
                .head(convertHead(headList)).sheet("sheet1")
                .doWrite(convertData(headList, dataList));
    }

    /**
     * 多sheet(Map写入)
     * @param response 响应对象
     * @param headMap 头部Map数据
     * @param dataMap 数据Map数据
     * @param sheetMap sheet Map数据
     */
    public static void multipleWrite(HttpServletResponse response
            , Map<String,List<ExcelHead>> headMap
            , Map<String,List<Map<String, Object>>> dataMap
            , Map<String,String> sheetMap) throws IOException {
        //日期转换器
        TimestampStringConverter converter = new TimestampStringConverter();
        ExcelWriter excelWriter = EasyExcel.write()
                .registerConverter(converter)
                .registerWriteHandler(new ColumnWidthStyleStrategy())
                .file(response.getOutputStream())
                .excelType(ExcelTypeEnum.XLSX).autoCloseStream(true).build();
        int i = 0;
        for (Map.Entry<String,List<ExcelHead>> entry : headMap.entrySet()) {
            WriteSheet writeSheet = EasyExcel
                    .writerSheet(i++, sheetMap.get(entry.getKey()))
                    .head(convertHead(entry.getValue())).build();
            excelWriter.write(convertData(entry.getValue(), dataMap.get(entry.getKey())), writeSheet);
        }
        excelWriter.finish();
    }

    /**
     * 实体写入
     * @param response 响应对象
     * @param sheetName sheet名称
     * @param c 实体类
     * @param list 实体数据
     */
    public static <T> void writeSheet(HttpServletResponse response, String sheetName, Class<T> c, List<T> list) throws IOException {
        EasyExcel.write(response.getOutputStream(), c).sheet(sheetName).doWrite(list);
    }

    /**
     * 读取并存储到实体
     * @param fileName 路径地址
     * @param sheetName sheet名称
     * @param c 实体类
     */
    public static <T> List<T> read(String fileName, String sheetName, Class c) {
        List<T> list = new ArrayList();
        EasyExcel.read(fileName, c, new ReadListener<T>() {
            @Override
            public void invoke(T o, AnalysisContext analysisContext) {
                list.add(o);
            }
            @Override
            public void doAfterAllAnalysed(AnalysisContext analysisContext) {

            }
        }).sheet(sheetName).doRead();
        return list;
    }

    /**
     * 读取并存储到实体
     * @param fileName 路径地址
     * @param sheetNo 指定sheet
     */
    public static Map<String,Object> readToMap(String fileName, Integer sheetNo) {
        Map<String,Object> result = new HashMap<>();
        List<Map<String,Object>> dataList = new ArrayList();
        //头部map
        Map<String,String> headMap = new HashMap<>();
        //头部拼音map
        Map<String,String> pinyinMap = new HashMap<>();
        EasyExcel.read(fileName,new AnalysisEventListener<Map<Integer, Object>>() {
            @Override
            public void invoke(Map<Integer, Object> data, AnalysisContext context) {
                Map<String,Object> map = new HashMap<>();
                for (Integer key : data.keySet()) {
                    if(key!=null && data.get(key)!=null) {
                        map.put("field_" + key.toString(), data.get(key));
                    }
                }
                dataList.add(map);
            }
            @Override
            public void doAfterAllAnalysed(AnalysisContext analysisContext) {
            }
            @Override
            public void invokeHead(Map<Integer, ReadCellData<?>> head, AnalysisContext context) {
                for (Integer key : head.keySet()) {
                    if(key!=null && head.get(key)!=null && StringUtils.isNotBlank(head.get(key).getStringValue())) {
                        headMap.put("field_" + key.toString(), head.get(key).getStringValue());
                        pinyinMap.put("field_" + key.toString(), Pinyin4jUtils.getPinYinHeadChar(head.get(key).getStringValue()));
                    }
                }
            }
        }).sheet(sheetNo).headRowNumber(1).doRead();
        result.put("headMap",headMap);
        result.put("pinyinMap",pinyinMap);
        result.put("dataList",dataList);
        result.put("count",dataList.size());
        return result;
    }

    /**
     * 读取表头并存储到实体
     * @param fileName 路径地址
     * @param sheetNo 指定sheet
     */
    public static Map<String,Object> readToMapHead(String fileName, Integer sheetNo) {
        Map<String,Object> result = new HashMap<>();
        //头部map
        Map<String,String> headMap = new HashMap<>();
        //头部拼音map
        Map<String,String> pinyinMap = new HashMap<>();
        EasyExcel.read(fileName,new AnalysisEventListener<Map<Integer, Object>>() {
            @Override
            public void invoke(Map<Integer, Object> data, AnalysisContext context) {
            }
            @Override
            public void doAfterAllAnalysed(AnalysisContext analysisContext) {
            }
            @Override
            public void invokeHead(Map<Integer, ReadCellData<?>> head, AnalysisContext context) {
                for (Integer key : head.keySet()) {
                    if(key!=null && head.get(key)!=null && StringUtils.isNotBlank(head.get(key).getStringValue())) {
                        headMap.put("field_" + key.toString(), head.get(key).getStringValue());
                        pinyinMap.put("field_" + key.toString(), Pinyin4jUtils.getPinYinHeadChar(head.get(key).getStringValue()));
                    }
                }
            }
        }).sheet(sheetNo).headRowNumber(1).doRead();
        result.put("headMap",headMap);
        result.put("pinyinMap",pinyinMap);
        return result;
    }

    /**
     * 头部转换
     * @param headList 头部集合
     */
    private static List<List<String>> convertHead(List<ExcelHead> headList) {
        List<List<String>> list = new ArrayList<>();
        for (ExcelHead head : headList) {
            list.add(Lists.newArrayList(head.getTitle()));
        }
        //沒有搞清楚head的参数为List<List<String>>,用List<String>就OK了
        return list;
    }

    /**
     * 数据转换
     * @param headList 头部集合
     * @param dataList 数据集合
     */
    private static List<List<Object>> convertData(List<ExcelHead> headList, List<Map<String, Object>> dataList) {
        List<List<Object>> result = new ArrayList();
        //对dataList转为easyExcel的数据格式
        for (Map<String, Object> data : dataList) {
            List<Object> row = new ArrayList();
            for (ExcelHead h : headList) {
                Object o = data.get(h.getFieldName());
                //需要对null的处理,比如age的null,要转为-1
                row.add(handler(o, h.getNullValue()));
            }
            result.add(row);
        }
        return result;
    }

    /**
     * 空值处理
     * @param o 数值
     * @param nullValue 空值置换
     */
    private static Object handler(Object o, Object nullValue) {
        return o != null ? o : nullValue;
    }
}

头部实体类(要和工具类在同一个module或项目下)

java 复制代码
/**
 * Excel头部实体
 */
public class ExcelHead<T> {

    private String fieldName;

    private String title;

    private T nullValue;

    public ExcelHead(String fieldName, String title) {
        this.fieldName = fieldName;
        this.title = title;
    }

    public ExcelHead(String fieldName, String title, T nullValue) {
        this.fieldName = fieldName;
        this.title = title;
        this.nullValue = nullValue;
    }

    public String getFieldName() {
        return fieldName;
    }

    public void setFieldName(String fieldName) {
        this.fieldName = fieldName;
    }

    public String getTitle() {
        return title;
    }

    public void setTitle(String title) {
        this.title = title;
    }

    public T getNullValue() {
        return nullValue;
    }

    public void setNullValue(T nullValue) {
        this.nullValue = nullValue;
    }
}

注意:真正导出表格的是ExcelWriterSheetBuilder类中的方法,前面只是封装,这个是真正导出用到的;这个类是EasyExcel自带的。

日期转换器

java 复制代码
/**
 * 日期转换器
 */
public class TimestampStringConverter implements Converter<Timestamp> {

    @Override
    public Class<?> supportJavaTypeKey() {
        return Timestamp.class;
    }

    @Override
    public CellDataTypeEnum supportExcelTypeKey() {
        return CellDataTypeEnum.STRING;
    }

    @Override
    public WriteCellData<?> convertToExcelData(Timestamp value, ExcelContentProperty contentProperty,
                                               GlobalConfiguration globalConfiguration) {
        WriteCellData cellData = new WriteCellData();
        String cellValue;
        if (contentProperty == null || contentProperty.getDateTimeFormatProperty() == null) {
            cellValue = DateUtils.format(value.toLocalDateTime(), null, globalConfiguration.getLocale());
        } else {
            cellValue = DateUtils.format(value.toLocalDateTime(), contentProperty.getDateTimeFormatProperty().getFormat(),
                    globalConfiguration.getLocale());
        }
        cellData.setType(CellDataTypeEnum.STRING);
        cellData.setStringValue(cellValue);
        cellData.setData(cellValue);
        return cellData;
    }
}
相关推荐
PythonFun19 分钟前
Python批量下载PPT模块并实现自动解压
开发语言·python·powerpoint
Death20022 分钟前
Qt 6 相比 Qt 5 的主要提升与更新
开发语言·c++·qt·交互·数据可视化
机器视觉知识推荐、就业指导25 分钟前
使用Qt实现实时数据动态绘制的折线图示例
开发语言·qt
快乐就好ya1 小时前
Java多线程
java·开发语言
IT学长编程1 小时前
计算机毕业设计 二手图书交易系统的设计与实现 Java实战项目 附源码+文档+视频讲解
java·spring boot·毕业设计·课程设计·毕业论文·计算机毕业设计选题·二手图书交易系统
CS_GaoMing2 小时前
Centos7 JDK 多版本管理与 Maven 构建问题和注意!
java·开发语言·maven·centos7·java多版本
艾伦~耶格尔2 小时前
Spring Boot 三层架构开发模式入门
java·spring boot·后端·架构·三层架构
man20172 小时前
基于spring boot的篮球论坛系统
java·spring boot·后端
2401_858120533 小时前
Spring Boot框架下的大学生就业招聘平台
java·开发语言
S hh3 小时前
【Linux】进程地址空间
java·linux·运维·服务器·学习