JMeter 源码二次开发函数示例
一、JMeter 5.0 版本
实际测试中,依靠jmeter自带的函数已经无法满足我们需求,这个时候就需要二次开发。本次导入的是jmeter 5.0的源码进行实际的函数开发。
二、开发函数
在src/functions新建class-IntThreeSum(函数求和)
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 org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;
/**
* @ClassName IntThreeSum三个参数求和
* @Description TODO
* @Author Hardy.Feng
* @Date 2018/12/20 14:00
* @Version 1.0
**/
public class IntThreeSum extends AbstractFunction { //继承AbstractFunction类,必须对父类的代码重写
private static final Logger log = LoggerFactory.getLogger(IntThreeSum.class);
private static final List<String> desc = new LinkedList<>(); //描述
private static final String KEY = "__IntThreeSum"; //方法描述,必须双下划线
static {
desc.add(JMeterUtils.getResString("first_param"));
desc.add(JMeterUtils.getResString("secone_param"));
desc.add(JMeterUtils.getResString("third_param"));
}
private Object[] values;
public IntThreeSum() {
}
@Override
public String execute(SampleResult previousResult, Sampler currentSampler) throws InvalidVariableException {
JMeterVariables vars = getVariables();
int sum = 0;
String varName = ((CompoundVariable) values[values.length - 1]).execute().trim();
log.info("varName==>:{}", varName);
//遍历获取3个数之和
for (int i = 0; i < values.length - 1; i++) {
sum += Integer.parseInt(((CompoundVariable) values[i]).execute());
}
try {
sum += Integer.parseInt(varName);
varName = null;
} catch (NumberFormatException ignored) {
}
String totalString = Integer.toString(sum);
if (vars != null && varName != null) {
vars.put(varName.trim(), totalString);
log.info("varName:", vars.get(varName.trim()));
}
return totalString;
}
@Override
public void setParameters(Collection<CompoundVariable> parameters) throws InvalidVariableException {
//对入参进行检查,最小3个参数
checkMinParameterCount(parameters,3);
values = parameters.toArray();
}
@Override
public String getReferenceKey() {
return KEY;
}
@Override
public List<String> getArgumentDesc() {
return desc;
}
}
三、测试类Test
在test/src新建测试类IntThreeSumTest
package org.apache.jmeter.functions;
import org.apache.jmeter.JMeter;
import org.apache.jmeter.engine.util.CompoundVariable;
import org.apache.jmeter.junit.JMeterTestCase;
import org.apache.jmeter.threads.JMeterContext;
import org.apache.jmeter.threads.JMeterContextService;
import org.apache.jmeter.threads.JMeterVariables;
import org.apache.log4j.Logger;
import org.junit.Before;
import org.junit.Test;
import org.slf4j.LoggerFactory;
import static org.apache.jmeter.functions.FunctionTestHelper.makeParams;
import java.util.Collection;
/**
* @ClassName IntThreeSumTest
* @Description TODO
* @Author hardy.Feng
* @Date 2018/12/20 14:30
* @Version 1.0
**/
public class IntThreeSumTest extends JMeterTestCase {
@Test
public void sumTest() throws Exception {
IntThreeSum intThreeSum = new IntThreeSum();
checkInvalidParameterCounts(intThreeSum,3); //检查一下参数最小入参3位
Collection<CompoundVariable> params = makeParams("1","2","3");
intThreeSum.setParameters(params);
String totalString = intThreeSum.execute();
System.out.println("total:" + totalString);
}
}
四、导出jar文件
如果测试类测试没有问题,就可以导出class为jar文件。放到lib/ext目录下面,如下图所示
五、启动jmeter
函数已经生效。欢迎留言讨论。
公众号:大家可以关注: