Jmeter扩展组件开发-自定义java请求

使用JMeter做性能测试时，有时Jmeter本身的功能并不能很好的满足实际的需求，这时可能就需要自己来扩展开发了，Jmeter是支持我们自定义编写测试脚本的，本文简单记录下使用Jmeter扩展组件开发-自定义java请求的流程。
项目工程github：https://github.com/testerwan/JmeterExtended

1、使用IDEA创建一个Maven工程，pom.xml中添加依赖的jar包（ApacheJMeter_core、ApacheJMeter_java）

       
            org.apache.jmeter
            ApacheJMeter_core
            3.2
        
        
            org.apache.jmeter
            ApacheJMeter_java
            3.2
        

2、idea中创建一个Class com.wan.Second 该类实现了JavaSamplerClient接口 实现接口的方法

package com.wan;
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerClient;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
import org.apache.jmeter.samplers.SampleResult;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;

/**
 * Created by wan on 19/5/21.
 */
public class Second implements JavaSamplerClient{

    //URLNAME 就是在图形化界面当中显示的变量名称
    private static final String URLNAME = "URL";
    //设置界面当中默认显示的变量的值
    private static final String DEFAULTURL = "http://www.baidu.com";
    /**
     * 这个方法决定了在jmeter当中要显示哪些属性
     * @return arguments
     */

    //用来存储响应的数据，目的是将响应结果放到察看结果树当中
    private String resultData;
    public Arguments getDefaultParameters() {
        System.out.println("getDefaultParameters run");
        Arguments arguments = new Arguments();
        arguments.addArgument(URLNAME,DEFAULTURL);
        return arguments;
    }

    private String inputUrl;
    /**
     * 这个方法就是一个初始化方法，我们所有的初始化的动作都可以在这里写
     * @param javaSamplerContext
     */
    public void setupTest(JavaSamplerContext javaSamplerContext) {
        inputUrl = javaSamplerContext.getParameter(URLNAME,DEFAULTURL);
        System.out.println("setupTest run");
        System.out.println("用户输入的url地址是:"+inputUrl);
    }

    /**
     * 这个方法就是实现你具体功能逻辑的方法
     * @param javaSamplerContext
     * @return
     */
    public SampleResult runTest(JavaSamplerContext javaSamplerContext) {
        SampleResult result = new SampleResult();
        System.out.println("runTest run");
        StringBuffer sb = new StringBuffer();

        try {
            URL url = new URL(inputUrl);
            URLConnection conn = url.openConnection();
            byte[] buffer = new byte[1024];
            int len;
            result.sampleStart();//标记事务的开始
            InputStream in = conn.getInputStream();
            while ((len=in.read(buffer))!=-1){
                resultData = new String(buffer,"UTF-8");
                sb.append(resultData);
            }
        } catch (MalformedURLException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }


//        resultData = "这就是响应结果";
        resultData = sb.toString();
        result.setSampleLabel("自定义java请求访问");
        result.setSuccessful(true);//告诉查看结果树访问是否成功
        result.setResponseData(resultData,null);
        result.setDataType(SampleResult.TEXT);
        return result;
    }

    /**
     * 这个方法就是来做一些收尾的工作的。
     * @param javaSamplerContext
     */
    public void teardownTest(JavaSamplerContext javaSamplerContext) {
        System.out.println("teardownTest run");
    }
}

3、对工程进行打包:mvn clean package

image.png

4、把打包后的jar包 拷贝到jmeter安装目录下的bin/lib/ext文件夹下

image.png

5、启动Jmeter，创建一个线程组，添加一个java请求 选择类名为com.wan.Second 执行线程组后 查看结果树中 能看到请求相应能正常看到，说明自定java请求脚本执行成功

image.png

image.png

image.png