使用AOP+反射实现Excel数据的读取

简介: 使用AOP+反射实现Excel数据的读取

如果我们遇到把excel表格中的数据导入到数据库,首先我们要做的是:将excel中的数据先读取出来。
因此,今天就给大家分享一个读取Excel表格数据的代码示例:

为了演示方便,首先我们创建一个Spring Boot项目;具体创建过程这里不再详细介绍;

示例代码主要使用了Apache下的poi的jar包及API;因此,我们需要在pom.xml文件中导入以下依赖:

        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.13</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.13</version>
        </dependency>

主要代码:

ExcelUtils.java

import com.example.springbatch.xxkfz.annotation.ExcelField;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

/**
 * @author xxkfz
 * Excel工具类
 */

@Slf4j
public class ExcelUtils {
   
   

    private HSSFWorkbook workbook;

    public ExcelUtils(String fileDir) {
   
   
        File file = new File(fileDir);
        try {
   
   
            workbook = new HSSFWorkbook(new FileInputStream(file));
        } catch (FileNotFoundException e) {
   
   
            e.printStackTrace();
        } catch (IOException e) {
   
   
            e.printStackTrace();
        }
    }

    /**
     * 读取Excel数据
     *
     * @param sheetName
     * @param object
     * @return
     */
    public List readFromExcelData(String sheetName, Object object) {
   
   
        List result = new ArrayList();

        // 获取该对象的class对象
        Class class_ = object.getClass();

        // 获得该类的所有属性
        Field[] fields = class_.getDeclaredFields();

        // 读取excel数据  获得指定的excel表
        HSSFSheet sheet = workbook.getSheet(sheetName);

        // 获取表格的总行数
        int rowCount = sheet.getLastRowNum() + 1; // 需要加一
        if (rowCount < 1) {
   
   
            return result;
        }

        // 获取表头的列数
        int columnCount = sheet.getRow(0).getLastCellNum();

        // 读取表头信息,确定需要用的方法名---set方法
        // 用于存储方法名
        String[] methodNames = new String[columnCount]; // 表头列数即为需要的set方法个数

        // 用于存储属性类型
        String[] fieldTypes = new String[columnCount];

        // 获得表头行对象
        HSSFRow titleRow = sheet.getRow(0);

        // 遍历表头列
        for (int columnIndex = 0; columnIndex < columnCount; columnIndex++) {
   
   

            // 取出某一列的列名
            String colName = titleRow.getCell(columnIndex).toString();
/*
            // 将列名的首字母字母转化为大写
            String UColName = Character.toUpperCase(colName.charAt(0)) + colName.substring(1, colName.length());

            // set方法名存到methodNames
            methodNames[columnIndex] = "set" + UColName;
*/
            //
            String fieldName = fields[columnIndex].getName();
            String UpperFieldName = Character.toUpperCase(fieldName.charAt(0)) + fieldName.substring(1, fieldName.length());
            methodNames[columnIndex] = "set" + UpperFieldName;

            // 遍历属性数组
            for (int i = 0; i < fields.length; i++) {
   
   

                // 获取属性上的注解name值
                String name = fields[i].getAnnotation(ExcelField.class).name();

                // 属性与表头相等
                if (Objects.nonNull(name) && colName.equals(name)) {
   
   
                    //  将属性类型放到数组中
                    fieldTypes[columnIndex] = fields[i].getType().getName();
                }
            }
        }

        // 逐行读取数据 从1开始 忽略表头
        for (int rowIndex = 1; rowIndex < rowCount; rowIndex++) {
   
   
            // 获得行对象
            HSSFRow row = sheet.getRow(rowIndex);
            if (row != null) {
   
   
                Object obj = null;
                // 实例化该泛型类的对象一个对象
                try {
   
   
                    obj = class_.newInstance();
                } catch (Exception e1) {
   
   
                    e1.printStackTrace();
                }

                // 获得本行中各单元格中的数据
                for (int columnIndex = 0; columnIndex < columnCount; columnIndex++) {
   
   
                    String data = row.getCell(columnIndex).toString();
                    // 获取要调用方法的方法名
                    String methodName = methodNames[columnIndex];

                    obj = this.valueConvert(fieldTypes[columnIndex], methodName, class_, obj, data);
                }
                result.add(obj);
            }
        }
        return result;
    }

