(转)Java + Excel 接口自动化

最近项目比较悠闲，想找点事干，写了个 Excel 接口测试的 "框架" 以前用 python 写过一个，这次用 java, 应该说框架都不算，反正就是写了，能帮我解决问题就行。

当然咯，也许会问干嘛那么麻烦直接用 feed4testng, 或者 testng 就行了，没事找事干还专门写个这玩意... 呵呵，就闲的蛋疼！

文笔有限不知道怎么写，直接上代码：

欢迎各位指定，或提出好的意见，总觉得还有很多不好的地方。

结构就这破样了， E 文也不好， 随便捣鼓，开心就好。 哈哈

 

ExcelUtil.java 类：

  1 package com.hozhu.excel;
  2 
  3 import java.io.BufferedInputStream;
  4 import java.io.File;
  5 import java.io.FileInputStream;
  6 import java.io.FileOutputStream;
  7 import java.io.IOException;
  8 import java.util.ArrayList;
  9 import java.util.HashMap;
 10 import java.util.List;
 11 import java.util.Map;
 12 
 13 import org.apache.log4j.Logger;
 14 import org.apache.poi.ss.usermodel.CellStyle;
 15 import org.apache.poi.ss.usermodel.IndexedColors;
 16 import org.apache.poi.xssf.usermodel.XSSFCell;
 17 import org.apache.poi.xssf.usermodel.XSSFCellStyle;
 18 import org.apache.poi.xssf.usermodel.XSSFRow;
 19 import org.apache.poi.xssf.usermodel.XSSFSheet;
 20 import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 21 
 22 /**
 23  * 只支持读取 .xlsx 所有方法读取数据时, 未对 Excel 格式类型进行判断处理 如 Excel 中有特殊数据类型, 需在 Excel
 24  * 中标注为文本类型, 程序才能正常处理
 25  * 
 26  * @author Roger
 27  * @version 1.0
 28  */
 29 public class ExcelUtil {
 30     private static Logger logger = Logger.getLogger(ExcelUtil.class);
 31 
 32     private String filePath = null;
 33     private String sheetName = null;
 34 
 35     public String getSheetName() {
 36         return sheetName;
 37     }
 38 
 39     public void setSheetName(String sheetName) {
 40         this.sheetName = sheetName;
 41     }
 42 
 43     public String getFilePath() {
 44         return filePath;
 45     }
 46 
 47     public void setFilePath(String filePath) {
 48         this.filePath = filePath;
 49     }
 50 
 51     /**
 52      * 判断成员变量 filePath 是否为空
 53      * 
 54      * @return
 55      */
 56     private boolean isFilePathEmpty() {
 57         boolean flag = false;
 58         if ((null != filePath && (!filePath.equals("")))) {
 59             flag = true;
 60         }
 61         return flag;
 62     }
 63 
 64     private boolean isSheetNameEmpty() {
 65         boolean flag = false;
 66         if ((null != sheetName && (!sheetName.equals("")))) {
 67             flag = true;
 68         }
 69         return flag;
 70     }
 71 
 72     // 使用静态内部类创建外部类对象
 73     private static class Excel {
 74         private static ExcelUtil excelUtil = new ExcelUtil();
 75     }
 76 
 77     private ExcelUtil() {
 78     }
 79 
 80     // 获取 ExcelUtil 实例
 81     public static ExcelUtil getInstance() {
 82         return Excel.excelUtil;
 83     }
 84 
 85     /**
 86      * 检查传入的文件后缀是否为 xlsx
 87      * 
 88      * @param filePacht
 89      * @return
 90      */
 91     public boolean checkXlsx(File file) {
 92         boolean flag = false;
 93 
 94         if (file.exists()) {
 95             String str = file.getName();
 96             if (str.substring(str.lastIndexOf(".") + 1).equals("xlsx"))
 97                 flag = true;
 98         } else {
 99             logger.info(file.getName() + ", 文件不存在...");
100             // System.out.println("文件不存在...");
101         }
102 
103         return flag;
104     }
105 
106     /**
107      * 
108      * @param file
109      * @return true: 文件未被操作; false: 文件正在被操作.
110      */
111     public boolean isFile(File file) {
112         return file.renameTo(file);
113     }
114 
115     public File createFile() {
116         File file = null;
117         if (isFilePathEmpty()) {
118             file = new File(filePath);
119         } else {
120             logger.error("filePath 不能为: "
121                     + filePath
122                     + ", 请先使用 'ExcelUtil.getInstance().setFilePath(filePath)' 设置!");
123             // System.out.println("filePath 不能为: " + filePath +
124             // ", 请先使用 'ExcelUtil.getInstance().setFilePath(filePath)' 设置!");
125             System.exit(-1);
126         }
127         return file;
128     }
129 
130     /**
131      * 创建 XSSFWorkbook
132      * 
133      * @param 文件路径
134      * @return
135      */
136     public XSSFWorkbook createExcelWorkBook() {
137         File file = createFile();
138 
139         BufferedInputStream in = null;
140         XSSFWorkbook book = null;
141 
142         if (checkXlsx(file)) {
143             try {
144                 in = new BufferedInputStream(new FileInputStream(file));
145                 book = new XSSFWorkbook(in);
146             } catch (IOException e) {
147                 e.printStackTrace();
148             }
149         }
150         return book;
151     }
152 
153     /**
154      * 创建 XSSFSheet
155      * 
156      * @param sheetName
157      * @return
158      */
159     public XSSFSheet createExcelSheet(String sheetName) {
160         XSSFSheet sheet = null;
161         if (isSheetNameEmpty()) {
162             XSSFWorkbook book = createExcelWorkBook();
163             if (book != null) {
164                 int sheetCount = book.getNumberOfSheets();
165                 for (int i = 0; i < sheetCount; i++) {
166                     if (sheetName.equals(book.getSheetName(i))) {
167                         sheet = book.getSheet(sheetName);
168                         break;
169                     }
170                 }
171             }
172         } else {
173             logger.error("sheetName 不能为: "
174                     + sheetName
175                     + ", 请先使用 'ExcelUtil.getInstance().setSheetName(SheetName)' 设置!");
176             // System.out.println("sheetName 不能为: " + sheetName +
177             // ", 请先使用 'ExcelUtil.getInstance().setSheetName(SheetName)' 设置!");
178             System.exit(-1);
179         }
180 
181         return sheet;
182     }
183 
184     /**
185      * 获取指定行
186      * 
187      * @param sheetName
188      * @param line
189      *            行索引, 从 0 开始
190      * @return
191      */
192     public XSSFRow getExcelRow(int line) {
193         XSSFRow row = null;
194         if (line <= getSheetMaxRow())
195             row = createExcelSheet(sheetName).getRow(line);
196         return row;
197     }
198 
199     /**
200      * 获取指定 sheet 中最大行数
201      * 
202      * @param sheetName
203      * @return
204      */
205     public int getSheetMaxRow() {
206         int maxRow = -1;
207         if (createExcelSheet(sheetName) != null)
208             maxRow = createExcelSheet(sheetName).getLastRowNum();
209         return maxRow;
210     }
211 
212     /**
213      * 使用正则表达式去掉多余的.与0
214      * 
215      * @param s
216      * @return
217      */
218     private String subZeroAndDot(String s) {
219         if (s.indexOf(".") > 0) {
220             // 去掉多余的 0
221             s = s.replaceAll("0+?$", "");
222             // 如果最后一位是.则去掉
223             s = s.replaceAll("[.]$", "");
224         }
225         return s;
226     }
227 
228     /**
229      * 获取 Excel 指定行数据
230      * 
231      * @param sheetName
232      * @param line
233      * @return 返回一个一维数组
234      */
235     public String[] readExcelRows(int line) {
236         String[] result = null;
237         XSSFRow row = getExcelRow(line);
238         int maxRow = getSheetMaxRow();
239         if (row != null && maxRow > -1) {
240             int columnNum = row.getLastCellNum();
241             result = new String[columnNum];
242             for (int i = 0; i < columnNum; i++) {
243                 // 判断单元格是否为空, 不进行判断时, 如遇到空白单元格时, 抛出空指针异常
244                 if (null != row.getCell(i))
245                     result[i] = subZeroAndDot(row.getCell(i).toString().trim());
246             }
247         }
248         return result;
249     }
250 
251     /**
252      * 获取指定单元格中内容
253      * 
254      * @param sheetName
255      * @param line
256      *            行
257      * @param column
258      *            列
259      * @return
260      */
261     public String readExcelCell(int line, int column) {
262         String[] value = null;
263         String result = "";
264         value = readExcelRows(line);
265         if (value != null) {
266             result = value[column];
267         }
268         return result;
269     }
270 
271     /**
272      * 从指定行开始读取 Excel 中所有数据
273      * 
274      * @param sheetName
275      *            sheet 名称
276      * @param line
277      *            表标题所在行
278      * @return 返回一个包含 HashMap 的 ArrayList; 格式：[{第一行数据},{第二行数据}]
279      *         {列 1 标题 = 列 1 值, 列 2 标题 = 列 2 值}
280      */
281     public List> readExcelAllData(int line) {
282         List> result = new ArrayList>();
283         // 读取标题行
284         String[] titleRow = readExcelRows(line);
285         int maxRow = getSheetMaxRow();
286         XSSFRow row;
287         if (null != titleRow && maxRow != -1) {
288             // i = line + 1; 标题的下一行开始取数据
289             for (int i = line + 1; i <= maxRow; i++) {
290                 row = getExcelRow(i);
291                 Map map = new HashMap();
292                 for (int j = 0; j < row.getLastCellNum(); j++) {
293                     if (null != row.getCell(j)) {
294                         String value = subZeroAndDot(row.getCell(j).toString()
295                                 .trim());
296                         map.put(titleRow[j], value);
297                     }
298                 }
299                 result.add(map);
300             }
301         }
302         return result;
303     }
304 
305     /**
306      * 从指定数据行开始读取 Excel 所有数据
307      * 
308      * @param sheetName
309      *            sheet 名称
310      * @param line
311      *            数据读取开始行
312      * @return
313      */
314     public String[][] readDataArrayAll(int line) {
315         ArrayList