jmeter-增加自定义函数【java】

场景:需要一个生成可用的身份证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(); //将值存入类变量中
    }

}

你可能感兴趣的:(jmeter-增加自定义函数【java】)