Selenium自动化-调用Mysql数据库

上几篇博客发布了几篇Selenium入门知识和进阶，

　　现在附上如何 从数据库中取值

　　　　能够逐行取值，并且返回二维数组

 

import java.io.FileInputStream;
import java.io.IOException;
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.XSSFWorkbook;

public class MyExcel {
    public static Object[][] getExcelData(String filePath, String fileName, String sheetName) throws IOException {
        java.io.File file = new java.io.File(filePath + fileName);
        FileInputStream inputStream = new FileInputStream(file);
        Workbook workbook = null;
        String fileExtensionName = fileName.substring(fileName.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.getLastRowNum() - sheet.getFirstRowNum();
        List<Object> records = new ArrayList<Object>();
        for (int i = 1; i < rowCount + 1; i++) {
            Row row = sheet.getRow(i);
            String fields[] = new String[row.getLastCellNum()];
            for (int j = 0; j < row.getLastCellNum(); j++) {
                try {
                    fields[j] = row.getCell(j).getStringCellValue();
                }
                catch (Exception e) {
                    fields[j] = null;
                }
            }
            records.add(fields);
        }
        Object[][] results = new Object[records.size()][];
        for (int i = 0; i < records.size(); i++) {
            results[i] = (Object[]) records.get(i);
        }
        return results;
    }

}