用 EasyExcel 实现 Excel 的导入导出

目录

一、引入依赖

二、创建 Excel 对应的实体类

三、导出数据为 Excel

四、读取 Excel 中的数据

五、通过 Web 导出数据为 Excel 文件

六、通过 Web 来导出 Excel 中的数据

七、总结


        在各种系统当中,Excel 的导入与导出功能是最为基础的功能,无论是商品信息的导入、工资的导出,实在太常见了。在 Java 项目中,可以完成 Excel 导入导出的类库有很多种,这里来介绍一下 阿里巴巴 开源的 EasyExcel ,该库的特点如下:

Java解析、生成Excel比较有名的框架有Apache poi、jxl。但他们都存在一个严重的问题就是非常的耗内存,poi有一套SAX模式的API可以一定程度的解决一些内存溢出的问题,但POI还是有一些缺陷,比如07版Excel解压缩以及解压后存储都是在内存中完成的,内存消耗依然很大。easyexcel重写了poi对07版Excel的解析,能够原本一个3M的excel用POI sax依然需要100M左右内存降低到几M,并且再大的excel不会出现内存溢出,03版依赖POI的sax模式。在上层做了模型转换的封装,让使用者更加简单方便

https://github.com/alibaba/easyexcel

        从上面可以看出,EasyExcel 占用内存方面比 Apache poi、jxl 要小很多,在使用上也较为简单。本篇文章通过简单的实例,来快速的上手 EasyExcel 的使用。下面给一个 EasyExcel 内存使用图,该图来自 EasyExcel 的官网,如下图。

用 EasyExcel 实现 Excel 的导入导出_第1张图片

一、引入依赖

        使用 EasyExcel 的第一步就是引入依赖。我们创建一个 SpringBoot 的项目,然后在 pom.xml 文件中引入 EasyExcel 的依赖,依赖如下:




    com.alibaba
    easyexcel
    2.1.1

        引入依赖后就可以按照套路来使用 EasyExcel 了。

二、创建 Excel 对应的实体类

        在各种系统中,导入导出的数据一般都是通过 Excel 来完成的,在实现具体的导入导出功能前,首先是确定要导入导出哪些数据,也就是确定 Excel 文件的数据格式。在实际的导入导出功能中,数据一般是通过查询数据库获取的,这部分功能由程序员自己来完成实现,而 Excel 的导入导出实际是由类库帮助我们完成。

        同样的,对于导入导出功能,我们需要将我们指定的数据格式告诉导入导出 Excel 的类库。那么,首先需要建立一个和 Excel 表格对应的数据格式实体类。这里来写一个简单的学生类。


@Data
public class Student {

    /**
     * 姓名
     */
    @ExcelProperty("姓名")
    private String name;

    /**
     * 性别
     */
    @ExcelProperty("性别")
    private Integer gender;

    /**
     * 电话
     */
    @ExcelProperty("电话")
    private String phone;
}

        在上面定义的实体类中通过 @ExcelProperty 注解来指定属性对应的 Excel 中列的名称。EasyExcel 还提供了类似设置 列宽、列高 等注解,比如:

@ContentRowHeight(10)
@HeadRowHeight(20)
@ColumnWidth(25)

        具体各种注解的用法及细节可以参考官网文档进行学习和了解。

三、导出数据为 Excel

        有了与 Excel 对应的实体类后,我们先来完成一个导出 Excel 的功能。这里,我们创建一个测试用例,来模拟将数据导出为 Excel 的过程。

@Test
void Export()
{
    List stuList = new ArrayList<>();

    for (int i = 0; i < 10; i ++) {
        Student student = new Student();

        student.setName("stu" + i);
        student.setGender(i % 2);
        student.setPhone("1391234432" + i);

        stuList.add(student);
    }

    String exportName = "./export.xlsx";

    EasyExcel.write(exportName, Student.class)
            .sheet("学生信息")
            .doWrite(stuList);
}

        在上面的代码中,通过循环创建了 10 条数据保存在了 List 集合当中,然后将 List 集合导出到 Excel 中。导出的 Excel 结果如下图所示:

用 EasyExcel 实现 Excel 的导入导出_第2张图片

        在实际的项目中,导出的数据通常来自于数据库,这里为了 Demo 的简洁,通过 List 集合来进行模拟,但是实质是相同的。

