主要调整了一下代码类的封装。
以前的代码sheetIndex散落在多个方法上，而方法是要求按顺序执行的，没有被封装起来，会导致直接用部分方法会报错。
1 import java.io.FileInputStream;
2 import java.io.InputStream;
3 import java.util.ArrayList;
4 import java.util.List;
5
6 import org.apache.logging.log4j.LogManager;
7 import org.apache.logging.log4j.Logger;
8 import org.apache.poi.ss.usermodel.Cell;
9 import org.apache.poi.ss.usermodel.DateUtil;
10 import org.apache.poi.ss.usermodel.Row;
11 import org.apache.poi.ss.usermodel.Sheet;
12 import org.apache.poi.ss.usermodel.Workbook;
13 import org.apache.poi.ss.usermodel.WorkbookFactory;
14
15 /**
16 * ClassName:ExcelReader.java Author: wenbin.ji CreateTime: Jan 28, 2011
17 * 11:16:29 AM Description:Excel数据读取工具类，POI实现，兼容Excel2003，及Excel2007
18 **/
19
20 public class ExcelReader {
21
22 private static Logger logger = LogManager.getLogger(ExcelReader.class);
23
24 private Sheet sheet=null;
25 private List<String[]> dataList = null;
26
27
28 public ExcelReader(String path,int sheetIndex) {
29 try {
30 InputStream inp = new FileInputStream(path);
31 Workbook wb = WorkbookFactory.create(inp);
32 sheet = wb.getSheetAt(sheetIndex);
33 dataList=this.getAllData();
34 } catch (Exception e) {
35 e.printStackTrace();
36 }
37 }
38
39 /**
40 * 得到正式数据，头部除外
41 * @param sheetIndex
42 * @param startRow
43 * @return
44 */
45 public List<String[]> getBodyData(int startRow) {
46 List<String[]> allData = this.getAllData();
47 return allData.subList(startRow,allData.size());
48 }
49
50
51 /**
52 * 获得最大数量的表格行。不考虑无数据行
53 * @param sheetIndex
54 * @return
55 */
56 private List<String[]> getAllData() {
57 return this.getAllData(Integer.MAX_VALUE);
58 }
59
60 /**
61 * 取Excel所有数据，包含header
62 * @return List<String[]>
63 */
64 private List<String[]> getAllData(int endRow) {
65 int columnNum = 0;
66 dataList = new ArrayList<String[]>(100);
67 int lastRow=this.getRowNum();
68 //如果总行数大于指定结束行。那么以指定行为准
69 lastRow=lastRow>endRow?endRow:lastRow;
70 logger.error("lastRow=========>:"+lastRow);
71 if (sheet.getRow(0) != null) {
72 columnNum = sheet.getRow(0).getLastCellNum()
73 - sheet.getRow(0).getFirstCellNum();
74 }
75 if (columnNum > 0) {
76 for (Row row : sheet) {
77 //到最后一行结束循环
78 if(row.getRowNum()>lastRow){
79 break;
80 }
81
82 String[] singleRow = new String[columnNum];
83 int n = 0;
84 for (int i = 0; i < columnNum; i++) {
85 Cell cell = row.getCell(i, Row.CREATE_NULL_AS_BLANK);
86 switch (cell.getCellType()) {
87 case Cell.CELL_TYPE_BLANK:
88 singleRow[n] = "";
89 break;
90 case Cell.CELL_TYPE_BOOLEAN:
91 singleRow[n] = Boolean.toString(cell
92 .getBooleanCellValue());
93 break;
94 // 数值
95 case Cell.CELL_TYPE_NUMERIC:
96 if (DateUtil.isCellDateFormatted(cell)) {
97 singleRow[n] = String.valueOf(cell
98 .getDateCellValue());
99 } else {
100 cell.setCellType(Cell.CELL_TYPE_STRING);
101 String temp = cell.getStringCellValue();
102 // 判断是否包含小数点，如果不含小数点，则以字符串读取，如果含小数点，则转换为Double类型的字符串
103 if (temp.indexOf(".") > -1) {
104 singleRow[n] = String.valueOf(new Double(temp))
105 .trim();
106 } else {
107 singleRow[n] = temp.trim();
108 }
109 }
110 break;
111 case Cell.CELL_TYPE_STRING:
112 singleRow[n] = cell.getStringCellValue().trim();
113 break;
114 case Cell.CELL_TYPE_ERROR:
115 singleRow[n] = "";
116 break;
117 case Cell.CELL_TYPE_FORMULA:
118 cell.setCellType(Cell.CELL_TYPE_STRING);
119 singleRow[n] = cell.getStringCellValue();
120 if (singleRow[n] != null) {
121 singleRow[n] = singleRow[n].replaceAll("#N/A", "")
122 .trim();
123 }
124 break;
125 default:
126 singleRow[n] = "";
127 break;
128 }
129 n++;
130 }
131 if ("".equals(singleRow[0])) {
132 continue;
133 }// 如果第一行为空，跳过
134 dataList.add(singleRow);
135 }
136 }
137 return dataList;
138 }
139
140 /**
141 * 返回Excel最大行index值，实际行数要加1
142 *
143 * @return
144 */
145 public int getRowNum() {
146 return sheet.getLastRowNum();
147 }
148
149 /**
150 * 返回数据的列数
151 *
152 * @return
153 */
154 public int getColumnNum() {
155 Row row = sheet.getRow(0);
156 if (row != null && row.getLastCellNum() > 0) {
157 return row.getLastCellNum();
158 }
159 return 0;
160 }
161
162 /**
163 * 获取某一行数据
164 * @param rowIndex
165 * 计数从0开始，rowIndex为0代表header行
166 * @return
167 */
168 public String[] getRowData(int rowIndex) {
169 String[] dataArray = null;
170 if (rowIndex > this.getRowNum()) {
171 return dataArray;
172 } else {
173 dataArray = new String[this.getColumnNum()];
174 return this.dataList.get(rowIndex);
175 }
176
177 }
178
179 /**
180 * 获取某一列数据
181 *
182 * @param colIndex
183 * @return
184 */
185 public String[] getColumnData(int colIndex) {
186 String[] dataArray = null;
187 if (colIndex > this.getColumnNum()) {
188 return dataArray;
189 } else {
190 if (this.dataList != null && this.dataList.size() > 0) {
191 dataArray = new String[this.getRowNum() + 1];
192 int index = 0;
193 for (String[] rowData : dataList) {
194 if (rowData != null) {
195 dataArray[index] = rowData[colIndex];
196 index++;
197 }
198 }
199 }
200 }
201 return dataArray;
202
203 }
204 }
时间： 2024-11-03 22:49:07