實現(xiàn)工具類
利用注解實現(xiàn)簡單的excel數(shù)據(jù)讀取，利用注解對類的屬性和excel中的表頭映射，使用Apache的poi就不用在業(yè)務(wù)代碼中涉及row,rows這些屬性了。
定義注解：
java" id="highlighter_485959">
|
1
2
3
4
5
6
7
|
@Retention(RetentionPolicy.RUNTIME)@Target(ElementType.FIELD)public @interface Excel { String name();} |
由于本例中只涉及根據(jù)Excel表頭部分對Excel進(jìn)行解析，只定義了一個name作為和Excel表頭的隱射。
工具類完整代碼如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
public class ExcelUtil<T> { Class<T> clazz; public ExcelUtil(Class<T> clazz) { this.clazz = clazz; } public List<T> importExcel(String sheetName, InputStream input) { int maxCol = 0; List<T> list = new ArrayList<T>(); try { Workbook workbook = WorkbookFactory.create(input); Sheet sheet = workbook.getSheet(sheetName); // 如果指定sheet名,則取指定sheet中的內(nèi)容. if (!sheetName.trim().equals("")) { sheet = workbook.getSheet(sheetName); } // 如果傳入的sheet名不存在則默認(rèn)指向第1個sheet. if (sheet == null) { sheet = workbook.getSheetAt(0); } int rows = sheet.getPhysicalNumberOfRows(); // 有數(shù)據(jù)時才處理 if (rows > 0) { List<Field> allFields = getMappedFiled(clazz, null); // 定義一個map用于存放列的序號和field. Map<Integer, Field> fieldsMap = new HashMap<Integer, Field>(); // 第一行為表頭 Row rowHead = sheet.getRow(0); Map<String, Integer> cellMap = new HashMap<>(); int cellNum = rowHead.getPhysicalNumberOfCells(); for (int i = 0; i < cellNum; i++){ cellMap.put(rowHead.getCell(i).getStringCellValue().toLowerCase(), i); } for (Field field : allFields) { // 將有注解的field存放到map中. if (field.isAnnotationPresent(Excel.class)) { Excel attr = field.getAnnotation(Excel.class); // 根據(jù)Name來獲取相應(yīng)的failed int col = cellMap.get(attr.name().toLowerCase()); field.setAccessible(true); fieldsMap.put(col, field); } } // 從第2行開始取數(shù)據(jù) for (int i = 1; i < rows; i++) { Row row = sheet.getRow(i); T entity = null; for (int j = 0; j < cellNum; j++) { Cell cell = row.getCell(j); if (cell == null) { continue; } int cellType = cell.getCellType(); String c = ""; if (cellType == HSSFCell.CELL_TYPE_NUMERIC) { DecimalFormat df = new DecimalFormat("0"); c = df.format(cell.getNumericCellValue()); } else if (cellType == HSSFCell.CELL_TYPE_BOOLEAN) { c = String.valueOf(cell.getBooleanCellValue()); } else { c = cell.getStringCellValue(); } if (c == null || c.equals("")) { continue; } entity = (entity == null ? clazz.newInstance() : entity); // 從map中得到對應(yīng)列的field. Field field = fieldsMap.get(j); if (field == null) { continue; } // 取得類型,并根據(jù)對象類型設(shè)置值. Class<?> fieldType = field.getType(); if (String.class == fieldType) { field.set(entity, String.valueOf(c)); } else if ((Integer.TYPE == fieldType) || (Integer.class == fieldType)) { field.set(entity, Integer.valueOf(c)); } else if ((Long.TYPE == fieldType) || (Long.class == fieldType)) { field.set(entity, Long.valueOf(c)); } else if ((Float.TYPE == fieldType) || (Float.class == fieldType)) { field.set(entity, Float.valueOf(c)); } else if ((Short.TYPE == fieldType) || (Short.class == fieldType)) { field.set(entity, Short.valueOf(c)); } else if ((Double.TYPE == fieldType) || (Double.class == fieldType)) { field.set(entity, Double.valueOf(c)); } else if (Character.TYPE == fieldType) { if (c.length() > 0) { field.set(entity, c.charAt(0)); } } } if (entity != null) { list.add(entity); } } } } catch (Exception e) { e.printStackTrace(); } return list; } /** * 得到實體類所有通過注解映射了數(shù)據(jù)表的字段 * * @param clazz * @param fields * @return */ private List<Field> getMappedFiled(Class clazz, List<Field> fields) { if (fields == null) { fields = new ArrayList<Field>(); } // 得到所有定義字段 Field[] allFields = clazz.getDeclaredFields(); // 得到所有field并存放到一個list中. for (Field field : allFields) { if (field.isAnnotationPresent(Excel.class)) { fields.add(field); } } if (clazz.getSuperclass() != null && !clazz.getSuperclass().equals(Object.class)) { getMappedFiled(clazz.getSuperclass(), fields); } return fields; }} |
代碼很簡單，獲取sheet，解析第一行，并和實體類標(biāo)有注解的字段一一對應(yīng)，用hashMap記錄下來，然后循環(huán)取得Excel中剩下所有的數(shù)據(jù)，根據(jù)map的對應(yīng)關(guān)系將值set到對應(yīng)字段。
基本使用
待解析表格如下：
定義實體類:
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
public class User { @Excel(name = "filed1") private String name; @Excel(name = "filed2") private String nameEn; @Excel(name = "filed3") private Integer age; @Excel(name = "filed4") private String six; @Excel(name = "filed5") private String weight; // ...getter setter} |
使用工具類：
|
1
2
3
4
5
6
7
8
9
10
11
|
public static void main (String[] args) { FileInputStream fileInputStream = null; try { fileInputStream = new FileInputStream("D://data.xlsx"); } catch (FileNotFoundException e) { e.printStackTrace(); } ExcelUtil<User> util = new ExcelUtil<>(User.class); List<User> jalanHotelList = util.importExcel("user", fileInputStream); // do something} |
利用這個思路可以擴展出導(dǎo)出excel功能，利用注解指定導(dǎo)出的excel表頭，甚至可以輕松控制excel表頭的顏色，合并屬性等等，在xdemo中有詳細(xì)復(fù)雜的示例，可以研究下。由于我的需求很簡單，就不整那么復(fù)雜啦。
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持服務(wù)器之家。
