做性能测试时，我们通常都是需要自己编写测试脚本，才能满足测试用例的需求，Jmeter也是支持我们自定义编写测试脚本的，而且支持Java、Python等多种语言。在这里，我们选择使用Java语言开发测试脚本，是因为Jmeter本身也是用Java实现的，所以使用Java开发脚本会更好。
首先，使用Eclipse或者IntelliJ IDEA来新建一个项目，然后新建一个包，再从Jmeter的lib/ext目录下中拷贝ApacheJMeter_java.jar和ApacheJMeter_core.jar两个文件，然后引入这两个JAR包到项目中。在IntelliJ IDEA中的操作如下：
- 点击File->Project Structure
- 点击Project Structure界面左侧的Modules
- 在Dependencies标签界面下
- 点击底部的+号，
- 选择第一个选项JARs or directories...
- 选择相应的jar包，点击OK
IntelliJ IDEA下的导入操作
然后，我们在包中新建一个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中的操作如下：
- 点击File->Project Structure
- 点击Project Structure界面左侧的Artifact
- 依次点击底部的+>JAR>Empty
- 输入这个jar的Name，下面有这个jar包生成的路径Output directory
- 在Output Layout中点击+>Module Output给这个jar包添加Module Output
- 点击OK退出Project Structure面板
- 最后依次点击Build>Build Artifacts...>TestNum>Build/Rebuild
生成jar包后，将jar包拷贝到Jmeter的lib/ext目录下。之后启动Jmeter，添加线程组和Java请求，同时为了方便，我们用随机数函数来随机抽取数字，验证与resultNum匹配的结果，再添加察看结果树和聚合报告。
配置自定义Java脚本
设置线程数为10，执行后可以看到，10个请求中，有3个请求回答正确，响应数据对应了我们的代码，表示执行成功。
自定义Java脚本执行结果
通过使用Jmeter自定义Java测试代码，再配合Jmeter自带的各种函数，就可以实现绝大多数的性能测试需求。
网友评论