SQL查询指定字段自动写入CSV文件（供Jmeter性能测试参数化使用）

场景：
在使用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