JMeter性能测试-自定义Java脚本

使用JMeter做性能测试时，我们通常都是需要自己编写测试脚本，才能满足测试用例的需求，Jmeter也是支持我们自定义编写测试脚本的，而且支持Java、Python等多种语言。在这里，我们选择使用Java语言开发测试脚本，是因为Jmeter本身也是用Java实现的，所以使用Java开发脚本会更好。

首先，使用Eclipse或者IntelliJ IDEA来新建一个项目，然后新建一个包，再从Jmeter的lib/ext目录下中拷贝ApacheJMeter_java.jar和ApacheJMeter_core.jar两个文件，然后引入这两个JAR包到项目中。在IntelliJ IDEA中的操作如下：

1. 点击File->Project Structure
2. 点击Project Structure界面左侧的Modules
3. 在Dependencies标签界面下
4. 点击底部的+号，
5. 选择第一个选项JARs or directories…
6. 选择相应的jar包，点击OK

然后，我们在包中新建一个TestNum类，该类继承ApacheJMeter_java.jar包中的AbstractJavaSamplerClient类，引用它就可以调用了。在继承AbstractJavaSamplerClient类的时候，我们需要实现4个方法：

• setupTest()：初始化方法，用于初始化性能测试时的每个线程，实际运行时每个线程仅执行一次
• getDefaultParameters()：用于设置传入的参数，可以设置多个，已设置的参数会显示到Jmeter参数列表中
• runTest()：性能测试时的线程运行体，即测试执行的循环体，根据线程数和循环次数的不同可执行多次
• teardownTest()：测试结束方法，用于结束性能测试中的每个线程，实际运行时，每个线程仅执行一次，在测试方法运行结束后执行

具体的代码实现如下：

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;

public class TestNum extends AbstractJavaSamplerClient {
    // SampleResult类封装了从入口样本返回的各种信息
    private SampleResult results;
    // 输入的数字
    private String inNum;
    // 需要匹配的数字
    private String resultNum;

    // 用于设置传入的参数，可以设置多个，已设置的参数会显示到Jmeter参数列表中
    // Arguments类表示一组参数对象
    public Arguments getDefaultParameters() {
        // 声明定义Arguments类
        Arguments params = new Arguments();
        // 添加一个新参数
        params.addArgument("inNum", "");
        params.addArgument("resultNum", "66");
        return params;
    }

    // 初始化方法，用于初始化性能测试时的每个线程，实际运行时每个线程仅执行一次
    //
    // JavaSamplerContext类用于向JavaSamplerClient实现提供上下文信息
    // 这当前由在GUI中指定的初始化参数组成，其他数据将来可以通过JavaSamplerContext访问
    public void setupTest(JavaSamplerContext jsc) {
        // 定义SampleResult类
        results = new SampleResult();
        // 以String形式获取指定参数的值，或者如果未指定该值，则返回指定的默认值
        inNum = jsc.getParameter("inNum", "");
        resultNum = jsc.getParameter("resultNum", "");
        if (inNum != null && inNum.length() > 0) {
            // 设置采样器数据
            results.setSamplerData(inNum);
        }
        if (resultNum != null && resultNum.length() > 0) {
            results.setSamplerData(resultNum);
        }
    }

    // 性能测试时的线程运行体，即测试执行的循环体，根据线程数和循环次数的不同可执行多次
    public SampleResult runTest(JavaSamplerContext arg0) {
        boolean flag = false;
        // 定义一个事务，表示这是事务的起始点
        results.sampleStart();
        for (int i = inNum.length(); --i >= 0; ) {
            // 使用isDigit判断是否为数字，charAt方法用于返回指定索引处的字符
            if (!Character.isDigit(inNum.charAt(i))) {
                flag = false;
            } else {
                flag = true;
            }
        }
        for (int j = resultNum.length(); --j >= 0; ) {
            if (!Character.isDigit(resultNum.charAt(j))) {
                flag = false;
            } else {
                flag = true;
            }
        }
        // 定义一个事务，表示这是事务的结束点
        results.sampleEnd();
        if (flag) {
            Integer num = Integer.parseInt(inNum);
            Integer rsNum = Integer.parseInt(resultNum);
            if (num == rsNum) {
                // 设置SampleResult对象的编码和响应数据
                results.setResponseData("恭喜你，答对了O(∩_∩)O~\n答案是【" + resultNum + "】", "UTF-8");
                // 设置SampleResult对象的成功属性，即返回正确还是错误
                results.setSuccessful(true);
            } else if (num > rsNum) {
                results.setResponseData("好像大了点~~~~(>_<)~~~~ \n您输入的是【" + inNum + "】", "UTF-8");
                results.setSuccessful(false);
            } else {
                results.setResponseData("好像小了点~~~~(>_<)~~~~ \n您输入的是【" + inNum + "】", "UTF-8");
                results.setSuccessful(false);
            }
        } else {
            results.setResponseData("请输入数字：~~~~(>_<)~~~~ \n您输入的inNum是【" + inNum + "】，resultNum是【" + resultNum + "】", "UTF-8");
            results.setSuccessful(false);
        }
        return results;
    }

    // 测试结束方法，用于结束性能测试中的每个线程，实际运行时，每个线程仅执行一次，在测试方法运行结束后执行
    public void teardownTest(JavaSamplerContext arg0) {
    }
}

然后将上面的代码打包成jar包，生成的包名称为TestNum.jar，在IntelliJ IDEA中的操作如下：

1. 点击File->Project Structure
2. 点击Project Structure界面左侧的Artifact
3. 依次点击底部的+>JAR>Empty
4. 输入这个jar的Name，下面有这个jar包生成的路径Output directory
5. 在Output Layout中点击+>Module Output给这个jar包添加Module Output
6. 点击OK退出Project Structure面板
7. 最后依次点击Build>Build Artifacts…>TestNum>Build/Rebuild

生成jar包后，将jar包拷贝到Jmeter的lib/ext目录下。之后启动Jmeter，添加线程组和Java请求，同时为了方便，我们用随机数函数来随机抽取数字，验证与resultNum匹配的结果，再添加察看结果树和聚合报告。

设置线程数为10，执行后可以看到，10个请求中，有3个请求回答正确，响应数据对应了我们的代码，表示执行成功。

通过使用Jmeter自定义Java测试代码，再配合Jmeter自带的各种函数，就可以实现绝大多数的性能测试需求。