springboot使用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
到此這篇關於springboot使用AOP+反射實現Excel數據的讀取的文章就介紹到這瞭,更多相關springboot實現Excel讀取內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!
推薦閱讀:
- sql查詢返回值使用map封裝多個key和value實例
- Java使用poi做加自定義註解實現對象與Excel相互轉換
- SpringBoot @Cacheable自定義KeyGenerator方式
- mybatis-plus3.0.1枚舉返回為null解決辦法
- SpringBoot+EasyPoi實現excel導出功能