【Excel操作的白月光】「EasyExcel技术专题」摒除OOM！让你的Excel操作变得更加优雅和安全

<dependency>
  <groupId>com.alibaba</groupId>
  <artifactId>easyexcel</artifactId>
  <version>2.2.6</version>
</dependency>
复制代码

<!-- poi -->
 <dependency>
          <groupId>org.apache.poi</groupId>
          <artifactId>poi</artifactId>
          <version>3.17</version>
 </dependency>
 <dependency>
          <groupId>org.apache.poi</groupId>
          <artifactId>poi-ooxml</artifactId>
          <version>3.17</version>
 </dependency>
复制代码

public void testReadEntity() {
  // 被读取的文件绝对路径
  String fileName = "path/testDemo.xlsx";
  // 接收解析出的目标对象（Entity）指的是你的实体类
   List<Entity> entityList = new ArrayList<>();
  // 这里需要指定读用哪个class去读，然后读取第一个sheet文件流会自动关闭
  // excel中表的列要与对象的字段相对应
   EasyExcel.read(fileName, Entity.class, new AnalysisEventListener<Student>() {
    // 每解析一条数据都会调用该方法
    @Override
    public void invoke(Entity entity, AnalysisContext analysisContext) {
      System.out.println("解析一条Row行对象：" + JSON.toJSONString(entity));
      entityList.add(student);
  }
    // 解析完毕的回调方法
    @Override
    public void doAfterAllAnalysed(AnalysisContext analysisContext) {
        System.out.println("excel文件读取完毕！");
    }
  }).sheet().doRead();
}
// 通过Map作为整体的数据结构模型
public void readWithoutObj() {
  // 被读取的文件绝对路径
  String fileName = "path/testDemo.xlsx";
  // 接收结果集，为一个List列表，每个元素为一个map对象，key-value对为excel中每个列对应的值
  List<Map<Integer,String>> resultList = new ArrayList<>();
  EasyExcel.read(fileName, new AnalysisEventListener<Map<Integer,String>>() {
    @Override
    public void invoke(Map<Integer, String> map, AnalysisContext analysisContext) {
      System.out.println("解析到一条数据：" + JSON.toJSONString(map));
      resultList.add(map);
    }
    @Override
    public void doAfterAllAnalysed(AnalysisContext analysisContext) {
      System.out.println("excel文件解析完毕！" + JSON.toJSONString(resultList));
    }
  }).sheet().doRead();
}
复制代码

@Data
  public class TestEntity {
    /**
     * 从0开始，2代表强制读取第三个，
     * 不建议 index 和 name 同时用
     */
    @ExcelProperty(index = 2)
    private Double doubleData;
    /**
     * 用名字去匹配，这里需要注意，
     * 名字重复，会导致只有一个字段读取到数据
     */
    @ExcelProperty("字符串标题")
    private String string;
    @ExcelProperty("日期标题")
    private Date date;
  }
复制代码

@Data
  public class MultiHeaderEntity implements Serializable {
    @ExcelProperty(value = {"一层信息","二层1"})
    private Integer id;
    @ExcelProperty(value = {"一层信息","二层2"})
    private String name;
    @ExcelProperty(value = {"一层信息","二层4"})
    private String description;
    @ExcelProperty(value = {"一层信息","二层3"})
    private Date birthday;
  }
复制代码

@Data
  public class ConverterData {
    /**
     * 我自定义 转换器，不管数据库传过来什么 。我给他加上“自定义：”
     */
    @ExcelProperty(converter = CustomStringStringConverter.class)
    private String string;
    /**
     * 这里用string 去接日期才能格式化。我想接收年月日格式
     */
    @DateTimeFormat("yyyy年MM月dd日HH时mm分ss秒")
    private String date;
    /**
     * 我想接收百分比的数字
     */
    @NumberFormat("#.##%")
    private String doubleData;
  }
复制代码

@Data
public class DemoModel {
    private String attribute1;
    private Date attribute2;
    private Double attribute3;
}
复制代码

