jmeter 自定义 java测试脚本

1.打开Java编译器,新建一个项目"TestLength",然后新建一个包"app".
2.从Jmeter的安装目录lib\ext中拷贝两个文件"ApacheJMeter_core.jar" 和"ApacheJMeter_java.jar"到"Tester"的项目中,然后引入这两个JAR文件.(具体的引入方法参考各个Java编译器的使用方法)
3.在"app"包中新建一个类,名字叫"TestLength",不过这个类要继承"AbstractJavaSamplerClient"类,如果项目引入步骤二中的两个文件,就可以找到"AbstractJavaSamplerClient"类了.
4."TestLength"类在继承"AbstractJavaSamplerClient"类的同时也会继承四个方法,分别是"getDefaultParameters","setupTest","runTest"和"teardownTest"方法."getDefaultParameters"方法主要用于设置传入的参数;"setupTest"方法为初始化方法,用于初始化性能测试时的每个线程."runTest"方法为性能测试时的线程运行体;"teardownTest"方法为测试结束方法,用于结束性能测试中的每个线程.

 

5、把上面的例子打包，然后把生成的"TestLength。jar"文件拷贝到Jmeter的安装目录lib\ext下。
6、运行Jmeter，添加一个线程组，然后在该线程组下面添加一个Java请求(在Sampler中)，在Java请求的类名称中选择咱们刚创建的类"app。TestLength"，在下面参数列表的"testStr"后面输入要测试的字符串，然后添加一个监听器(聚合报告)，设置一下模拟的用户数就可以测试了。如果测试不成功，Jmeter会在它自己个输出框中抛出这个字符串。
　　通过上面的例子我们可以发现，使用Jmeter自定义Java测试代码，配合Jmeter自带的函数，就可以实现出LoadRunner中"Java Vuser"协议的绝大多数功能。

 

附上代码：

package cn.gbase.jmeterTest;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Statement;

import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
import org.apache.jmeter.samplers.SampleResult;

/**
 *
 * @author tianmao
 */
public class JmeterExtendJDBCTest extends AbstractJavaSamplerClient {
    private SampleResult results;
    private String arg1;

    /* Implements JavaSamplerClient.setupTest(JavaSamplerContext) */
    public void setupTest(JavaSamplerContext arg0) {

        results = new SampleResult();
        results.setSamplerData(toString());
        results.setDataType("text");
        results.setContentType("text/plain");
        results.setDataEncoding("UTF-8");

        results.setSuccessful(true);
        results.setResponseMessageOK();
        results.setResponseCodeOK();

        arg1 = arg0.getParameter("arg1", "");
        if (arg1 != null && arg1.length() > 0) {
            results.setSamplerData(arg1);
        }
    }

    /* Implements JavaSamplerClient.teardownTest(JavaSamplerContext) */
    public void teardownTest(JavaSamplerContext arg0) {
        System.out.println("treaddown");
    }

    /* Implements JavaSamplerClient.getDefaultParameters() */
    public Arguments getDefaultParameters() {
        Arguments params = new Arguments();
        params.addArgument("arg1", "abc"); // 定义一个参数，显示到Jmeter的参数列表中，第一个参数为参数默认的显示名称，第二个参数为默认值
        return params;
    }

    @Override
    public SampleResult runTest(JavaSamplerContext context) {
        int len = 0;
        results.sampleStart(); // 定义一个事务，表示这是事务的起始点，类似于LoadRunner的
                                // lr.start_transaction
        arg1 = context.getParameter("arg1");
        doJDBCDeal();
        results.sampleEnd(); // 定义一个事务，表示这是事务的结束点，类似于LoadRunner的
                                // lr.end_transaction
        return results;
    }
    public void doJDBCDeal() {
        try {
            Class.forName("com.gbase.jdbc.Driver");
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }
        Connection conn = null;
        Statement stm = null;
        ResultSet rs = null;
        String sql = "select * from testView";
        try {
            conn = DriverManager.getConnection("jdbc:gbase://192.168.1.11/kgdx?user=gbase&password=gbase20110531&connectTimeout=300000");
            stm = conn.createStatement();
            rs = stm.executeQuery(sql);
            results.setResponseCode("testStr:" + arg1);
            results.setResponseMessage("aaaaaaaaaaaaaaa");

            results.setResponseCode("UTF-8");
            results.setResponseCodeOK();
            results.setResponseData(("execute:"+ sql+ "\r\n" + rs2Str(rs)).getBytes("utf-8"));
        } catch (Exception e) {
            results.setSamplerData(e.getMessage());
            results.setSuccessful(false); // 用于设置运行结果的成功或失败，如果是"false"则表示结果失败，否则则表示成功
            return;
        } finally {
            try {
                if (rs != null)
                    rs.close();
            } catch (SQLException e) {
            }
            try {
                if (stm != null)
                    stm.close();
            } catch (SQLException e) {
            }
            try {
                if (conn != null)
                    conn.close();
            } catch (SQLException e) {
            }
        }
        results.setSuccessful(true);
    }

    public String rs2Str(ResultSet rs) {
        StringBuilder sb = new StringBuilder();

        try {
            ResultSetMetaData rsmd = rs.getMetaData();
            int c = rsmd.getColumnCount();
            while (rs.next()) {
                sb.append("----------------------------------------------------------").append("\r\n");;
                for (int i = 0; i < c; i++) {
                    sb.append(rsmd.getColumnLabel(i + 1)).append("=").append(rs.getObject(i + 1)).append("\r\n");
                }
                sb.append("\r\n");
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return sb.toString();
    }
}