    /**
     * @param fieldType  字段类型
     * @param methodName 方法名
     * @param class_
     * @param data
     * @return
     */
    private Object valueConvert(String fieldType, String methodName, Class class_, Object obj, String data) {
   
   
        Method method = null;
        if (Objects.isNull(fieldType) || Objects.isNull(methodName) || Objects.isNull(class_) || Objects.isNull(obj)) {
   
   
            return obj;
        }
        try {
   
   
            switch (fieldType) {
   
   
                case "java.lang.String":
                    method = class_.getDeclaredMethod(methodName, String.class);
                    method.invoke(obj, data); // 执行该方法
                    break;
                case "java.lang.Integer":
                    method = class_.getDeclaredMethod(methodName, Integer.class);
                    Integer value = Integer.valueOf(data);
                    method.invoke(obj, value); // 执行该方法
                    break;
                case "java.lang.Boolean":
                    method = class_.getDeclaredMethod(methodName, Boolean.class);
                    Boolean booleanValue = Boolean.getBoolean(data);
                    method.invoke(obj, booleanValue); // 执行该方法
                    break;
                case "java.lang.Double":
                    method = class_.getDeclaredMethod(methodName, Double.class);
                    double doubleValue = Double.parseDouble(data);
                    method.invoke(obj, doubleValue); // 执行该方法
                    break;
                case "java.math.BigDecimal":
                    method = class_.getDeclaredMethod(methodName, BigDecimal.class);
                    BigDecimal bigDecimal = new BigDecimal(data);
                    method.invoke(obj, bigDecimal); // 执行该方法
                    break;
            }
        } catch (Exception e) {
   
   
            e.printStackTrace();
        }
        return obj;
    }
}

ExcelField.java

import java.lang.annotation.*;

/**
 * @author xxkfz
 */
@Target({
   
   ElementType.METHOD, ElementType.FIELD, ElementType.TYPE}) //注解放置的目标位置,METHOD是可注解在方法级别上
@Retention(RetentionPolicy.RUNTIME) //注解在哪个阶段执行
@Documented
public @interface ExcelField {
   
   
    String name() default "";
}

实体类 ExcelFileField.java

@Data
@AllArgsConstructor
@NoArgsConstructor
@ToString
public class ExcelFileField {
   
   

    @ExcelField(name = "id")
    private String id;

    @ExcelField(name = "code")
    private String code;

    @ExcelField(name = "type")
    private String type;

    @ExcelField(name = "version")
    private String version;
}

函数测试

 @Test
    void readExcel() {
   
   
        ExcelUtils utils = new ExcelUtils("E:/test.xls");
        ExcelFileField interfaceField = new ExcelFileField();
        List list = utils.readFromExcelData("sheet1", interfaceField);
        for (int i = 0; i < list.size(); i++) {
   
   
            ExcelFileField item = (ExcelFileField) list.get(i);
            System.out.println(item.toString());
        }
    }

Excel表格数据

在这里插入图片描述

测试结果:

ExcelFileField(id=X0001, code=X0001, type=X0001, version=X0001)
ExcelFileField(id=X0002, code=X0002, type=X0002, version=X0002)

Process finished with exit code 0

<hr style=" border:solid; width:100px; height:1px;" color=#000000 size=1">

本篇文章到这里就基本结束了,如果这篇文章对你有帮助,希望大家能留下你的点赞、 关注、 分享、 留言❤️❤️❤️

相关文章
|
1月前
|
数据安全/隐私保护
杨老师课堂之Excel VBA 程序开发第七讲表格数据高亮显示
杨老师课堂之Excel VBA 程序开发第七讲表格数据高亮显示
23 1
|
2月前
|
前端开发 Java
基于Java爬取微博数据(二) 正文长文本+导出数据Excel
【5月更文挑战第12天】基于Java爬取微博数据,正文长文本+导出数据Excel
|
14天前
|
数据格式 Python
Python代码示例,读取excel表格,将行数据转为列数据。(10)
【7月更文挑战第10天】Python代码示例,读取excel表格,将行数据转为列数据。
32 2
|
1月前
|
存储 数据挖掘 Python
使用Python集合高效统计Excel数据
使用Python集合高效统计Excel数据
40 7
VBA如何用Excel数据批量生成Word文档
VBA|用Excel数据批量生成并修改用模板创建的Word文档
|
15天前
|
前端开发 JavaScript PHP
技术心得:前端点击按钮,导入excel文件,上传到后台,excel接收和更新数据
技术心得:前端点击按钮,导入excel文件,上传到后台,excel接收和更新数据
17 0
|
1月前
|
数据安全/隐私保护
杨老师课堂之Excel VBA 程序开发第六讲 根据制定列创建相应工作表及数据
杨老师课堂之Excel VBA 程序开发第六讲 根据制定列创建相应工作表及数据
20 1
|
1月前
|
easyexcel Java API
SpringBoot集成EasyExcel 3.x:高效实现Excel数据的优雅导入与导出
SpringBoot集成EasyExcel 3.x:高效实现Excel数据的优雅导入与导出
240 1
|
16天前
|
数据可视化 数据挖掘 数据处理
Python对Excel两列数据进行运算【从基础到高级的全面指南】
【7月更文挑战第6天】使用Python的`pandas`库处理Excel数据,涉及安装`pandas`和`openpyxl`,读取数据如`df = pd.read_excel(&#39;data.xlsx&#39;)`,进行运算如`df[&#39;Sum&#39;] = df[&#39;Column1&#39;] + df[&#39;Column2&#39;]`,并将结果写回Excel。`pandas`还支持数据筛选、分组、可视化、异常处理和性能优化。通过熟练运用这些功能,可以高效分析Excel表格。
21 0
|
1月前
|
SQL Rust 数据挖掘
4秒读取50w行Excel数据
4秒读取50w行Excel数据