基于 JMeter API 开发性能测试平台

V2AS问路

基于 JMeter API 开发性能测试平台

阅读原文时间：2023年08月30日阅读：3

背景：

JMeter 是一个功能强大的性能测试工具，若开发一个性能测试平台，用它作为底层执行引擎在合适不过。如要使用其API，就不得不对JMeter 整个执行流程，常见的类有清楚的了解。

常用的 JMeter 类和功能的解释：

TestPlan 类：代表一个测试计划，它是性能测试的顶级元素。您可以使用它来设置全局的测试属性，如测试名称、线程组、监听器等。
ThreadGroup 类：代表线程组，它定义了并发执行的线程数、启动延迟时间、循环次数等。线程组是性能测试的基本单位，所有的线程都在线程组内执行。
LoopController 类：代表循环控制器，它定义了循环的次数或条件。可以将循环控制器添加到线程组中，以便在每个线程中执行多次循环。
HTTPSampler 类：代表 HTTP 请求采样器，用于发送 HTTP 请求。您可以设置请求的 URL、方法（GET、POST 等）、请求参数、请求头等。
StandardJMeterEngine 类：代表 JMeter 引擎，负责配置和执行测试计划。通过将测试计划配置给引擎，可以启动和执行性能测试。
JMeterUtils 类：提供了一些 JMeter 的实用方法和属性。通过加载 JMeter 属性文件并初始化本地设置，可以确保 JMeter API 的正确运行。

　　为了能够在代码中创建和配置测试计划、线程组、循环控制器和其他元素，以及执行性能测试，这些类的功能和方法是我们必须要熟悉的。掌握了这些类，那么利用jmeter如何编写性能测

试脚本是怎样的，我们也来回顾下：

JMeter 编写性能测试脚本的大致流程示意图：

创建测试计划（Test Plan） --> 添加线程组(Thread Group) --> 添加配置元件（Config Elements）--> 添加逻辑控制器（Logic Controllers)

--> 添加取样器（Samplers）和断言（Assertions）--> 添加监听器（Listeners）--> 运行测试计划

源码实现方式：

(1) 环境初始化

需要在项目中引入上述JMeter核心API，这样我们才能使用 JMeter 提供的各种功能和类

import org.apache.jmeter.control.LoopController;
import org.apache.jmeter.engine.StandardJMeterEngine;
import org.apache.jmeter.protocol.http.sampler.HTTPSampler;
import org.apache.jmeter.testelement.TestPlan;
import org.apache.jmeter.threads.ThreadGroup;
import org.apache.jmeter.util.JMeterUtils;

(2) 环境初始化

JMeter API 在执行过程中，首先会读取 JMeter 软件安装目录文件下配置文件里的属性，所以我们要通过 JMeter API 读取指定的 JMeter 主配置文件的目录以及 JMeter 的安装目录，

其中环境初始化主要包括以下 2 个步骤：

通过 JMeterUtils.loadJMeterProperties 加载 JMeter 主配置文件 JMeter.properties，然后把 jmeter.properties 中的所有属性赋值给 JMeterUtils 对象，以便之后获取所需的配置；
设置 JMeter 的安装目录，JMeter API 会根据我们指定的目录加载脚本运行时需要的配置

JMeterUtils.loadJMeterProperties("jmeter.properties");
JMeterUtils.setJMeterHome(~/jemter)
JMeterUtils.initLocale();

(3) 创建测试计划

使用 TestPlan 类创建一个测试计划，并设置相关属性。

TestPlan testPlan = new TestPlan("My Test Plan");
testPlan.setProperty(TestElement.TEST_CLASS, TestPlan.class.getName());
testPlan.setProperty(TestElement.GUI_CLASS, TestPlanGui.class.getName());

(4) 创建 ThreadGroup

使用 ThreadGroup 类创建一个线程组，并设置相关属性，例如线程数、循环次数等

ThreadGroup threadGroup = new ThreadGroup();
threadGroup.setName("My Thread Group");
threadGroup.setNumThreads(10);
threadGroup.setRampUp(5);
threadGroup.setSamplerController(loopController);

(5) 创建循环控制器

这一步是一个可选项。我们在实际测试过程中，创建循环控制器是为了模拟一个用户多次进行同样操作的行为，不创建循环控制器则默认是只执行一次操作。循环控制器创建的代码如下：

LoopController loopController = new LoopController();

//设置循环次数，1 代表循环 1 次

loopController.setLoops(1);

loopController.setFirst(true);

loopController.setProperty(TestElement.TEST\_CLASS, LoopController.class.getName());

loopController.setProperty(TestElement.GUI\_CLASS, LoopControlPanel.class.getName());

loopController.initialize()

(6) 创建 Sampler

使用 HTTPSampler 类创建一个 HTTP 请求采样器，并设置请求的 URL、方法等属性

HTTPSampler httpSampler = new HTTPSampler();
httpSampler.setDomain("example.com");
httpSampler.setPort(80);
httpSampler.setPath("/api/endpoint");
httpSampler.setMethod("GET");

(7) 创建结果收集器

结果收集器可以保存每次 Sampler 操作完成之后的结果的相关数据，例如，每次接口请求返回的状态、服务器响应的数据。

ResultCollector resultCollector = new ResultCollector();

resultCollector.setName(ResultCollector.class.getName());

(8) 构建 tree，生成 jmx 脚本

以上步骤其实都是为了创建了一个 HashTree 节点做准备。把创建都添加到子 HashTree 节点中，然后把子 HashTree 加到 testplan 中，最后再把 tesplan 节点加到构建好的父 HashTree 节点，这样就生成了我们的脚本可执行文件 jmx。代HashTree subTree = new HashTree(

subTree.add(httpSamplerProxy);

subTree.add(loopController);

subTree.add(threadGroup);

subTree.add(resultCollector);

HashTree tree = new HashTree();

tree.add(testPlan,subTree);

SaveService.saveTree(tree, new FileOutputStream("test.jmx"));

} catch (IOException e) {