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

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

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

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

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

org.apache.poi

poi

3.13

org.apache.poi

poi-ooxml

3.13

主要代码：

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;

phttp://ublic 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);

http:// // 获取表格的总行数

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

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。