SpringBoot结合easyexcel处理Excel文件

Posted 朱季谦

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了SpringBoot结合easyexcel处理Excel文件相关的知识,希望对你有一定的参考价值。

文/朱季谦

假如有这样一个需求,每天需要读取以下表头的Excel文件,统计文件里击中黑名单的比例,该文件is_blacklist列的1表示击中了黑名单,0表示未击中黑名单。

基于该需求,可以在定时任务通过easyexcel工具进行处理。

一、首先需要在SpringBoot引入easyexcel的maven依赖

<dependency>
   <groupId>com.alibaba</groupId>
   <artifactId>easyexcel</artifactId>
   <version>3.1.1</version>
 </dependency>

二、定义一个接收Excel文件表头列数据的Java Bean——

@Data
 @NoArgsConstructor
 @AllArgsConstructor
 @EqualsAndHashCode
 public class BlackTitle 
   @ExcelProperty(value = "key")
   private String key;

   @ExcelProperty(value = "is_blacklist")
   private String is_blacklist; 

 

这注解@ExcelProperty是easyexcel包的注解,起到的作用是,例如可以将private String key通过 @ExcelProperty(value = "key")映射到表格的key值。简单理解,就是每一个BlackTitle对象,能接收到表格每一行到数据,表格的key列值映射到BlackTitle的熟悉key值,表格的is_blacklist列值映射到BlackTitle的熟悉is_blacklist值。

三、继承EasyExcel 框架中的 AnalysisEventListener 接口的类 BlackTitleListener,用于监听并处理 Excel 文件中的数据行。

当EasyExcel解析每一行数据时,这个invoke 方法就会被调用,该BlackTitleListener类里的重写方法实现将当前读取的行数据添加到 data 列表中。该类最后提供了一个 getData() 方法,用于返回解析出来的数据列表,即集合data——

public static class BlackTitleListener extends AnalysisEventListener<BlackTitle> 

   private List<BlackTitle> data = new ArrayList<>();
  
   @Override
   public void invoke(BlackTitle blackTitle, AnalysisContext analysisContext) 
     data.add(blackTitle);
   
  
   @Override
   public void doAfterAllAnalysed(AnalysisContext analysisContext) 
   
  
   public List<BlackTitle> getData()
     return data;
   
 

四、实现读取Excel文件的sheet表格——

String source = "C:\\\\Users\\\\Admin\\\\Desktop\\\\blacklist.xlsx";
BlackTitleListener listener = new BlackTitleListener();
//读取 Excel 文件的第一个 sheet 中的数据,并将其转换成 BlackTitle 对象,最终存储在 listener 对象中。
EasyExcel.read(source,listener).head(BlackTitle.class).sheet(0).headRowNumber(1).doReadSync();

这里创建一个名为 listenerBlackTitleListener 对象,在该对象里,可以监听读取每一行数据的操作。调用EasyExcel.read(...) 方法去读取Excel文件,通过 .head(BlackTitle.class) 设置表头信息,即将表头字段映射到BlackTitle类里属性。通过 .sheet(0) 指定要读取第一个sheet表格数据。.headRowNumber(1) 表示表头位于第一行(索引从 0 开始)。.doReadSync() 方法开始同步读取 Excel 数据,然后将数据存储到 listener 对象中。

五、获取读取存储到集合里的数据进行处理

获取读取到集合里的数据——

List<BlackTitle> data = listener.getData();

接下来,就可以通过简单的集合迭代来统计出表格is_blacklist列值为1的比例了——

 long blackSize = data.stream().filter(k->k.getIs_blacklist().equals("1")).count();

最后,完整代码如下——

public class BlackTitleTest 
   public static void main(String[] args) 
     String source = "C:\\\\Users\\\\Admin\\\\Desktop\\\\blacklist.xlsx";
     BlackTitleListener listener = new BlackTitleListener();
     EasyExcel.*read*(source,listener).head(BlackTitle.class).sheet(0).headRowNumber(1).doReadSync();
     List<BlackTitle> data = listener.getData();
     long blackSize = data.stream().filter(k->k.getIs_blacklist().equals("1")).count();
    
   

   public static class BlackTitleListener extends AnalysisEventListener<BlackTitle> 

     private List<BlackTitle> data = new ArrayList<>();
     
     @Override
     public void invoke(BlackTitle blackTitle, AnalysisContext analysisContext) 
       data.add(blackTitle);
     
     
     @Override
     public void doAfterAllAnalysed(AnalysisContext analysisContext) 
     
     
     
     public List<BlackTitle> getData()
       return data;
     

   
 