四、读取 Excel 中的数据

        在读取 Excel 的内容时,EasyExcel 类库为我们提供了同步读取,或者创建监听器的方式进行读取。同步读取的方式较为简单,但是官方并不建议使用,因为会将大量的数据放入内存。

        官网使用监听器的方式进行 Excel 文件的读取,首先需要创建一个继承 AnalysisEventListener 类的监听器类,其中会实现几个方法,比如 invoke 和 doAfterAllAnalysed 。invoke 方法是每一条数据解析都会调用该方法,在所有的数据都解析完成了会调用 doAfterAllAnalysed 方法

        我们的 Demo 只完成简单的输出,并不实际写入数据库,因此比较简单。具体代码如下:

public class ExcelListener extends AnalysisEventListener {
    @Override
    public void invoke(Student student, AnalysisContext analysisContext) {
        System.out.println(student);
    }

    @Override
    public void invokeHead(Map headMap, AnalysisContext context) {
        System.out.println("表头信息" + headMap);
    }

    @Override
    public void doAfterAllAnalysed(AnalysisContext analysisContext) {

    }
}

        编写一个测试用例,使用 EasyExcel 来读取 Excel 的内容,代码如下:


@Test
void Import()
{
    String importName = "./export.xlsx";

    EasyExcel.read(importName, Student.class, new ExcelListener())
            .sheet().doRead();
}

        执行上面的测试用例,输出结果如下:

表头信息{0=姓名, 1=性别, 2=电话}
Student(name=stu0, gender=0, phone=13912344320)
Student(name=stu1, gender=1, phone=13912344321)
Student(name=stu2, gender=0, phone=13912344322)
Student(name=stu3, gender=1, phone=13912344323)
Student(name=stu4, gender=0, phone=13912344324)
Student(name=stu5, gender=1, phone=13912344325)
Student(name=stu6, gender=0, phone=13912344326)
Student(name=stu7, gender=1, phone=13912344327)
Student(name=stu8, gender=0, phone=13912344328)
Student(name=stu9, gender=1, phone=13912344329)

        上面的输出结果就是读入我们前面导出的数据。

五、通过 Web 导出数据为 Excel 文件

        在项目中,通常是通过一个页面来完成数据导入,我们可以写一个接口来供前端进行使用。通过 Web 页面导入 Excel 需要先上传 Excel 文件,然后再进行导入,上传 Excel 文件就是 Spring MVC 的知识了,使用 MultipartFile 即可完成。代码如下:

@RestController("")
public class DemoController {

    @PostMapping("importStudent")
    public String importData(MultipartFile file) {
        try {
            EasyExcel.read(file.getInputStream(), Student.class, new ExcelListener())
                    .sheet()
                    .doRead();
        } catch (IOException e) {
            e.printStackTrace();
        }

        return "Success";
    }
}

        启动该项目,然后通过 PostMan 来导入前面的 Excel 文件,如下图:

用 EasyExcel 实现 Excel 的导入导出_第3张图片

        上图中,KEY 的类型选择为 “File”,在控制台中输出结果如下:

表头信息{0=姓名, 1=性别, 2=电话}
Student(name=stu0, gender=0, phone=13912344320)
Student(name=stu1, gender=1, phone=13912344321)
Student(name=stu2, gender=0, phone=13912344322)
Student(name=stu3, gender=1, phone=13912344323)
Student(name=stu4, gender=0, phone=13912344324)
Student(name=stu5, gender=1, phone=13912344325)
Student(name=stu6, gender=0, phone=13912344326)
Student(name=stu7, gender=1, phone=13912344327)
Student(name=stu8, gender=0, phone=13912344328)
Student(name=stu9, gender=1, phone=13912344329)
Student(name=stu0, gender=0, phone=13912344320)
Student(name=stu1, gender=1, phone=13912344321)
Student(name=stu2, gender=0, phone=13912344322)
Student(name=stu3, gender=1, phone=13912344323)
Student(name=stu4, gender=0, phone=13912344324)
Student(name=stu5, gender=1, phone=13912344325)
Student(name=stu6, gender=0, phone=13912344326)
Student(name=stu7, gender=1, phone=13912344327)
Student(name=stu8, gender=0, phone=13912344328)
Student(name=stu9, gender=1, phone=13912344329)

        这里沿用了前面导入的 Listener,这里补充一点,EasyExcel 的 Listener 并不被 Spring 进行管理,因此每次使用都需要 new 一个新的实例。那么,如果需要将数据写入数据库,需要通过构造方法,将 Dao 传入,而不能直接将 Dao 进行注入。

        当我们导入数据时,不需要逐条将数据写库,频繁的写库效率太低,也不要待数据全部读完再一次写库,如果读入的数量过多时,则会导致占用较大的内存。

        在 EasyExcel 官网中给出了一段 Demo 供大家参考,代码我直接复制过来,供大家阅读,代码如下:

// 有个很重要的点 DemoDataListener 不能被spring管理,要每次读取excel都要new,然后里面用到spring可以构造方法传进去
public class DemoDataListener extends AnalysisEventListener {
    private static final Logger LOGGER = LoggerFactory.getLogger(DemoDataListener.class);
    /**
     * 每隔5条存储数据库,实际使用中可以3000条,然后清理list ,方便内存回收
     */
    private static final int BATCH_COUNT = 3000;
    /**
     * 缓存的数据
     */
    private List list = new ArrayList<>(BATCH_COUNT);
    /**
     * 假设这个是一个DAO,当然有业务逻辑这个也可以是一个service。当然如果不用存储这个对象没用。
     */
    private DemoDAO demoDAO;

    public DemoDataListener() {
        // 这里是demo,所以随便new一个。实际使用如果到了spring,请使用下面的有参构造函数
        demoDAO = new DemoDAO();
    }

    /**
     * 如果使用了spring,请使用这个构造方法。每次创建Listener的时候需要把spring管理的类传进来
     *
     * @param demoDAO
     */
    public DemoDataListener(DemoDAO demoDAO) {
        this.demoDAO = demoDAO;
    }

    /**
     * 这个每一条数据解析都会来调用
     *
     * @param data    one row value. Is is same as {@link AnalysisContext#readRowHolder()}
     * @param context
     */
    @Override
    public void invoke(DemoData data, AnalysisContext context) {
        LOGGER.info("解析到一条数据:{}", JSON.toJSONString(data));
        list.add(data);
        // 达到BATCH_COUNT了,需要去存储一次数据库,防止数据几万条数据在内存,容易OOM
        if (list.size() >= BATCH_COUNT) {
            saveData();
            // 存储完成清理 list
            list = new ArrayList<>(BATCH_COUNT);
        }
    }

    /**
     * 所有数据解析完成了 都会来调用
     *
     * @param context
     */
    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        // 这里也要保存数据,确保最后遗留的数据也存储到数据库
        saveData();
        LOGGER.info("所有数据解析完成!");
    }

    /**
     * 加上存储数据库
     */
    private void saveData() {
        LOGGER.info("{}条数据,开始存储数据库!", list.size());
        demoDAO.save(list);
        LOGGER.info("存储数据库成功!");
    }
}

        代码中有很好的注释,在大家写监听器时,可以充分的进行参考。

六、通过 Web 来导出 Excel 中的数据

        同 Excel 的导入相同,实际的 Excel 导出也是通过 Web 页面进行导出的,代码如下:

@GetMapping("exportStudent")
public void exportData(HttpServletResponse response) {
    response.setContentType("application/vnd.ms-excel");
    response.setCharacterEncoding("utf-8");
    String fileName = "student";
    response.setHeader("Content-disposition", "attachment:filename=" + fileName + ".xlsx");

    List stuList = new ArrayList<>();

    // 实体替换为查询数据库
    for (int i = 0; i < 10; i ++) {
        Student student = new Student();

        student.setName("stu" + i);
        student.setGender(i % 2);
        student.setPhone("1391234432" + i);

        stuList.add(student);
    }

    // 调用方法进行写操作
    try {
        EasyExcel.write(response.getOutputStream(), Student.class)
                .sheet("学生信息").doWrite(stuList);
    } catch (IOException e) {
        e.printStackTrace();
    }
}

        代码基本上分为三步,准备导出数据,设置 Http 头,数据导出。这里通过浏览器来访问我们的接口地址,然后 Excel 文件则会顺利的导出。

七、总结

        EasyExcel 的使用还是比较简单,功能也比较多,它在导出 Excel 时可以进行合并单元格、设置复杂的表格格式、自定义样式等,还可以提前设置 Excel 的模板,将数据进行填充(填充效果如下图)。在读取 Excel 数据时,可以读取多个 Sheet 页,可以格式化数值、日期等格式。具体可以参考 EasyExcel 的手册和官网,地址如下:

https://alibaba-easyexcel.github.io/index.html

https://github.com/alibaba/easyexcel

用 EasyExcel 实现 Excel 的导入导出_第4张图片

图片

你可能感兴趣的:(后端功能,EasyExcel类库,Excel导入,Excel导出,Excel导入导出)