通过Loadrunner读取excel表数据并批量更新到Oracle中

本文以Loadrunner的Java_Vuser脚本为例,来做一个简化版的自动化测试框架(以excel作为数据驱动),实现批量更新Oracle业务数据库的目的,通过本文例子我们还可以实现将Loadrunner由性能测试工具,转换成一个接口自动化测试工具(Loadrunner的多用户和循环action脚本模式,是多么类似单元测试工具的Test Case调用,既@Test模式)。

1、在loadrunner中新建脚本(本文以LoadRunner11为例),要求选择协议类型为Java->Java Vuser

2、在Run-time Settings设置JDK路径,由于LoadRunner11不支持jdk1.8,本次测试是拷贝了一份低版本的JDK1.6,所以路径选择固定路径模式,如下所示:

通过Loadrunner读取excel表数据并批量更新到Oracle中_第1张图片

3、上网下载一份Oracle的JAR包和jxl的JAR包(官网和网上都能搜到),我已经把两包加上mysql的驱动包都放一起,有兴趣的可以直接下载:http://download.csdn.net/detail/smooth00/9875368,将JAR包放到Loadrunner的include目录下或其它指定目录下,并在Run-time Settings中配置Classpath

通过Loadrunner读取excel表数据并批量更新到Oracle中_第2张图片

4、建立excel文件(要求另存为Excel97-2003的文件格式,保存的路径是在LR脚本目录下新建一个目录DataSource,将excel放到这个目录下),并按以下数据格式创建参数表(每个Sheet以真实表命名,colName是表的字段表,colValue是要更新的字段值):

通过Loadrunner读取excel表数据并批量更新到Oracle中_第3张图片

5、在Loadrunner中以Java Vuser协议创建脚本,脚本样例如下:

/*

* LoadRunner Java script. (Build: _build_number_)

* Script Description:         

*/

import java.io.File;

import java.io.IOException;

import java.sql.Connection; 

import java.sql.DriverManager; 

import java.sql.PreparedStatement; 

import java.sql.ResultSet; 

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.HashMap;

import jxl.Cell;

import jxl.read.biff.BiffException;

import jxl.Sheet;

import jxl.Workbook;

import lrapi.lr;

public class Actions

{

    //设定数据库驱动,数据库连接地址、端口、名称,用户名,密码 

        String driverName="oracle.jdbc.OracleDriver"; 

        String url="jdbc:oracle:thin:@172.16.1.65:1521:orcl";  //orcl为数据库名称,1521为连接数据库的默认端口 

        String user="test";  //用户名 

        String password="123456";  //密码 

        PreparedStatement pstmt = null; 

        ResultSet rs = null;

        //数据库连接对象 

        Connection conn = null;

    public Workbook workbook;

    public Sheet sheet;

    public Cell cell;

    int rows;

    int columns;

    //public String fileName;

    //public String caseName;

    public ArrayList arrkey = new ArrayList();

    static String sourceFile;

public void connection(){//连接Oracle数据库

    try {

//反射Oracle数据库驱动程序类 

Class.forName(driverName); 


//获取数据库连接 

conn = DriverManager.getConnection(url, user, password); 


//输出数据库连接 

System.out.println(conn);

    } catch (ClassNotFoundException e) { 

e.printStackTrace(); 

    } catch (SQLException e) { 

e.printStackTrace(); 

    }

}

public void deconnection(){//断开数据库连接

    try{ 

    if(rs != null){ 

rs.close(); 

    } 

    if(pstmt != null){ 

pstmt.close(); 

    } 

    if(conn != null){ 

conn.close(); 

    } 

} catch (SQLException e) { 

    e.printStackTrace(); 

}   

}

    /**

    * 以HashMap获得excel表中的数据

    * @caseName Excel的Sheet名

    * @fileName Excel的文件名

    */

    public Object[][] getExcelData(String caseName,String fileName) throws BiffException, IOException {

        workbook = Workbook.getWorkbook(new File(getPath(fileName)));

        sheet = workbook.getSheet(caseName);

        rows = sheet.getRows();

        columns = sheet.getColumns();

        // 为了返回值是Object[][],定义一个多行单列的二维数组

        HashMap[][] arrmap = new HashMap[rows - 1][1];

        // 对数组中所有元素hashmap进行初始化

        if (rows > 1) {

            for (int i = 0; i < rows - 1; i++) {

                arrmap[i][0] = new HashMap();

            }

        } else {

            System.out.println("excel中没有数据");

        }

        // 获得首行的列名,作为hashmap的key值

        for (int c = 0; c < columns; c++) {

            String cellvalue = sheet.getCell(c, 0).getContents();

            arrkey.add(cellvalue);

        }

        // 遍历所有的单元格的值添加到hashmap中

        for (int r = 1; r < rows; r++) {

            for (int c = 0; c < columns; c++) {

                String cellvalue = sheet.getCell(c, r).getContents();

                arrmap[r - 1][0].put(arrkey.get(c), cellvalue);

            }

        }

        return arrmap;

    }

    /**

    * 获得excel文件的路径

    * @return

    * @throws IOException

    */

    public String getPath(String fileName) throws IOException {

        File directory = new File(".");

        sourceFile = directory.getCanonicalPath() + "\\DataSource\\"

                + fileName + ".xls";

        return sourceFile;

    }

    /**

    * 以ArrayList格式获得excel表中的数据

    * @caseName Excel的Sheet名

    * @fileName Excel的文件名

    */

