操縱Excel文件的 ExcelUtil 類 ！

 

 

package FileDemo1;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.List;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

public class ExcleUtil {
    private static XSSFSheet ExcelWSheet;
    private static XSSFWorkbook ExcelWBook;
    private static XSSFCell Cell;
    private static XSSFRow Row;

    // 設定要設置的Excel的文件路徑和Excel 中Sheet名；
    // 在讀/寫Excel 的時候先要調用此方法
    public static void setExcleFile(String FilePath, String sheetName) throws Exception {
        FileInputStream ExcleFile;
        try {
            // 實例化Excle文件的FileInputStream 對象；
            ExcleFile = new FileInputStream(FilePath);
            // 實例化Excle文件的XSSFWorkbook 對象；
            ExcelWBook = new XSSFWorkbook(ExcleFile);
            /*
             * 實例化XSSFSheet 對象，指定ExcelFile中的sheet名稱，用於后續對sheet中行和列的操作；
             * 
             */
            ExcelWSheet = ExcelWBook.getSheet(sheetName);

        } catch (Exception e) {
            e.getStackTrace();
        }
    }
    /*
     * 讀取excle文件指定單元格的函數 ；
     * 
     */

    public static String getCell(int row, int col) throws Exception {

        try {
            // 通過函數參數指定單元格的行號和列，獲取指定單元格的對象；
            Cell = ExcelWSheet.getRow(row).getCell(col);
            /*
             * 1.如果單元格的類型為字符串類型，使用getStringCellValue();來獲取單元格的內容；
             * 2.如果單元格的類型為數字類型，使用getNumberricCellValue();來獲取單元格的內容；
             * 注意：getNumberricCellValue();返回的值為double類型，轉為為字符串類型，必須在
             * getNumberricCellValue();前面加上（" "）雙引號，用於強制轉換為String類型，不加雙引號
             * 則會拋錯；double類型無法轉換為String類型的異常；
             * 
             */
            String CellData = Cell.getCellType() == XSSFCell.CELL_TYPE_STRING ? Cell.getStringCellValue() + ""
                    : String.valueOf(Math.round(Cell.getNumericCellValue()));
            return CellData;
        } catch (Exception e) {
            e.getStackTrace();
            return "";
        }

    }
    /*
     * 在Excle中執行單元格寫入數據；
     * 
     * 
     */

    public static void setCellData(int rownum, int colnum, String Result) throws Exception {

        try {
            // 獲取excle文件的中行對象；
            Row = ExcelWSheet.getRow(rownum);
            // 如果單元格為空則返回null；
            Cell = Row.getCell(colnum, org.apache.poi.ss.usermodel.Row.RETURN_BLANK_AS_NULL);
            if (Cell == null) {
                // 當單元格為空是則創建單元格
                // 如果單元格為空無法調用單元格對象的setCellValue方法設定單元格的值 ；
                Cell = Row.createCell(colnum);
                // 創建單元格和后可以通過調用單元格對象的setCellValue方法設置單元格的值了；
                Cell.setCellValue(Result);
            } else {
                // 單元格中有內容，則可以直接調用單元格對象的 setCellValue 方法來設置單元格的值；
                Cell.setCellValue(Result);
            }
            FileOutputStream fileout = new FileOutputStream(Constant.testDataExcelFilePath);
            // 將內容寫到Excel文件中 ；
            ExcelWBook.write(fileout);
            // j調用flush方法強制刷新寫入文件；
            fileout.flush();
            fileout.close();

        } catch (Exception e) {
            System.out.println(e.getMessage() + e.getStackTrace());
            throw (e);
        }

    }