SpringBoot系列之集成EasyExcel导入合并行数据

最近在做Excel导入功能,是一种一对多的数据,涉及到合并单元格的,考虑到使用poi去学,要自己去做处理,所以,看看有什么开源的框架,找到两个合适的框架,一个是easypoi是能支持这种的,这个框架提供了特定注解;还有一种是EasyExcel,阿里开源的,不过功能相对没easypoi齐全,比如这种合并单元格数据导入,就没有特定的注解,不过通过搜索资料,是可以实现的,不过要自己写工具类做处理,工具类整理自网上教程


不过在官网看到EasyExcel更专注于性能和易用,虽然功能支持不如easypoi,所以还是想尝试使用一下这个框架。所以,业务比较复杂,可以使用easypoi,业务简单点可以使用easyexcel,图来自EasyExcel官网:

加上easyexcel配置:

<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>easyexcel</artifactId>
    <version>2.2.10</version>
</dependency>

加上ExcelProperty注解,加上value和index,index是excel对应的列

package com.example.easyexcel.model.dto;

import com.alibaba.excel.annotation.ExcelProperty;
import lombok.Data;
import org.hibernate.validator.constraints.Length;

import javax.validation.constraints.NotBlank;


@Data
public class UserExcelDto 


    @ExcelProperty(value = "序号",index = 0)
    @NotBlank(message = "序号必须填!")
    private String seq;

    @ExcelProperty(value = "用户名" ,index = 1)
    @NotBlank(message = "用户名必须填!")
    private String name;

    @ExcelProperty(value = "密码", index = 2)
    @Length(min = 0 , max = 8 , message = "密码最多8位!")
    private String password;

    @ExcelProperty(value = "描述",index = 3)
    private String addressName;

    @ExcelProperty(value = "邮政编码",index = 4)
    private String code;





package com.common.excel;

import cn.hutool.core.util.ReflectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.excel.metadata.CellExtra;
import lombok.extern.slf4j.Slf4j;

import java.util.ArrayList;
import java.util.List;

@Slf4j
public class EasyExcelListener<T> extends AnalysisEventListener<T> 

    private List<T> datas;

    private Integer rowIndex;

    private List<CellExtra> extraMergeInfoList;

    public EasyExcelListener(Integer rowIndex) 
        this.rowIndex = rowIndex;
        datas = new ArrayList<>();
        extraMergeInfoList = new ArrayList<>();
    

    @Override
    public void invoke(T data, AnalysisContext context) 
        // 是否忽略空行数据,因为自己要做数据校验,所以还是加上,可以根据业务情况使用
        context.readWorkbookHolder().setIgnoreEmptyRow(false);
        ReflectUtil.invoke(data, "setIndex", StrUtil.toString(context.readRowHolder().getRowIndex()));
        datas.add(data);
    

    @Override
    public void doAfterAllAnalysed(AnalysisContext context) 
        log.info("所有数据解析完成!");
    

    @Override
    public void extra(CellExtra extra, AnalysisContext context) 
        switch (extra.getType()) 
            case MERGE:
                if (extra.getRowIndex() >= rowIndex) 
                    extraMergeInfoList.add(extra);
                
                break;
            default:
        
    

    public List<T> getData() 
        return datas;
    


    public List<CellExtra> getExtraMergeInfoList() 
        return extraMergeInfoList;
    




EasyExcel.read(file.getInputStream(), UserExcelDto.class, easyExcelListener)
                .extraRead(CellExtraTypeEnum.MERGE)
                .sheet(sheetNo)
                .headRowNumber(headRowNumber)
                .doRead();
        
System.out.println(JSONUtil.toJsonPrettyStr(easyExcelListener.getData()));
List<CellExtra> extraMergeInfoList = easyExcelListener.getExtraMergeInfoList();


EasyExcel Merge默认数据读取,这种情况,那些非合并的数据,比如姓名,密码这些数据,只会在第一个附上,框架之后的读取,不能读取到

