定制JMeter取样器

• JMeter运行你区定义自己的取样器sampler，方法很简单，只需继承类org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient，并重写若干方法即可。

• 首先，创建一个maven项目，并加入jmeter相关的依赖包：

<dependencies>
    <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter</artifactId>
        <version>2.10</version>
    </dependency>
    <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter_core</artifactId>
        <version>2.10</version>
    </dependency>
    <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter_http</artifactId>
        <version>2.10</version>
    </dependency>
    <dependency>
        <groupId>org.apache.jmeter</groupId>
        <artifactId>ApacheJMeter_java</artifactId>
        <version>2.10</version>
    </dependency>
</dependencies>

• 然后，创建一个类，继承自AbstractJavaSamplerClient，并重写setupTest、teardownTest、runTest与getDefaultParameters方法即可

• 代码示例如下：

package pengliu.me.jmeter.plugins;

import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;

import java.util.Iterator;
import java.util.Random;

public class MySampler extends AbstractJavaSamplerClient
{
    private static final Logger log = LoggingManager.getLoggerForClass();

    @Override
    public void setupTest(JavaSamplerContext context)
    {
        Iterator<String> it = context.getParameterNamesIterator();
        while (it.hasNext())
        {
            String paramName =  it.next();
            this.log.info(String.format("-----------------paramName: %s, paramValue: %s-----------------",
                    paramName,
                    context.getParameter(paramName)));
        }

        this.log.info(String.format("-----------------Get param 'name' from JMeter: %s-----------------",
                context.getParameter("name", "Default Name")));
        this.log.info(String.format("-----------------Current Thread Id: %s, Start 'setupTest(JavaSamplerContext context)'--------------------",
                Thread.currentThread().getId()));
    }

    @Override
    public void teardownTest(JavaSamplerContext context)
    {
        this.log.info(String.format("-----------------Current Thread Id: %s, Start 'teardownTest(JavaSamplerContext context)'--------------------",
                Thread.currentThread().getId()));
    }

    public SampleResult runTest(JavaSamplerContext javaSamplerContext)
    {
        SampleResult result = new SampleResult();
        result.sampleStart();
        try
        {
            Random r = new Random();
            long sleepTime = (long)r.nextInt(5000);
            this.log.info(String.format("-----------------Current Thread Id: %s, Running Test, sleep time is %s--------------------",
                    Thread.currentThread().getId(),
                    sleepTime));
            Thread.sleep(sleepTime);

            if(sleepTime%2 == 0)
            {
                throw new Exception("--------This is my exception from runTest!!!---------");
            }

            result.sampleEnd();
            result.setSuccessful(true);
            result.setSampleLabel("My Lable: SUCCESS: " + Thread.currentThread().getId());
        }
        catch (Throwable e)
        {
            result.sampleEnd();
            result.setSuccessful(false);
            result.setSampleLabel("My Lable: FAIL: " + Thread.currentThread().getId());

            e.printStackTrace();
        }

        return result;
    }

    @Override
    public Arguments getDefaultParameters() {

        Arguments params = new Arguments();

        params.addArgument("name", "edw");

        return params;
    }
}

• 其中setupTest与teardownTest方法，在一个线程只被执行一次，无论此Thread Group配置的Loop Count是多少，每个线程只运行一次这两个方法。

• 其中runTest方法，如果一个线程配置了多次的Loop Count，那么此runTest就会运行多次

• 其中getDefaultParameters方法，返回的是JMeter界面的参数列表，如下图：
  

• 使用”mvn package”命令打包上面的sampler:

• 将打包好的jar包放到JMeter的lib/ext目录下：

 

• 启动JMeter, 并加入一个Java Request的sampler: 

 

• Classname选择代码中自定义的sampler类： 

• 配置Thread Group，线程数为２，循环数为５： 

• 选中Java Request这个sampler，可以设置参数name的值

• 开始执行后，JMeter会产生两个线程，每个线程都跑只跑一遍setupTest与teardownTest方法，但会跑５遍runTest方法

• 查看JMeter的日志输出：

• 查看”View Results in Table”中的结果：