    public static void TangsetCellData(int RowNum, int ColNum, String Result) {
        try {
            // 獲取行對象
            Row = ExcelWSheet.getRow(RowNum);
            // 如果單元格為空，則返回null
            Cell = Row.getCell(ColNum, Row.RETURN_BLANK_AS_NULL);
            if (Cell == null) {
                // 當單元格對象是Null時，則創建單元格
                // 如果單元格為空，無法直接調用單元格的setCellValue方法設定單元格的值
                Cell = Row.createCell(RowNum);
                // 調用setCellValue方法設定單元格的值
                Cell.setCellValue(Result);
            } else {
                // 單元格中有內容，則可以直接調用seCellValue方法設定單元格的值
                Cell.setCellValue(Result);
            }
            // 實例化寫入Excel文件的文件輸出流對象
            FileOutputStream fileOut = new FileOutputStream(Constant.testDataExcelFilePath);
            // 將內容寫入Excel中
            ExcelWBook.write(fileOut);
            fileOut.flush();
            fileOut.close();
        } catch (Exception e) {
            // TODO: handle exception
            e.printStackTrace();
        }
    }

    // 從excel 文件中獲取測試數據的靜態方法；
    public static Object[][] getTestData(String excelFilePath, String sheetName) throws Exception {
        // 根據參數傳入的數據文件路徑和文件名稱，組合出Excel 數據文件的絕對路徑
        // 聲明一個文件；
        File file = new File(excelFilePath);
        // 創建FileInputStream 來讀取Excel文件內容；
        FileInputStream inputStream = new FileInputStream(file);
        // 聲明Workbook 對象；
        Workbook workbook = null;
        // 獲取文件名參數的擴展名，判斷是“.xlsx” 還是 “.xls” ；
        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);
        // 獲取Excel 數據文件Sheet1 中數據的行數，getLastRowNum 方法獲取數據的最后一行的行號，
        // getFistRowNum 獲取第一行 最后一行減去第一行就是總行數了
        // 注意excle 的行和列都是從0開始的；
        int rowCount = sheet.getLastRowNum() - sheet.getFirstRowNum();
        // 創建名為records 的List對象來存儲從Excel文件中讀取的數據；
        List<Object[]> records = new ArrayList<Object[]>();
        // 使用for循環遍歷Excel 數據文件的所有數據（除了第一行，第一行為標題行），所以i從1開始而不是從0開始；

        for (int i = 1; i < rowCount + 1; i++) {
            // 使用getRow來獲取行對象；
            Row row = sheet.getRow(i);
            /*
             * 聲明一個數據，用來存儲Excel數據文件每行中的測試用例和數據，數據的大小用getLastCellNum-2
             * 來進行動態聲明，實現測試數據個數和數組大小一致，
             * 因為Excel數據文件中的測試數據行的最后一個單元格是測試執行結果，倒數第二個單元格為此測試數據行是否運行的狀態位，
             * 所以最后倆列的單元格數據並
             * 不需要傳入測試方法中，所以是用getLastCellNum-2的方式去掉每行中的最后倆個單元格數據，計算出需要存儲的測試數據個數，
             * 並作為測試數據數組的初始化大小
             * 
             */
            String fields[] = new String[row.getLastCellNum() - 2];

            /*
             * 判斷數據行是否要參與測試的執行，Excel 文件的倒數第二列為數據行的狀態位， 標記為“y”
             * 表示此數據行要被測試腳本執行，標記為非“y”的數據行均被認為不會參數測試腳本執行，會被跳過；
             */

            if (row.getCell(row.getLastCellNum() - 2).getStringCellValue().equals("y")) {
                for (int j = 0; j < row.getLastCellNum() - 2; j++) {
                    /*
                     * 判斷Excel 單元格的內容是數字還是字符， 字符格式調用：
                     * row.getCell(j).getStringCellValue()；
                     * 數字格式調用：row.getCell(j).getNumericCellValue()；
                     */
                    fields[j] = (String) (row.getCell(j).getCellType() == XSSFCell.CELL_TYPE_STRING
                            ? row.getCell(j).getStringCellValue() : "" + row.getCell(j).getNumericCellValue());

                }
                // fields 存儲到數組當中；
                records.add(fields);

            }
        }

        /*
         * 定義函數的返回值，即Object[] [] 將存儲測試數據的list 轉換為一個Object 的二維數組；
         */
        Object[][] results = new Object[records.size()][];
        for (int i = 0; i < records.size(); i++) {
            results[i] = records.get(i);
        }

        return results;

    }

    public static int getLastColumnNum() {

        return ExcelWSheet.getRow(0).getLastCellNum() - 1;
    }

}