使用POI读取Excel表格中数据 2

待我称王封你为后i 2023-02-21 12:58 139阅读 0赞

使用POI读取Excel表格中数据 2

背景

自己最近在研究自动换测试,在进行参数化时候,会对把用例写在Excel表格中,然后通过Java来取表格中数据,进而进行参数化。

问题

在使用POI读取Excel表格中数据 1中,如下图情况,如果,现在出现不适第一行,第一列的情况,我们又该怎么解决,如第二张图。

在这里插入图片描述
在这里插入图片描述
表格中的单元格里出现没有输入的情况(图片中红色部分),如何通过POI进行读取,或者如何处理?预期结果不用考虑。

代码

代码优化。
只贴了获取表格中数据代码,其他不是本文内容。

  1. import org.apache.poi.hssf.usermodel.HSSFDateUtil;
  2. import org.apache.poi.hssf.usermodel.HSSFWorkbook;
  3. import org.apache.poi.ss.usermodel.*;
  4. import org.apache.poi.xssf.usermodel.XSSFRow;
  5. import org.apache.poi.xssf.usermodel.XSSFSheet;
  6. import org.apache.poi.xssf.usermodel.XSSFWorkbook;
  7. import java.io.File;
  8. import java.io.FileInputStream;
  9. import java.io.IOException;
  10. import java.text.SimpleDateFormat;
  11. public class ExcelUtil {
  12. public static Object[][] getTestData(String excelFilePath, String SheetName) throws IOException {
  13. File file = new File(excelFilePath);
  14. FileInputStream inputStream = new FileInputStream(file);
  15. Workbook workbook = null;
  16. //获取文件扩展名
  17. String fileExtensionName = excelFilePath.substring(excelFilePath.indexOf("."));
  18. if (fileExtensionName.equals(".xlsx")) {
  19. workbook = new XSSFWorkbook(inputStream);
  20. } else if (fileExtensionName.equals(".xls")) {
  21. workbook = new HSSFWorkbook(inputStream);
  22. }
  23. Sheet sheet = workbook.getSheet(SheetName);
  24. //获得实际有多少行
  25. int rowCount = sheet.getPhysicalNumberOfRows();
  26. //rowCount = sheet.getLastRowNum() - sheet.getFirstRowNum() + 1;
  27. //获得有数据的第一行有多少列
  28. int colCount = sheet.getRow(sheet.getFirstRowNum()).getPhysicalNumberOfCells();
  29. //获得有数据第一行第一列所在的位置
  30. int rowColFirst = sheet.getRow(sheet.getFirstRowNum()).getFirstCellNum();
  31. //创建一个Object二维数组,用来作为参数返回。
  32. //因为第一行是说明,则跳过第一行,所以行数减一
  33. Object[][] data = new Object[rowCount - 1][];
  34. // 因为第一行是说明,则跳过第一行,所以行数减一
  35. //sheet.getFirstRowNum() 获得第一行开始的位子,可能不从第一行开始(i=0)
  36. //rowCount - 1 表示除去第一行说明,实际数据有多少行
  37. for (int i = sheet.getFirstRowNum(), i1 = 0; i1 < rowCount - 1; i++, i1++) {
  38. //得到第i+1行,即第二行的实际数据
  39. Row row = sheet.getRow(i + 1);
  40. // 创建一个字符数组,用来保存一行中,每一列的数据
  41. String[] strings = new String[colCount];
  42. // rowColFirst 不一定等于0,所以用k 来做为字符数组游标
  43. for (int j = rowColFirst, k = 0; k < colCount; j++, k++) {
  44. Cell cell = row.getCell(j);
  45. //是否为空
  46. if (cell != null) {
  47. //是否为数字类型
  48. if (cell.getCellType().equals(CellType.NUMERIC)) {
  49. //是
  50. //是否为时间格式
  51. if (HSSFDateUtil.isCellDateFormatted(cell)) {
  52. //是
  53. strings[k] = new SimpleDateFormat("yyyy-MM-dd").format(cell.getDateCellValue());
  54. } else {
  55. row.getCell(j).setCellType(CellType.STRING);
  56. //字符串类型
  57. strings[k] = row.getCell(j).getStringCellValue();
  58. }
  59. } else if (cell.getCellType().equals(CellType.STRING)) {
  60. row.getCell(j).setCellType(CellType.STRING);
  61. //字符串类型
  62. strings[k] = row.getCell(j).getStringCellValue();
  63. }
  64. } else {
  65. //为空转换为空字符串
  66. strings[k] = "";
  67. }
  68. }
  69. //把获得数据添加到data
  70. data[i1] = strings;
  71. }
  72. return data;
  73. }
  74. }

后言

当然,该方法不是一个很好的方法,有些还有些特殊情况处理不了。
如果你有什么改进或者更好的方法,欢迎留言。

发表评论

表情:
评论列表 (有 0 条评论,139人围观)

还没有评论,来说两句吧...

相关阅读