场景：
在使用Jmeter 做性能测试时，大量场景中的字段需要参数化。总结有以下三种方式可以处理。
1、从每次jmeter请求的响应结果中断言获取
2、jmeter连接数据库，创建jdbc request，获取sql查询结果
3、手动准备需要参数化的字段值的CSV文件
前两种方式，耗时，不能达到性能测试的预期效果。第三种方式，如果场景较多，数据量大时，也很繁琐。所以，能不能将从数据库中查询出指定需要参数化字段的值直接写入csv文件呢？这种方法，方便省时省力，还可以重复使用。
以下介绍详细步骤。
1、准备各场景需要参数化字段的对应sql。可以将所有的sql放在一个文件中然后读取（方便后期维护或替换）。这里直接使用常量，即定义类文件，将所有sql分别定义为静态常量。即下面的SqlList类
public class SqlList {
//1.打开定损单
public static final String sql_011 =" select *\n" +
" from (select tsu.account,\n" +
" tc.claim_id,\n" +
" UTL_RAW.CAST_TO_VARCHAR2(UTL_ENCODE.BASE64_ENCODE(UTL_RAW.CAST_TO_RAW(tc.claim_id)))as encode\n" +
" from t_sys_user tsu\n" +
" join t_claim_wf twf on twf.user_id = tsu.user_id\n" +
" join t_claim tc on tc.claim_id = twf.claim_id\n" +
" where \n" +
" tsu.account ='vip'\n" +
" and tsu.company_id = 2373\n" +
" and twf.claim_status = '06')\n" +
" order by dbms_random.value()";
}
2、创建JDBC连接（JDBCUtils类），执行指定sql语句。将查询结果存放到List中，因为数据是二维的，所以要创建List如： ArrayList<ArrayList<String>>。这里每个sql都指定了要查询的字段，故在向List中添加元素时，直接使用Arrays.asList。如：data.add(new ArrayList<String>(Arrays.asList(resultSet.getString(col1), resultSet.getString(col2), resultSet.getString(col3))))
JDBCUtils类：
import java.io.IOException;
import java.io.InputStream;
import java.sql.*;
import java.util.Properties;
public class JDBCUtils {
private static Connection connection;
private static Statement statement;
static String driverClass = null;
static String url = null;
static String username = null;
static String password = null;
public JDBCUtils(){}
static {
try {
InputStream in = JDBCUtils.class.getClassLoader().getResourceAsStream("jdbc.properties");
Properties pro = new Properties();
pro.load(in);
driverClass = pro.getProperty("driverClass");
url = pro.getProperty("url");
username = pro.getProperty("username");
password = pro.getProperty("password");
Class.forName(driverClass);
connection = DriverManager.getConnection(url,username,password);
statement = connection.createStatement();
} catch (ClassNotFoundException e) {
e.printStackTrace();
} catch (SQLException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
public static Connection getConnection(){
return connection;
}
public static ResultSet selectSingle(String sql) {
ResultSet resultSet = null;
try {
resultSet = statement.executeQuery(sql);
} catch (SQLException e) {
e.printStackTrace();
}
return resultSet;
}
public static void close(){
try {
statement.close();
} catch (SQLException e) {
e.printStackTrace();
}finally {
try {
connection.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
}
}
JDBCData 类：
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Arrays;
public class JDBCData {
public static ArrayList<ArrayList<String>> selectCase(String col1,String col2,String col3) {
String sql = SqlList.sql_011;
ResultSet resultSet = JDBCUtils.selectSingle(sql);
ArrayList<ArrayList<String>> data=new ArrayList<ArrayList<String>>();
try {
while (resultSet.next()){
data.add(new ArrayList<String>(Arrays.asList(resultSet.getString(col1), resultSet.getString(col2), resultSet.getString(col3))));
}
} catch (SQLException e) {
e.printStackTrace();
}finally {
JDBCUtils.close();
}
return data;
}
}
3、将上述List集合，通过字符流的方式写入到指定的CSV文件中。针对每列，每个参数值之间以逗号分隔，所以需要在遍历List集合时，判断参数值是不是最后一列，如果不是就在参数值后写逗号，反之则不需要。针对每行，判断参数值是不是最后一行，如果不是就在每行写完后换行，反之则不需要。
CSVUtils类：
import java.io.BufferedWriter;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
public class CSVUtils {
//导出到csv文件
public static int Array2CSV(ArrayList<ArrayList<String>> data, String path)
{
try {
//使用字符流将List中的内容写入到文件中
BufferedWriter out =new BufferedWriter(new OutputStreamWriter(new FileOutputStream(path),"UTF-8"));
for (int i = 0; i < data.size(); i++)
{
ArrayList<String> onerow=data.get(i);
for (int j = 0; j < onerow.size(); j++)
{
//如果不是最后一列，就在字段值后面追加逗号;是最后一列，就不追加逗号
if(j != (onerow.size()-1)){
out.write(onerow.get(j));
out.write(",");
}else {
out.write(onerow.get(j));
}
}
if(i !=(data.size()-1)){
out.newLine();
}
}
out.flush();
out.close();
} catch (Exception e) {
e.printStackTrace();
}
return data.size();
}
}
4、测试，调用第3步中的写入CSV方法，生成CSV文件。
SaveCSV类：
public class SaveCSV {
static int rows = 0;
public static void main(String[] args) {
test();
}
public static void test() {
rows = CSVUtils.Array2CSV(JDBCData.selectCase("account","claim_id","encode"),"D:/workspaces/jmeter script/定损_打开定损单_测试数据.csv");
System.out.println("01_定损_打开定损单_测试数据"+"---------"+rows);
}
}
用记事本打开生成的CSV文件如下
image.png
网友评论