Jmeter作为基于java开源的压测工具,其扩展性是很便捷的。之前自己编写实践了如何扩展jmeter的function函数,大家可以按照如下方法尽情的扩展自己想要实现的任何基础函数。


1、总体思路

扩展JMeter的function可以分成下面几个步骤:

1、在Eclipse中新建Maven项目,引入扩展JMeter function所需的依赖包;

2、编写实现自定义function的代码,并对其编译打包;

3、将编译好的包拷贝至JMeter的扩展目录下;

4、运行查看自定义的function是否正确;

2、新建项目

在Eclipse中新建一个Maven项目,File > New > Project,选择Maven Project;然后引入扩展jmeter function的2个依赖包,我用的是jmeter3.3版本。如下图。

1.jpeg

2.jpeg

3、编写JMeter function代码

要实现扩展JMeter function,主要有两点:

1、 实现function的类的package声明必须包含".functions"。

JMeter设计让一些核心的类可以在非UI的方式下运行的时候能被加载进来,这些类会被优先加载。加载这些类的时候是通过命名规则来实现的。所有实现function的类必需包含".functions",所以我们自定义实现的类里必须包含".functions",比如如下申明的包名为:"jmeter.test.functions"。

3.jpeg

2、需要继承AbstractFunction,并且实现相应的方法。

AbstractFunction类提供了4个抽象方法,在扩展的时候需要实现他们。

1、public void setParameters(Collection
 parameters) throws InvalidVariableException

这个方法在用于传递用户在执行过程当中传入的实际参数值。该方法在function没有参数情况下也会被调用。一般该方法传入的参数会被保存在类内全局变量里,并被后面调用的execute方法中使用到。

4.jpeg

2、public String execute(SampleResult previousResult, Sampler currentSampler)  throws InvalidVariableException

JMeter会将上次运行的SampleResult和当前的Sampler作为参数传入到该方法里,返回值就是在运行该function后得到的值,以String类型返回。该方法如果操作了非线程安全的对象(比如文件),则需要将对该方法进行线程同步保护。

5.jpeg

3、public String getReferenceKey()

这个就是function的名字。JMeter的命名规则是在方法名前面加入双下划线"__"。比如"__MyRandomFunc",function的名字跟实现该类的类名应该一致,而且该名字应该以static final的方式在实现类中定义好,避免在运行的时候更改它。

6.png

 
4、public List
 getArgumentDesc()

最后在你的实现类中还需要提供一个方法来告诉JMeter关于你实现的function的描述。

7.png

4、测试function正确性

右击项目选址export 》选择jar file,如下打包成一个jar包。

8.jpeg

把这个JAR拷贝至$JMETER__HOME/lib/ext目录下,重新启动JMeter。打开选项中的函数助手对话框,如果配置正确的话就能出现自己定义的function,如下所示。每次点击右下角的生产按钮,会生成调用该function随机数字,每点击一次,这个值就不一样,可见编写的函数是正确的。

9.jpeg

5、完整的源码

大家可以参考如下的代码,修改成自己想要扩展的,今后就可以随心所欲的编写一些基础功能的函数了。

如果需要完整的工程,可在微信公众号大话性能发送:jmeter函数工程,即可获取下载工程密码。

完整工程下载地址:

package jmeter.test.functions; import java.util.Collection; import java.util.LinkedList; import java.util.List; import java.util.Random; import org.apache.jmeter.engine.util.CompoundVariable; import org.apache.jmeter.functions.AbstractFunction; import org.apache.jmeter.functions.InvalidVariableException; import org.apache.jmeter.samplers.SampleResult; import org.apache.jmeter.samplers.Sampler; public class MyRandomFunc extends AbstractFunction {
    // 自定义function的描述     private static final List
 desc = new LinkedList
();     static {
        desc.add("Get a random int within specified parameter value.");     }     // function名称     private static final String KEY = "__MyRandomFunc";     private static final int MAX_PARA_COUNT = 1;     private static final int MIN_PARA_COUNT = 1;     // 传入参数的值     private Object[] values;     private Random r = new Random();     @Override     public List
 getArgumentDesc() {
        return desc;     }     @Override     public String execute(SampleResult previousResult, Sampler currentSampler) throws InvalidVariableException {
        try {
            int max = new Integer(((CompoundVariable) values[0]).execute().trim());             int val = r.nextInt(max);             return String.valueOf(val);         } catch (Exception ex) {
            throw new InvalidVariableException(ex);         }     }     @Override     public String getReferenceKey() {
        return KEY;     }     @Override     public void setParameters(Collection
 parameters) throws InvalidVariableException {
        checkParameterCount(parameters, MIN_PARA_COUNT, MAX_PARA_COUNT); // 检查参数的个数是否正确         values = parameters.toArray(); // 将值存入类变量中。     } }

更多其他测试内容都在公众号大话性能可以参考链接文末