Jmeter扩展组件开发（10） – 自定义扩展函数助手的开发

CODE

package com.functions;

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;

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

public class Plus extends AbstractFunction {

    private Object[] values;//声明变量values用来接用户输入的参数；为什么是Object，用户输入的不代表是int型的数据
    private CompoundVariable first,second;//存储第一个和第二个参数

    /*
     *执行方法
     * @param sampleResul
     * @param sampler
     * @return
     * @throws InvalidVariableException
     */
    public String execute(SampleResult sampleResult, Sampler sampler) throws InvalidVariableException {
        System.out.println("execute run!!!");

        first = (CompoundVariable) values[0];//强转，因为定义first时，定义的是CompoundVariable
        System.out.println("第一个参数是" + first);
        second = (CompoundVariable) values[1];
        System.out.println("第二个参数是" + second);

        int count = new Integer(first.execute().trim()) + new Integer(second.execute().trim());//声明一个“和”变量；将first和second转换为int，将两段空格去掉

        System.out.println("两数相加的合是" + count);
        return String.valueOf(count); //因为public接收的是string，所以返回去转成string
    }

    /*
    *设置参数，接收用户传递的参数
    * @param collection
    * @throws InvalidVariableException
     */
    public void setParameters(Collection<CompoundVariable> collection) throws InvalidVariableException {
        checkMinParameterCount(collection,2);//有两个参数，这个方法是apacheJmeter自带的检查参数写的是否合法，接收参数
        values = collection.toArray();//接收到用户输入的参数，把它变成一个数组
        System.out.println("setParameters run!!!");
    }

    /*
    * 功能名称
    * @return
     */
    public String getReferenceKey() {
        System.out.println("getReferenceKey run!!!");
        return "__MyDemoPlus";
    }

    /*
    * 功能描述，参数描述
    * @return
     */
    public List<String> getArgumentDesc() {
        System.out.println("getArgumentDesc run!!!");
        List desc = new ArrayList();
        desc.add("第一个数字");//第一行参数展示
        desc.add("第二个数字");//第二行参数展示
        return desc;
    }
}

Step

敲代码打包放到Jmeter目录Jmeter重启校验添加__MyDemoPlus函数，运用出结果，OK简单的函数助手开发完成