【二十五】springboot使用EasyExcel和线程池实现多线程导入Excel数据

小z666

发布于 2024-06-21 17:44:34

630

发布于 2024-06-21 17:44:34

文章被收录于专栏：javajava

在公司开发时，遇到一个很常见的导入功能的需求，需要导入Excel文件，由此想到了阿里巴巴的EasyExcel这个方便的工具，当客户给我说需要支持大数据量导入时，我想到了使用线程池来多线程处理导入数据库这个操作。由此本章记录一下这次操作。

首先，整体的大概流程差不多是这个样子：

效果：

真实项目上是公司封装的RPC框架，这次demo直接使用的前面的整合mybatis-plus的demo来写的。

首先看下目录结构：

图片中框选部分就是本章需要修改到的或者新创建的文件，后面一一解读，下面的描述可以对标这个图片中的类名。

然后介绍下本章需要做的准备工作：

数据库表
修改application配置文件，修改tomcat的最大文件上传限制（否则excel文件太大，上传会报错）
开启mybatis-plus的批量插入功能，mybatis-plus默认只有insert这个单条插入功能（若自己的项目不使用这个，则不需要，这只是我的demo上没有批量插入方法）
创建excel多线程导入接口所需的各个类

一、准备数据库和Excel文件

二、引入所需依赖

三、修改配置文件，修改文件大小默认限制

四、开启mybatis-plus的批量保存功能

五、创建所需工具类

六、创建业务各层代码

七、创建easyExcel事件监听器

八、创建自定义线程类

九、测试单线程和多线程处理的效率

一、准备数据库和Excel文件

CREATE TABLE `deadman` (
  `uid` varchar(255) CHARACTER SET utf8 COLLATE utf8_bin NOT NULL COMMENT '主键',
  `idCard` varchar(255) CHARACTER SET utf8 COLLATE utf8_bin DEFAULT NULL COMMENT '身份证号',
  `userName` varchar(255) CHARACTER SET utf8 COLLATE utf8_bin DEFAULT NULL COMMENT '死者姓名',
  `sex` varchar(255) CHARACTER SET utf8 COLLATE utf8_bin DEFAULT NULL COMMENT '死者性别',
  `age` varchar(255) CHARACTER SET utf8 COLLATE utf8_bin DEFAULT NULL COMMENT '死者年龄',
  `reason` varchar(255) CHARACTER SET utf8 COLLATE utf8_bin DEFAULT NULL COMMENT '死因',
  `house` varchar(255) CHARACTER SET utf8 COLLATE utf8_bin DEFAULT NULL COMMENT '安排地狱层数',
  PRIMARY KEY (`uid`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb3 COLLATE=utf8_bin;

二、引入所需依赖

主要是导入easyExcel所涉及的依赖，如下：

三、修改配置文件，修改文件大小默认限制

这个必须修改，不然后面接口都进不去，默认tomcat服务器是限制了上传文件的大小的。

spring:
  servlet:
    multipart:
      max-file-size: 100MB
      max-request-size: 100MB

server:
  tomcat:
    max-swallow-size: 100MB

四、开启mybatis-plus的批量保存功能

这次demo整合的rpc框架是Mybatis-plus，所以找了一个方法去实现批量保存，如下：

1、新建一个SpiceSqlInjector类

PS：注意框选部分类名。

2、创建SpiceBaseMapper接口类

PS：本来是业务的接口层去继承BaseMapper，此处又它去继承BaseMapper，然后加如上图这个方法，注意名称必须是这个否则后面调用该方法会报错。

3、创建业务所需的Mapper层

继承刚才自定义的接口类。

由此就可以通过注入DeadManMapper调用上面的方法实现批量插入了。

如下：

五、创建所需工具类

在通过EasyExcel导入时，会涉及到一些数据类型不满足入参之类的，本次demo涉及到如下工具类。

1、MultipartFileToFileUtils

将传入的MultipartFile类型转为File类型，Controller接收到的是MultipartFile类型，EasyExcel.read方法所需要的是File类型。

2、SpringJobBeanFactory

在监听器类注入mappe时会报空指针，使用如下工具类继承ApplicationContextAware，获取bean对象。

六、创建业务各层代码

1、controller层

2、映射数据库的实体类

关于mybatis-plus相关的注解可以去看看前面的文章。

3、easyExcel的实体类

index指的是excel表上面的行编号，例如

4、service层

5、service的实现类层

上面的方法是多线程处理的事件监听器，下面的是单线程的事件监听器，后面针对两种方式都做一下对比。

七、创建easyExcel事件监听器

1、单线程的事件监听器

解析：该监听器继承自AnalysisEventListener类，泛型指定为上面easyExcel指定的实体类对象类。重写该类的两个方法：invoke和doAfterAllAnalysed。 invoke（）：该方法会从excel表的第二行开始读取数据。 doAfterAllAnalysed（）：当invoke将excel的数据全部解析完后，会执行该方法，所以在该类进行数据的入库即可。

此时，单线程的导入已经完成，我们最后再对比测试两种监听器的效率，现在创建多线程事件监听器。

2、多线程的事件监听器

解析：同样是重写上面说的那两个方法。invoke方法的处理不变，修改doAfterAllAnalysed方法的处理，在该方法通过创建线程池的方法，将创建的线程任务提交到线程池，让线程池进行多线程任务的执行，从而实现多线程执行导入操作。流程：创建线程池——》计算每个线程需要处理的数据——》创建CountDownLatch对象（保证最后每个线程会回到主线程）——》循环线程数量并提交线程任务到线程池——》执行CountDownLatch对象的await方法，让当前线程处于等待状态，等待CountDownLatch减少为1后会唤醒当前线程——》每个线程处理自己的数据并在处理完后执行CountDownLatch对象的countDown方法，让CountDownLatch对象的值减一——》当CountDownLatch的值为0时（说明线程池类的线程任务以及执行完成），执行主线程的代码——》关闭线程池