[
    
        "code": "510000",
        "password": "********",
        "name": "user1",
        "addressName": "广州天河区",
        "seq": "1"
    ,
    
        "code": "510001",
        "addressName": "广州天河区"
    ,
    
        "code": "510002",
        "password": "***",
        "name": "user2",
        "addressName": "广州天河区",
        "seq": "2"
    
]

所以借助网上一个工具类实现,具体参考ExcelMergeHelper

工具类,思路也是读取excel数据,然后读取ExcelProperty注解,注意必须加上index,通过反射机制读取数据

package com.example.easyexcel.core.excel;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.annotation.ExcelProperty;
import com.alibaba.excel.enums.CellExtraTypeEnum;
import com.alibaba.excel.metadata.CellExtra;
import com.alibaba.excel.util.CollectionUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.Field;
import java.util.List;

public class ExcelMergeHelper<T> 

    private static final Logger LOGGER = LoggerFactory.getLogger(ExcelMergeHelper.class);

    /**
     * 返回解析后的List
     *
     * @param: fileName 文件名
     * @param: clazz Excel对应属性名
     * @param: sheetNo 要解析的sheet
     * @param: headRowNumber 正文起始行
     * @return java.util.List<T> 解析后的List
     */
    public List<T> getList(String fileName, Class<T> clazz, Integer sheetNo, Integer headRowNumber) 
        UserEasyExcelListener<T> listener = new UserEasyExcelListener<>(headRowNumber);
        try 
            EasyExcel.read(fileName, clazz, listener).extraRead(CellExtraTypeEnum.MERGE).sheet(sheetNo).headRowNumber(headRowNumber).doRead();
         catch (Exception e) 
            LOGGER.error(e.getMessage());
        
        List<CellExtra> extraMergeInfoList = listener.getExtraMergeInfoList();
        if (CollectionUtils.isEmpty(extraMergeInfoList)) 
            return listener.getData();
        
        List<T> data = explainMergeData(listener.getData(), extraMergeInfoList, headRowNumber);
        return data;
    

    /**
     * 处理合并单元格
     *
     * @param data               解析数据
     * @param extraMergeInfoList 合并单元格信息
     * @param headRowNumber      起始行
     * @return 填充好的解析数据
     */
    public List<T> explainMergeData(List<T> data, List<CellExtra> extraMergeInfoList, Integer headRowNumber) 
        //循环所有合并单元格信息
        extraMergeInfoList.forEach(cellExtra -> 
            int firstRowIndex = cellExtra.getFirstRowIndex() - headRowNumber;
            int lastRowIndex = cellExtra.getLastRowIndex() - headRowNumber;
            int firstColumnIndex = cellExtra.getFirstColumnIndex();
            int lastColumnIndex = cellExtra.getLastColumnIndex();
            //获取初始值
            Object initValue = getInitValueFromList(firstRowIndex, firstColumnIndex, data);
            //设置值
            for (int i = firstRowIndex; i <= lastRowIndex; i++) 
                for (int j = firstColumnIndex; j <= lastColumnIndex; j++) 
                    setInitValueToList(initValue, i, j, data);
                
            
        );
        return data;
    

    /**
     * 设置合并单元格的值
     *
     * @param filedValue  值
     * @param rowIndex    行
     * @param columnIndex 列
     * @param data        解析数据
     */
    public void setInitValueToList(Object filedValue, Integer rowIndex, Integer columnIndex, List<T> data) 
        T object = data.get(rowIndex);

        for (Field field : object.getClass().getDeclaredFields()) 
            //提升反射性能,关闭安全检查
            field.setAccessible(true);
            ExcelProperty annotation = field.getAnnotation(ExcelProperty.class);
            if (annotation != null) 
                if (annotation.index() == columnIndex) 
                    try 
                        field.set(object, filedValue);
                        break;
                     catch (IllegalAccessException e) 
                        LOGGER.error("设置合并单元格的值异常:"+e.getMessage());
                    
                
            
        
    


    /**
     * 获取合并单元格的初始值
     * rowIndex对应list的索引
     * columnIndex对应实体内的字段
     *
     * @param firstRowIndex    起始行
     * @param firstColumnIndex 起始列
     * @param data             列数据
     * @return 初始值
     */
    private Object getInitValueFromList(Integer firstRowIndex, Integer firstColumnIndex, List<T> data) 
        Object filedValue = null;
        T object = data.get(firstRowIndex);
        for (Field field : object.getClass().getDeclaredFields()) 
            //提升反射性能,关闭安全检查
            field.setAccessible(true);
            ExcelProperty annotation = field.getAnnotation(ExcelProperty.class);
            if (annotation != null) 
                if (annotation.index() == firstColumnIndex) 
                    try 
                        filedValue = field.get(object);
                        break;
                     catch (IllegalAccessException e) 
                        LOGGER.error("设置合并单元格的初始值异常:"+e.getMessage());
                    
                
            
        
        return filedValue;
    


