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

测试小白代码能力不强，也可以玩转、轻松扩展 jmeter。


http://dwz.cn/SnoXI5P3

总体思路

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

1、在 Eclipse 中新建 Maven 项目，引入扩展 JMeter function 所需的依赖包；
2、编写实现自定义 function 的代码，并对其编译打包；
3、将编译好的包拷贝至 JMeter 的扩展目录下；
4、运行查看自定义的 function 是否正确；

示例代码

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

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(); // 将值存入类变量中。
    }

}

大家可持续关注大话性能公众号，不断学习测试实战技能和高薪岗位内推。