    public ArrayList GetExcelParameter(String caseName,String fileName){

ArrayList parameters = new ArrayList();

try {

Workbook workbook = Workbook.getWorkbook(new File(getPath(fileName)));

Sheet sheet = workbook.getSheet(caseName);

for (int i = 1; i < sheet.getRows(); i++) {//不包括第一行

    for (int j = 0; j < sheet.getColumns(); j++) {

        parameters.add(sheet.getCell(j, i).getContents()); 

System.out.println(parameters.get(j)+"  "); 

    }

}

} catch (BiffException e) {

// TODO Auto-generated catch block

e.printStackTrace();

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

}

return parameters;

}

/**

  * 批量更新表中的指定字段(excel中每个Sheet是一张表,Sheet名与表名同名,第一列ID序号,第二列字段名,第三列是需要更新的字段值)

  * @TableName Excel的Sheet名(设置成与数据表同名)

  * @ColumnIDs 需要修改的字段名,用@分隔不同字段

  * @fileName Excel的文件名

  */

public void updateField(String TableName,String ColumnIDs,String fileName){

    try{

    Workbook workbook = Workbook.getWorkbook(new File(getPath(fileName)));

    Sheet sheet = workbook.getSheet(TableName);

    String[] ColumnID=ColumnIDs.split("@");

    String update_sql="update "+TableName+" Set ";

    for(int k=0;k

for (int i = 1; i < sheet.getRows(); i++) {//不包括第一行

    String ColumnName=sheet.getCell(1, i).getContents();

    String ColumnNameValue=sheet.getCell(2, i).getContents();

    //lr.output_message("The parameter1's value is "+ColumnName+" "+ColumnID[i]);

    if(ColumnName.equals(ColumnID[k])){

if(k==ColumnID.length-1)

  update_sql+=ColumnName+"='"+ColumnNameValue+"' WHERE ROWNUM<=10";//修改前10条

else

    update_sql+=ColumnName+"='"+ColumnNameValue+"',";

    //lr.output_message("The parameter1's value is "+ColumnName);

break;

    }   

}

    }

    System.out.println("【" + update_sql + "】");

    //创建该连接下的PreparedStatement对象 

    pstmt = conn.prepareStatement(update_sql);

    //循环获取ResultSet对象中信息

        rs = pstmt.executeQuery();           

    } catch (Exception e) { 

e.printStackTrace(); 

    }

}

/**

  * 批量修改指定Sheet(指定表名)中的字段,按照表中第一条记录更新第一个字段,第二条记录更新第二个字段,以此类推

  * @TableName Excel的Sheet名(设置成与数据表同名)

  * @fileName Excel的文件名

  */

public void updateData(String TableName,String fileName){

try{

Workbook workbook = Workbook.getWorkbook(new File(getPath(fileName)));

Sheet sheet = workbook.getSheet(TableName);

String update_sql=null;

for (int i = 1; i < sheet.getRows(); i++) {//不包括第一行

    PreparedStatement pstmt2 = null;  //将数据集定义PreparedStatement放到for循环内,效率不高,但为了避免游标数超限,只能这么做

    ResultSet rs2 = null;

    String ColumnName=sheet.getCell(1, i).getContents();

    String ColumnNameValue=sheet.getCell(2, i).getContents();

    update_sql="update "+TableName+" Set "+ColumnName+"='"+ColumnNameValue+"' where rowid in ( Select rid from (Select Rownum r_n,rowid rid,a.* from "+TableName+" a)where r_n="+i+")";//修改第i条

    System.out.println("【" + update_sql + "】");

    //创建该连接下的PreparedStatement对象 

    pstmt2 = conn.prepareStatement(update_sql);

    //循环获取ResultSet对象中信息

    rs2 = pstmt2.executeQuery();

    if(rs2 != null){ 

rs2.close(); 

    } 

    if(pstmt2 != null){ 

pstmt2.close(); 

    }

}


  } catch (Exception e) { 

e.printStackTrace(); 

  }

}

/**

  * 循环调用整个excel中的不同Sheet参数表,并按上一函数的要求更新数据

  * @fileName Excel的文件名

  */

public void updateData(String fileName){

    try{

Workbook workbook = Workbook.getWorkbook(new File(getPath(fileName)));

Sheet[] sheets = workbook.getSheets();// 获取所有的sheet

String sheetname = null;

for (int x = 0; x < sheets.length; x++) { 

  Sheet sheet = workbook.getSheet(x);// 获取sheet

  if(sheet.getName().toString().indexOf("Sheet")!=-1){//循环到以Sheet命名的附表则退出

    break;

  }else{

updateData(sheet.getName());

  }

}     

    } catch (Exception e) { 

e.printStackTrace(); 

    }

}

public int init() throws Throwable {

  connection();

return 0;

}//end of init

public int action() throws Throwable {

    //Object[][] getExcelOb=getExcelData("GTDJ_BGSX_LC","DataTable");

    //System.out.println("【" + getExcelOb + "】");

    //ArrayList gextExcelPa=GetExcelParameter("GTDJ_BGSX_LC","DataTable");

    //System.out.println("【" + gextExcelPa + "】");

    // -----------------------------------------------------------------------------

    //updateField("GTDJ_BGSX_LC","RECID@OPENO@PRIPID@ALTITEM","DataTable");//修改指定字段,用@分隔字段名

    //updateData("GTDJ_BGSX_LC","DataTable");//修改不同的字段-每条记录只改一个字段(指定表名)

    updateData("DataTable");//修改excel所有表的不同字段-每条记录只改一个字段

    //--------------------------------------------------------------------------


return 0;

}//end of action

public int end() throws Throwable {

    deconnection();

return 0;

}//end of end

}

以上脚本仅供参考,主要目的是为了表达Loadrunner所提供的Java和.Net调用接口,可以实现用数据驱动的方式对Java和.Net进行自动化测试,而且可以适当扩展成单元测试工具,毕竟工具都是死的,人和思想才是活的。

你可能感兴趣的:(通过Loadrunner读取excel表数据并批量更新到Oracle中)