Jmeter二次开发实现自定义functions函数（九）

在Jmeter->选项->函数助手对话框中我们可以看到Jmeter内置的一些常用函数，但考虑到测试过程中的实际情况，我们经常需要在脚本引用或者实现自定义的函数。那么如何在“函数助手对话框中”看到我们自定义的函数呢，下面将介绍详细步骤：

Jmeter源码导入参考上一篇：https://www.cnblogs.com/xiaozhaoboke/p/14103865.html

一、在目录..\apache-jmeter-5.0\src\functions\org\apache\jmeter\functions新建自定义class文件，例如GetPhoneNumber.class并继承AbstractFunction重写以下方法：

package org.apache.jmeter.functions;

import org.apache.jmeter.engine.util.CompoundVariable;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.samplers.Sampler;

import java.util.Collection;
import java.util.List;

public class GetPhoneNumber extends AbstractFunction{

    @Override
    public String execute(SampleResult previousResult, Sampler currentSampler) throws InvalidVariableException {
        return null;
    }

    @Override
    public void setParameters(Collection<CompoundVariable> parameters) throws InvalidVariableException {

    }

    @Override
    public String getReferenceKey() {
        return null;
    }

    @Override
    public List<String> getArgumentDesc() {
        return null;
    }
}

二、实现函数逻辑后代码如下：

package org.apache.jmeter.functions;

import org.apache.jmeter.engine.util.CompoundVariable;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.samplers.Sampler;
import org.apache.jmeter.threads.JMeterVariables;
import org.apache.jmeter.util.JMeterUtils;

import java.util.Collection;
import java.util.LinkedList;
import java.util.List;

public class GetPhoneNumber extends AbstractFunction{
    //定义function名称
    private static final String KEY = "__MobileGenerator";
    private static final List<String> desc = new LinkedList<String>();
    private static final String[] telFirst = "134,135,136,137,138,139,150,151,152,157,158,159,130,131,132,155,156,133,153 ".split(",");
    //自定义function描述
    static{
        desc.add(JMeterUtils.getResString("Name of variable in which to store the result (optional)"));
    }

    private CompoundVariable varName;

    //执行部分
    @Override
    public String execute(SampleResult previousResult, Sampler currentSampler) throws InvalidVariableException {
        int index = getNum(0, telFirst.length - 1);
        String first = telFirst[index];
        String second = String.valueOf(getNum(1, 888) + 10000).substring(1);
        String third = String.valueOf(getNum(1, 9100) + 10000).substring(1);
        String mobile = first + second + third;
        if (varName != null){
            JMeterVariables vars = getVariables();
            final String varTrim = varName.execute().trim();
            if (vars != null && varTrim.length() > 0) {
                vars.put(varTrim, mobile);
            }
        }
        return mobile;
    }

    //设置参数
    @Override
    public void setParameters(Collection<CompoundVariable> parameters)throws InvalidVariableException
    {
        checkParameterCount(parameters, 0, 1);
        Object[] values = parameters.toArray();
        if (values.length > 0) {
            varName = (CompoundVariable) values[0];
        } else {
            varName = null;
        }
    }

    /** {@inheritDoc} */
    @Override
    public String getReferenceKey() {
        return KEY;
    }

    /** {@inheritDoc} */
    public List<String> getArgumentDesc() {
        return desc;
    }

    private static int getNum(int start,int end)
    {
        return (int)(Math.random()*(end-1));
    }

}

三、运行编译

也可以写个测试类

编译后得到GetPhoneNumber.class文件

四、将编译后的class文件放到ApacheJMeter_functions.jar中

　　使用WinRAR打开ApacheJMeter_functions.jar，复制GetPhoneNumber.class文件至\org\apache\jmeter\functions\目录下保存；

五，重启Jmeter，打开函数助手

六、学习后总结，不足之处请指出，后续修正！

参考原文链接：https://blog.csdn.net/liujingqiu/article/details/87891378