对数据进行处理:

List<UserExcelDto> data = new ExcelMergeHelper().explainMergeData(easyExcelListener.getData(), extraMergeInfoList, headRowNumber);

封装后的数据,这种数据,我们就可以进行业务处理,过程相对比较麻烦

[
    
        "code": "510000",
        "password": "********",
        "name": "user1",
        "addressName": "广州天河区",
        "seq": "1"
    ,
    
        "code": "510001",
        "password": "********",
        "name": "user1",
        "addressName": "广州天河区",
        "seq": "1"
    ,
    
        "code": "510002",
        "password": "***",
        "name": "user2",
        "addressName": "广州天河区",
        "seq": "2"
    
]
package com.common.excel;

import com.alibaba.excel.annotation.ExcelProperty;
import com.alibaba.excel.metadata.CellExtra;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.Field;
import java.util.List;

public class ExcelMergeHelper<T> 

    private static final Logger LOGGER = LoggerFactory.getLogger(ExcelMergeHelper.class);

    /**
     * 处理合并单元格
     *
     * @param data               解析数据
     * @param extraMergeInfoList 合并单元格信息
     * @param headRowNumber      起始行
     * @return 填充好的解析数据
     */
    public List<T> explainMergeData(List<T> data, List<CellExtra> extraMergeInfoList, Integer headRowNumber) 
        //循环所有合并单元格信息
        extraMergeInfoList.forEach(cellExtra -> 
            int firstRowIndex = cellExtra.getFirstRowIndex() - headRowNumber;
            int lastRowIndex = cellExtra.getLastRowIndex() - headRowNumber;
            int firstColumnIndex = cellExtra.getFirstColumnIndex();
            int lastColumnIndex = cellExtra.getLastColumnIndex();
            //获取初始值
            Object initValue = getInitValueFromList(firstRowIndex, firstColumnIndex, data);
            //设置值
            for (int i = firstRowIndex; i <= lastRowIndex; i++) 
                for (int j = firstColumnIndex; j <= lastColumnIndex; j++) 
                    setInitValueToList(initValue, i, j, data);
                
            
        );
        return data;
    

    /**
     * 设置合并单元格的值
     *
     * @param filedValue  值
     * @param rowIndex    行
     * @param columnIndex 列
     * @param data        解析数据
     */
    public void setInitValueToList(Object filedValue, Integer rowIndex, Integer columnIndex, List<T> data) 
        if (!(data.size()>rowIndex))
            return;
        T object = data.get(rowIndex);

        for (Field field : object.getClass().getDeclaredFields()) 
            //提升反射性能,关闭安全检查
            field.setAccessible(true);
            ExcelProperty annotation = field.getAnnotation(ExcelProperty.class);
            if (annotation != null) 
                if (annotation.index() == columnIndex) 
                    try 
                        field.set(object, filedValue);
                        break;
                     catch (IllegalAccessException e) 
                        LOGGER.error("设置合并单元格的值异常:"+e.getMessage());
                    
                
            
        
    


    /**
     * 获取合并单元格的初始值
     * rowIndex对应list的索引
     * columnIndex对应实体内的字段
     *
     * @param firstRowIndex    起始行
     * @param firstColumnIndex 起始列
     * @param data             列数据
     * @return 初始值
     */
    private Object getInitValueFromList(Integer firstRowIndex, Integer firstColumnIndex, List<T> data) 
        Object filedValue = null;
        T object = data.get(firstRowIndex);
        for (Field field : objectSpringBoot系列之集成EasyExcel导入合并行数据

springboot中处理Excel数据

SpringBoot整合EasyExcel

SpringBoot整合EasyExcel

Springboot整合easyExcel导入导出Excel

SpringBoot入门教程:下载导出Excel(EasyExcel)