Jmeter实现自定义函数助手

以一个乘法计算器为例，实现一个简单的两数相乘的函数助手。

1.在pom文件中添加ApacheJmeter_functions依赖

<dependency>
    <groupId>org.apache.jmeter</groupId>
    <artifactId>ApacheJMeter_functions</artifactId>
    <version>5.0</version>
</dependency>

2.新建包名为com.xxx.functions的包，包名一定要是functions，否则Jmeter无法识别

3.新建一个MyMul类继承AbstractFunction类，然后实现他的4个方法

public class MyMul extends AbstractFunction {
    public String execute(SampleResult sampleResult, Sampler sampler) throws InvalidVariableException {
        return null;
    }
    public void setParameters(Collection<CompoundVariable> collection) throws InvalidVariableException {
    }
    public String getReferenceKey() {
        return null;
    }
    public List<String> getArgumentDesc() {
        return null;
    }
}

execute()：执行方法
setParamenters()：设置参数，接收用户传递的参数
getReferenceKey()：功能名称
getArgumentDesc()： 功能描述、参数描述

首先，实现getReferenceKey方法，这个方法的返回值就是显示在函数助手中的名称。

/**
* 功能名称,显示在函数助手中的名称
* @return 
*/
public String getReferenceKey() {
    return "__MyMul";
}

然后实现setParameters方法，这个方法用来接收用户输入的参数

/**
* 设置参数，接收用户传递的参数
* @param collection
* @throws InvalidVariableException
*/
private Object[] values; //定义一个数组用来保存用户参数
public void setParameters(Collection<CompoundVariable> collection) throws InvalidVariableException {
    checkParameterCount(collection,2);  //检查用户参数的数量，我们要实现两数相乘，所以这里要有两个参数
    values = collection.toArray();  //将用户输入的参数保存到values中
}

然后再实现具体的执行方法execute()

/**
* 执行方法
* @param sampleResult
* @param sampler
* @return
* @throws InvalidVariableException
*/
private CompoundVariable first,second; //定义两个参数，用来保存获取到的用户输入参数
public String execute(SampleResult sampleResult, Sampler sampler) throws InvalidVariableException {
    first = (CompoundVariable) values[0];   //从values数组中拿到第一个数
    second = (CompoundVariable) values[1];  //从values数组中拿到第二个数
    //删除两个数的前后空格然后转换成int类型，再进行相乘
    long mul = new Integer(first.execute().trim()) * new Integer(second.execute().trim()); 
    //在输出台打印出 两数之积
    System.out.println("两数之积："+mul);   
    //返回相乘结果
    return String.valueOf(mul);  
}

最后在实现参数描述方法getArgumentDesc()

/**
* 参数描述，显示在函数助手中的参数描述中
* @return
*/
public List<String> getArgumentDesc() {
    //定义一个List用来保存结果
    List<String> desc = new ArrayList<String>();
    desc.add("第一个数"); //添加第一个参数描述
    desc.add("第二个数"); //添加第二个参数描述
    return desc;
}

4.使用mvn clean package 命令打包之后放大jmeter目录下的/lib/ext/目录

5.重启Jmeter之后，打开函数助手对话框就可以看到自己写的函数助手

选择之后，添加第一个参数和第二个参数，点击生成

可以看到函数的结果是21。

 

在控制台可打印出了“两数之积：21”

 

然后我们在具体请求中使用一下吧。

在Jmeter中添加一个线程组，在线程组下添加一个BeanShell取样器和查看结果树，我们demo中大多数情况下使用BeanShell是为了更直观的显示我们的请求数据。

在BeanShell里面输入如下内容

 

执行之后在控制台打印出了我们输入的几个参数的乘积。

最近阿里云又开始做活动了，老用户和新用户都有超低的折扣，点击下面链接就可以看到了：

新用户>

老用户>