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<CompoundVariable> 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<String> 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函数工程，即可获取下载工程密码。
完整工程下载地址：https://share.weiyun.com/5JoDvIh
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<String> desc = new LinkedList<String>();
    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<String> 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<CompoundVariable> parameters) throws InvalidVariableException {
        checkParameterCount(parameters, MIN_PARA_COUNT, MAX_PARA_COUNT); // 检查参数的个数是否正确
        values = parameters.toArray(); // 将值存入类变量中。
    }

}


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