美文网首页
jmeter-增加自定义函数【java】

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

作者: abaeccdce206 | 来源:发表于2017-08-02 18:36 被阅读323次

    场景:需要一个生成可用的身份证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<CompoundVariable> parameters) throws InvalidVariableException;
    

    这个方法在用于传递用户在执行过程当中传入的实际参数值。该方法在function没有参数情况下也会被调用。一般该方法传入的参数会被保存在类内全局变量里,并被后面调用的execute方法中使用到。

       public String getReferenceKey();
    

    这个就是function的名字。JMeter的命名规则是在方法名前面加入双下划线"__"。比如"__GetEven",function的名字跟实现该类的类名应该一致,而且该名字应该以static final的方式在实现类中定义好,避免在运行的时候更改它。

       public List<String> 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<String> desc = new LinkedList<String>();
        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<String> 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<CompoundVariable> parameters) throws InvalidVariableException {
            checkParameterCount(parameters,MIN_PARA_COUNT,MAX_PARA_COUNT); //检查参数的个数是否正确
            values = parameters.toArray(); //将值存入类变量中
        }
    
    }

    相关文章

      网友评论

          本文标题:jmeter-增加自定义函数【java】

          本文链接:https://www.haomeiwen.com/subject/bxeylxtx.html