性能工具之JMeter二次开发总结

文章目录

- 一、前言
- 二、自定义脚本
- 三、自定义请求编写（Java Sampler）
- 四、自定义函数
- 五、小结

一、前言

掌握 JMeter 的脚本编写和执行，这基本已满足大部分的性能测试需求，但是面对各种各样的项目技术方案，有些需求是需要我们写代码实现的，那么性能测试工程师对 JMeter 二开就无法避免了，而 JMeter 作为 Apache 的开源性能测试工具允许使用者对其进行二次开发扩展，本文对常见的几种二开方法进行总结，方便大家学习查阅。

二、自定义脚本

JSR223 是一个用于将脚本嵌入 JMeter的Java源代码的框架。相当于一个小巧的 Java 源码解释器，简单来说就是可以在里面写代码，然后通过解释语言翻译成插件可以识别的指令去执行相关操作。

由于 JSR223 组件可用于增强对测试计划的控制，因此它们是使用 JMeter 进行性能测试的最有用工具之一。有一整套JMeter JSR223 组件：一个采样器，一个断言，前后之处理器，一个监听器和一个定时器。

在 JSR223 组件中使用一系列解释语言，常见的有 Groovy 和 BeanShell。

这里以 BeanShell 举例，其常被用于对请求或者返回内容进行获取或者加工，其中 prev 是对当前的取样进行访问，执行了对响应状态码、响应信息、请求头等的操作，

常见用法举例：

#获取响应的状态码
log.info("code is  "+prev.getResponseCode());

#获取响应信息
log.info("response is "+prev.getResponseDataAsString());

#获取头文件中ContentType类型
log.info("content_type  "+prev.getContentType());

#获取取样器请求首部字段
log.info("header "+prev.getRequestHeaders());

通过以上方式，基本实现了对请求的基本信息的获取，然后我们可以对这些信息做进一步的提取、判断等操作。

另外，我们可以使用 log 变量访问组件的 Logger 对象。Logger 类对于脚本编写非常重要，因为可以让我们更轻松地调试脚本，通过信息输出到 JMeter 日志，方便我们调试脚本。

BeanShell 日志打印示例如下：

#打印线程号
log.debug('Current thread number:'+ String.valueOf(ctx.getThreadNum()));

#打印启动采样器名称
log.info('Sampler '+ sampler.getName()+' started’);

#自定义告警
log.warn('Could not convert string, using default encoding');

#打印错误信息
log.error("Malformed URL detected:", e);

#打印响应数据
log.error("Response data:"+ prev.getResponseDataAsString());

其它使用方式可以参考本文：性能工具之 JMeter 快速入门

注意在 JMeter 高并发时不建议使用 BeanShell ，因为其消耗资源较多，相比 Groovy 要更高效。

三、自定义请求编写（Java Sampler）

实现 Java Sampler 自定义请求的两种方式

继承 AbstractJavaSamplerClient 抽象类；
实现 JavaSamplerClient 接口。

通过阅读源码可以发现 AbstractJavaSamplerClient 抽象类是 JavaSamplerClient 接口的子类，所以，我们可以新建一个 JavaClass，并继承 AbstractJavaSamplerClient。

AbstractJavaSamplerClient中默认实现了四个可以覆盖的方法，分别是 getDefaultParameters()， setupTest()， runTest()和 teardownTest()方法。

getDefaultParameters 方法：主要用于设置传入界面的参数，这个方法由 JMeter 在进行添加 JavaRequest 时第一个运行，它决定了你要在 GUI 中默认显示哪些属性。当每次在 GUI 里点击建立 java requst sampler 的时候会调用该方法。该方法设置了 parameters 的初始值，也可以在 sampler 的 GUI 界面做进一步的修改；
setupTest方法：初始化方法，用于初始化性能测试时的每个线程，用于初始化测试脚本里面用到的变量，这些变量会在后续执行中使用。
runTest方法：性能测试时的线程运行体，实现功能逻辑的主方法，每个线程会循环执行这个方法。
teardownTest：方法为测试结束方法，用于结束性能测试中的每个线程，在每个线程执行完所有的测试工作之后执行，有点像 finally 的功能，比如，我开了一个数据库的连接，那么我要在所有的线程完成工作后关闭。

使用案例：