@Slf4j
public class DemoModelAnalysisListener extends AnalysisEventListener<DemoData> {
    /**
     * 每隔500条存储数据库，然后清理list ，方便内存回收
     */
    static final int BATCH_COUNT = 500;
    List<DemoModel> list = new ArrayList<>();
    /**
     * 这个也可以是一个service。当然如果不用存储这个对象没用。
     */
    private DemoService demoService;
    public DemoModelAnalysisListener(DemoService demoService) {
        this.demoService = demoService;
    }
    /**
     * 这个每一条数据解析都会来调用
     * @param data
     *            one row value. Is is same as {@link AnalysisContext#readRowHolder()}
     * @param context
     */
    @Override
    public void invoke(DemoData data, AnalysisContext context) {
        log.info("解析到一条数据:{}", JSON.toJSONString(data));
        list.add(data);
        // 达到BATCH_COUNT了，需要去存储一次数据库，
    // 防止数据几万条数据在内存，容易OOM
        if (list.size() >= BATCH_COUNT) {
            saveData();
        }
    }
  /**
     * 所有数据解析完成了 都会来调用
     * @param context
     */
    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        // 这里也要保存数据，确保最后遗留的数据也存储到数据库
        saveData();
        log.info("所有数据解析完成！");
    }
    /**
     * 加上存储数据库
     */
    private void saveData() {
        log.info("{}条数据，开始存储数据库！", list.size());
        demoService.insert(list);
                // 存储完成清理 list
    list.clear();//解析结束销毁不用的资源
        log.info("存储数据库成功！");
    }
}
复制代码

@Test
    public void readExcelTest() {
        // 写法1：
    DemoService demoService = getService(); //获取service业务实现类
        String fileName = TestFileUtil.getPath() + "demo" + File.separator + "demo.xlsx";
        // 这里 需要指定读用哪个class去读，然后读取第一个sheet 文件流会自动关闭
        EasyExcel.read(fileName, DemoData.class, new DemoModelAnalysisListener(demoService)).sheet().doRead();
        // 写法2：
        fileName = TestFileUtil.getPath() + "demo" + File.separator + "demo.xlsx";
        ExcelReader excelReader = null;
        try {
            excelReader = EasyExcel.read(fileName, DemoData.class, new DemoModelAnalysisListener(demoService)).build();
            ReadSheet readSheet = EasyExcel.readSheet(0).build();
            excelReader.read(readSheet);
        } finally {
            if (excelReader != null) {
                // 这里千万别忘记关闭，读的时候会创建临时文件，到时磁盘会崩的
                excelReader.finish();
            }
        }
    }
复制代码

@Data
public class DemoData {
    @ExcelProperty("字符串标题")
    private String string;
    @ExcelProperty("日期标题")
    private Date date;
    @ExcelProperty("数字标题")
    private Double doubleData;
    /**
     * 忽略这个字段
     */
    @ExcelIgnore
    private String ignore;
}
复制代码

@Test
    public void demoWriteTest() {
        // 写法1
        String fileName = TestFileUtil.getPath() + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
        // 这里 需要指定写用哪个class去写，然后写到第一个sheet，名字为模板 然后文件流会自动关闭
        // 如果这里想使用03 则 传入excelType参数即可
        EasyExcel.write(fileName, DemoData.class).sheet("模板").doWrite(data());
    // 写法2
        fileName = TestFileUtil.getPath() + "simpleWrite" + System.currentTimeMillis() + ".xlsx";
        // 这里 需要指定写用哪个class去写
        ExcelWriter excelWriter = null;
        try {
            excelWriter = EasyExcel.write(fileName, DemoData.class).build();
            WriteSheet writeSheet = EasyExcel.writerSheet("模板").build();
            excelWriter.write(data(), writeSheet);
        } finally {
            // 千万别忘记finish 会帮忙关闭流
            if (excelWriter != null) {
                excelWriter.finish();
            }
        }
    }
复制代码

@Data
public class MultiHeaderEntity implements Serializable {
      @ExcelProperty(value = {"一层信息","二层1"})
      private Integer id;
      @ExcelProperty(value = {"一层信息","二层2"})
      private String name;
      @ExcelProperty(value = {"一层信息","二层4"})
      private String description;
      @ExcelProperty(value = {"一层信息","二层3"})
      private Date birthday;
    }
复制代码

@Test
    public void complexHeadWrite() {
        String fileName = TestFileUtil.getPath() + "complexHeadWrite" + System.currentTimeMillis() + ".xlsx";
        // 这里 需要指定写用哪个class去写，然后写到第一个sheet，名字为模板 然后文件流会自动关闭
        EasyExcel.write(fileName, MultiHeaderEntity.class).sheet("复杂").doWrite(datalist());
    }