使用POI讀取Excel表格中數據 2

使用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;
    }

}

後言

當然，該方法不是一個很好的方法，有些還有些特殊情況處理不了。
如果你有什麼改進或者更好的方法，歡迎留言。