场景:需要一个生成可用的身份证id的方法。【算法可以百度搜】
模仿求和方法 ${__intSum(1,2,)}
/**
*
* 身份证算法实现
*
* 1、号码的结构 公民身份号码是特征组合码,
* 由十七位数字本体码和一位校验码组成。
* 排列顺序从左至右依次为:六位数字地址码,八位数字出生日期码 三位数字顺序码和一位数字校验码。
*
* 2、地址码(前六位数) 表示编码对象常住户口所在县(市、旗、区)的行政区划代码,按GB/T2260的规定执行。
*
* 3、出生日期码(第七位至十四位) 表示编码对象出生的年、月、日,按GB/T7408的规定执行,年、月、日代码之间不用分隔符。
*
* 4、顺序码(第十五位至十七位)
* 表示在同一地址码所标识的区域范围内,对同年、同月、同日出生的人编定的顺序号,顺序码的奇数分配给男性,偶数分配给女性。
*
* 5、校验码(第十八位数)
* (1)十七位数字本体码加权求和公式 S = Sum(Ai * Wi), i = 0, ... , 16
* ,先对前17位数字的权求和
* Ai:表示第i位置上的身份证号码数字值
* Wi:表示第i位置上的加权因子 Wi: 7 9 10 5 8 4 2 1 6 3 7 9 10 5 8 4 2
* (2)计算模 Y = mod(S, 11)
* (3)通过模得到对应的校验码 Y: 0 1 2 3 4 5 6 7 8 9 10
* 校验码: 1 0 X 9 8 7 6 5 4 3 2
*
*
*/
可以利用之前的jmeter请求java方法实现。但是考虑到jmeter有
自带方法,所以准备模仿添加一个自定义方法。
百度上文章:都是说把写好的方法打包成jar,放入jmeter的lib/ext目录下,和java请求一样。
遇到问题:自定义的方法,依赖jmeter的jar,把整个工程
打包成jar放入jmeter目录下,会有包的冲突【因为新生成的jar包中已经有jmeter的jar了,
而且如果自定义方法多了,冗余严重】。
刚好想了解下jmeter源码:于是就决定在jmter源码基础上增加方法
优点是:1:直接利用jmeter的jar包,学习jmeter底层方法实现。
2:了解jmeter最新特性
3:不用每次写个方法或其他都要打包成jar放入jmeter的插件包中!!!!
jmeter源码github地址:https://github.com/apache/jmeter
准备工作 :工具部分见百度!~
1:安装git
2:安装ant
3:安装maven
git上文档都是英文,大概意思如下(都是躺过的坑)
安装依赖:在根目录执行
ant download_jars
写完代码后,用过想启动jmeter【每次改完代码后,如果要用执行下面2个命令!!!!】
ant
jmeter
准备就这么多了,具体步骤如下:
在工程目录下 src.org.apache.jmeter.functions下
copy一个其他方法如求和方法,将execute 方法的返回值改为你想要的返回值即可。
具体需要注意的方法如下,copy别人的
扩展AbstractFunction类
AbstractFunction类提供了4个抽象方法,在扩展的时候需要实现他们。
public String execute(SampleResult previousResult, Sampler currentSampler) throws InvalidVariableException
JMeter会将上次运行的SampleResult和当前的Sampler作为参数传入到该方法里,返回值就是在运行该function后得到的值,以String类型返回。该方法如果操作了非线程安全的对象(比如文件),则需要将对该方法进行线程同步保护。
public void setParameters(Collection parameters) throws InvalidVariableException;
这个方法在用于传递用户在执行过程当中传入的实际参数值。该方法在function没有参数情况下也会被调用。一般该方法传入的参数会被保存在类内全局变量里,并被后面调用的execute方法中使用到。
public String getReferenceKey();
这个就是function的名字。JMeter的命名规则是在方法名前面加入双下划线"__"。比如"__GetEven",function的名字跟实现该类的类名应该一致,而且该名字应该以static final的方式在实现类中定义好,避免在运行的时候更改它。
public List getArgumentDesc();
最后在你的实现类中还需要提供一个方法来告诉JMeter关于你实现的function的描述。
源代码实现
实现的源代码如下所示,重要的代码已经有注释。
/**
* Created by cxf on 2017/8/1.
*/
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;
import org.apache.jmeter.engine.util.CompoundVariable;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.samplers.Sampler;
import org.apache.jmeter.util.JMeterUtils;
public class GetIdCard extends AbstractFunction {
//自定义function的描述
private static final List desc = new LinkedList();
static {
desc.add(JMeterUtils.getResString("intsum_param_1")); //$NON-NLS-1$
desc.add(JMeterUtils.getResString("intsum_param_2")); //$NON-NLS-1$
desc.add(JMeterUtils.getResString("function_name_paropt")); //$NON-NLS-1$
}
//function名称
private static final String KEY = "_getIdCard";//你的方法名称
private static final int MAX_PARA_COUNT = 2;//参数做多2个
private static final int MIN_PARA_COUNT = 0;//参数最少0个
public GetIdCard() {
}
//传入参数的值
private Object[] values;
@Override
public List getArgumentDesc() {
return desc;
}
@Override
public String execute(SampleResult previousResult, Sampler currentSampler) throws InvalidVariableException {
try {
//
IdCardGenerator g = new IdCardGenerator();//单独生成身份证的类
String x=g.generate();//单独生成身份证的方法
if(x==null){
return "ID is null";
}else{
return "ID="+ g.generate();
}
} catch(Exception ex) {
throw new InvalidVariableException(ex);
}
}
@Override
public String getReferenceKey() {
return KEY;//你的方法名称
}
@Override
public void setParameters(Collection parameters) throws InvalidVariableException {
checkParameterCount(parameters,MIN_PARA_COUNT,MAX_PARA_COUNT); //检查参数的个数是否正确
values = parameters.toArray(); //将值存入类变量中
}
}