使用POI读取Excel表格中数据 2
背景
自己最近在研究自动换测试,在进行参数化时候,会对把用例写在Excel表格中,然后通过Java来取表格中数据,进而进行参数化。
问题
在使用POI读取Excel表格中数据 1中,如下图情况,如果,现在出现不适第一行,第一列的情况,我们又该怎么解决,如第二张图。
表格中的单元格里出现没有输入的情况(图片中红色部分),如何通过POI进行读取,或者如何处理?预期结果不用考虑。
代码
代码优化。
只贴了获取表格中数据代码,其他不是本文内容。
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.text.SimpleDateFormat;
public class ExcelUtil {
public static Object[][] getTestData(String excelFilePath, String SheetName) throws IOException {
File file = new File(excelFilePath);
FileInputStream inputStream = new FileInputStream(file);
Workbook workbook = null;
//获取文件扩展名
String fileExtensionName = excelFilePath.substring(excelFilePath.indexOf("."));
if (fileExtensionName.equals(".xlsx")) {
workbook = new XSSFWorkbook(inputStream);
} else if (fileExtensionName.equals(".xls")) {
workbook = new HSSFWorkbook(inputStream);
}
Sheet sheet = workbook.getSheet(SheetName);
//获得实际有多少行
int rowCount = sheet.getPhysicalNumberOfRows();
//rowCount = sheet.getLastRowNum() - sheet.getFirstRowNum() + 1;
//获得有数据的第一行有多少列
int colCount = sheet.getRow(sheet.getFirstRowNum()).getPhysicalNumberOfCells();
//获得有数据第一行第一列所在的位置
int rowColFirst = sheet.getRow(sheet.getFirstRowNum()).getFirstCellNum();
//创建一个Object二维数组,用来作为参数返回。
//因为第一行是说明,则跳过第一行,所以行数减一
Object[][] data = new Object[rowCount - 1][];
// 因为第一行是说明,则跳过第一行,所以行数减一
//sheet.getFirstRowNum() 获得第一行开始的位子,可能不从第一行开始(i=0)
//rowCount - 1 表示除去第一行说明,实际数据有多少行
for (int i = sheet.getFirstRowNum(), i1 = 0; i1 < rowCount - 1; i++, i1++) {
//得到第i+1行,即第二行的实际数据
Row row = sheet.getRow(i + 1);
// 创建一个字符数组,用来保存一行中,每一列的数据
String[] strings = new String[colCount];
// rowColFirst 不一定等于0,所以用k 来做为字符数组游标
for (int j = rowColFirst, k = 0; k < colCount; j++, k++) {
Cell cell = row.getCell(j);
//是否为空
if (cell != null) {
//是否为数字类型
if (cell.getCellType().equals(CellType.NUMERIC)) {
//是
//是否为时间格式
if (HSSFDateUtil.isCellDateFormatted(cell)) {
//是
strings[k] = new SimpleDateFormat("yyyy-MM-dd").format(cell.getDateCellValue());
} else {
row.getCell(j).setCellType(CellType.STRING);
//字符串类型
strings[k] = row.getCell(j).getStringCellValue();
}
} else if (cell.getCellType().equals(CellType.STRING)) {
row.getCell(j).setCellType(CellType.STRING);
//字符串类型
strings[k] = row.getCell(j).getStringCellValue();
}
} else {
//为空转换为空字符串
strings[k] = "";
}
}
//把获得数据添加到data
data[i1] = strings;
}
return data;
}
}
后言
当然,该方法不是一个很好的方法,有些还有些特殊情况处理不了。
如果你有什么改进或者更好的方法,欢迎留言。