EasyExcel从指定行写入详解

发布时间：2023-05-21

一、指定行写入方式

//创建对应的excel对象
ExcelWriter excelWriter = EasyExcel.write(fileName).build();
//写入方式一：从指定行开始写入，后续行逐行写入数据
WriteSheet writeSheet = EasyExcel.writerSheet(sheetName).build();
writeSheet.setStartRow(2); 
excelWriter.write(dataList, writeSheet);
//写入方式二：从指定行开始写入，同时写入指定条数的数据
WriteSheet writeSheet = EasyExcel.writerSheet(sheetName).build();
writeSheet.setStartRow(2);
writeSheet.setPageSize(100);
excelWriter.write(dataList, writeSheet);
excelWriter.finish();

从指定行写入是通过设置WriteSheet的startRow属性实现的。当设置了startRow后，通过ExcelWriter写入数据时，会从指定行开始写入。此外，还可以设置pageSize属性，控制每页写入多少条数据，以提高写入效率。

二、多线程写入数据

//创建对应的excel对象
ExcelWriter excelWriter = EasyExcel.write(fileName).build();
//构造多个写入任务，同时写入当前页对应的数据
List<WriteSheet> sheetList = new ArrayList<WriteSheet>();
for(int i=0; i < pageNum; i++){
    WriteSheet sheet = EasyExcel.writerSheet(i, sheetName).build();
    sheet.setStartRow(i * pageSize + 2);
    sheet.setPageSize(pageSize);
    sheetList.add(sheet);
}
//构造多线程的写入任务
ExecutorService executorService = ...;
try {
    CountDownLatch latch = new CountDownLatch(sheetList.size());
    for (WriteSheet sheet : sheetList) {
        executorService.execute(() -> {
            excelWriter.write(dataList, sheet);
            latch.countDown();
        });
    }
    latch.await();
} catch (Exception e) {
    e.printStackTrace();
} finally {
    excelWriter.finish();
    executorService.shutdown();
}

多线程写入是通过创建多个WriteSheet对象实现的。同样是设置startRow和pageSize属性，将数据均匀地分配到不同的线程中写入，以提高写入效率。需要注意的是，使用多线程写入时，需要确保数据需要是可重入的，要求写入数据不能有重复。

三、不同数据类型写入

public class Data{
    @ExcelProperty(value = "字符串标题", index = 0)
    private String str;
    @ExcelProperty(value = "日期标题", index = 1)
    @DateTimeFormat("yyyy年MM月dd日HH时mm分ss秒")
    private Date date;
    @ExcelProperty(value = "数字标题", index = 2)
    private Double doubleData;
}
//创建对应的excel对象
ExcelWriter excelWriter = EasyExcel.write(fileName, Data.class).build();
//调用write方法，进行数据写入
excelWriter.write(dataList, EasyExcel.writerSheet(sheetName).build());
excelWriter.finish();

EasyExcel支持不同的数据类型写入，只需要在对应的类中添加@ExcelProperty注解并指定对应的标题和索引位置，然后通过EasyExcel.write方法进行写入即可。支持的数据类型包括String、Date、Double、Boolean、LocalDate、BigDecimal等。

四、自定义数据转换器

public class DemoDataConverter implements Converter<String>{
    private static final Map<String, Integer> map = new HashMap<>();
    static {
        map.put("男", 1);
        map.put("女", 2);
    }
    @Override
    public Class supportJavaTypeKey() {
        return String.class;
    }
    @Override
    public CellData convertToExcelData(String value, ExcelContentProperty contentProperty, GlobalConfiguration globalConfiguration) throws Exception {
        if (value == null || value.isEmpty()) {
            return null;
        }
        return new CellData(map.get(value));
    }
    @Override
    public String convertToJavaData(CellData cellData, ExcelContentProperty contentProperty, GlobalConfiguration globalConfiguration) throws Exception {
        return map.entrySet().stream()
                .filter(entry -> Objects.equals(entry.getValue(), cellData.getNumberValue().intValue()))
                .map(Map.Entry::getKey)
                .findFirst()
                .orElse(null);
    }
}
//创建对应的excel对象
ExcelWriter excelWriter = EasyExcel.write(fileName, Data.class).registerConverter(new DemoDataConverter()).build();
//调用write方法，进行数据写入
excelWriter.write(dataList, EasyExcel.writerSheet(sheetName).build());
excelWriter.finish();

自定义数据转换器是通过实现Converter接口来实现的，需要重写convertToExcelData和convertToJavaData两个方法。其中，convertToExcelData用于将Java对象转换成对应的Excel单元格数据，convertToJavaData用于将Excel单元格数据转换成对应的Java对象。通过重写这两个方法，可以实现自定义的数据转换逻辑。

五、数据校验

public class DemoDataListener extends AnalysisEventListener<Data> {
    List<Data> dataList = new ArrayList<>();
    @Override
    public void invoke(Data data, AnalysisContext context) {
        if(data.getStr() == null || data.getDate() == null || data.getDoubleData() == null){
            throw new RuntimeException("数据校验失败！");
        }
        dataList.add(data);
        // 每1000条数据处理一次业务逻辑，防止数据量过大导致内存溢出
        if (dataList.size() >= 1000) {
            // TODO: 处理业务逻辑
            dataList.clear();
        }
    }
    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        //处理剩余的数据
        // TODO: 处理业务逻辑
        dataList.clear();
    }
}
//创建对应的excel对象
ExcelReader excelReader = EasyExcel.read(fileName, Data.class, new DemoDataListener()).build();
//调用read方法，进行数据读取
excelReader.read(EasyExcel.readerSheet(sheetName).head(Data.class).build());

数据校验是通过实现AnalysisEventListener接口来实现的。读取数据时，如果数据校验失败，可以抛出RuntimeException来停止数据读取。通过设置读取的pageSize属性，可以控制每页读取多少条数据，以提高读取效率。同时，在业务逻辑处理时，也需要注意对数据量过大的情况进行适当的分页处理。

推荐文章