简介

Jmeter已经内置了各种协议的取样器，已经能满足常用的性能压测需求。且在前面一章Jmeter扩展开发--自定义java取样器-程序员宅基地中也有关于Java取样器的扩展开发，不过有时候我们期望能定制自己的取样器和界面。为此，需要对Jmeter做扩展，自定义取样器开发

依赖&打包

基本的依赖和打包可以查看前一章Jmeter扩展开发--自定义java取样器-程序员宅基地

效果

自定义取样器的使用效果如下图所示：

如上图，成功编写自定义取样器之后，打包，将相关jar包放置在:Jmeter安装目录/lib/ext目录，重启Jmeter即可，当选择某个自定义插件之后，实例界面如下：

核心代码简介

在自定义取样器开发中有两个类需要重点介绍

org.apache.jmeter.samplers.gui.AbstractSamplerGui  

该类是自定义Jmeter取样器的ui显示类，可以直接像操作Swing一样编辑Jmeter取样器的ui界面，如上图2所示。该类包含如下重要方法介绍如下：

getStaticLabel

自定义取样器在Jmeter界面上显示的名字如下图：

示例代码如下：

public String getStaticLabel() {
        return "YC http请求";
    }

getLabelResource

同上，取样器名称的本地化方法，返回一样即可(其实上层组件调用的是getStaticLabel方法获取取样器名称，如果没有国际化的处理，直接放回相同即可)

public String getLabelResource() {
        return "YC http请求";
    }

createTestElement

该方法会创建一个AbstractSampler对象返回即可，必须对该对象中的字段做赋值的动作，一般的即是一个ui字段对应一个AbstractSampler字段即可，示例代码如下：

public TestElement createTestElement() {
        YcSample ycSample = new YcSample();
        ycSample.setUserName(this.userNameField.getText());
        ycSample.setAddress(this.addressField.getText());
        ycSample.setAge(this.ageField.getText());
        ycSample.setScore(this.scoreField.getText());
        return ycSample;
    }

modifyTestElement

该方法是将UI界面上配置的信息，传递给执行测试Samper对象完成测试，其实就是将界面上传递的参数传递给sampler，完成数据传递的过程，一般的，通过ui界面的字段取值，然后设置到sampler对应的字段值上去即可，实例代码如下：

public void modifyTestElement(TestElement element) {
        YcSample ycSample = (YcSample)element;
        ycSample.setUserName(this.userNameField.getText());
        ycSample.setAddress(this.addressField.getText());
        ycSample.setAge(this.ageField.getText());
        ycSample.setScore(this.scoreField.getText());
    }

构造器

ui的核心方法是构造器的，在构造器中可以直接使用Swing的方法构建ui界面即可，示例方法如下：

public YcHttpGui() {
        super(); // 此处一定要先调用父类构造器
        setLayout(new BorderLayout());
        setBorder(makeBorder());
        add(makeTitlePanel(), BorderLayout.NORTH);

        JPanel mainPanel = new VerticalPanel();
        add(mainPanel, BorderLayout.CENTER);


        JPanel jPanel = new JPanel();
        jPanel.setLayout(new GridLayout(2,3));

        jPanel.add(userNameField);
        jPanel.add(addressField);
        jPanel.add(ageField);
        jPanel.add(scoreField);

        VerticalPanel verticalPanel = new VerticalPanel();
        verticalPanel.add(jPanel);

        TitledBorder border = BorderFactory.createTitledBorder("用户信息");
        verticalPanel.setBorder(border);

        mainPanel.add(verticalPanel);
    }

以上基本是一些ui定式，如果还需要其他的ui元素，添加即可

org.apache.jmeter.samplers.AbstractSampler

该类是自定义Jmeter取样器的逻辑运行类，核心方法之后一个sample，说明如下

sample

该方法是自定义取样器的核心逻辑运行方法，只需要返回一个SampleResult对象即可(此处和扩展java取样器逻辑一样)，示例代码如下：

public SampleResult sample(Entry e) {
        SampleResult result = new SampleResult();
        result.sampleStart();
        log.info("userName is:{}",this.userName);
        System.out.println("运行时名称是:" + this.userName);
        result.sampleEnd();
        result.setSuccessful(true);
        result.setSamplerData("this is test!");
        result.setResponseCode("200");
        return result;
    }

完整代码

自定义Jmeter取样器UI完整代码如下：

import org.apache.jmeter.gui.util.VerticalPanel;
import org.apache.jmeter.samplers.gui.AbstractSamplerGui;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jorphan.gui.JLabeledTextField;

import javax.swing.*;
import javax.swing.border.TitledBorder;
import java.awt.*;

public class YcHttpGui extends AbstractSamplerGui {

    private JLabeledTextField userNameField = new JLabeledTextField("用户名");
    private JLabeledTextField addressField = new JLabeledTextField("地址");
    private JLabeledTextField ageField = new JLabeledTextField("年龄");
    private JLabeledTextField scoreField = new JLabeledTextField("score");

    public YcHttpGui() {
        super(); // 此处一定要先调用父类构造器
        setLayout(new BorderLayout());
        setBorder(makeBorder());
        add(makeTitlePanel(), BorderLayout.NORTH);

        JPanel mainPanel = new VerticalPanel();
        add(mainPanel, BorderLayout.CENTER);


        JPanel jPanel = new JPanel();
        jPanel.setLayout(new GridLayout(2,3));

        jPanel.add(userNameField);
        jPanel.add(addressField);
        jPanel.add(ageField);
        jPanel.add(scoreField);

        VerticalPanel verticalPanel = new VerticalPanel();
        verticalPanel.add(jPanel);

        TitledBorder border = BorderFactory.createTitledBorder("用户信息");
        verticalPanel.setBorder(border);

        mainPanel.add(verticalPanel);
    }

    @Override
    public String getStaticLabel() {
        return "YC http请求";
    }

    @Override
    public String getLabelResource() {
        return "YC http请求";
    }

    @Override
    public TestElement createTestElement() {
        YcSample ycSample = new YcSample();
        ycSample.setUserName(this.userNameField.getText());
        ycSample.setAddress(this.addressField.getText());
        ycSample.setAge(this.ageField.getText());
        ycSample.setScore(this.scoreField.getText());
        return ycSample;
    }

    @Override
    public void modifyTestElement(TestElement element) {
        YcSample ycSample = (YcSample)element;
        ycSample.setUserName(this.userNameField.getText());
        ycSample.setAddress(this.addressField.getText());
        ycSample.setAge(this.ageField.getText());
        ycSample.setScore(this.scoreField.getText());
    }

}

自定义Jmeter取样器sampler完整代码如下：

import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.apache.jmeter.samplers.AbstractSampler;
import org.apache.jmeter.samplers.Entry;
import org.apache.jmeter.samplers.SampleResult;

@Data
@Slf4j
public class YcSample extends AbstractSampler {

    private String userName;
    private String address;
    private String age;
    private String score;

    @Override
    public SampleResult sample(Entry e) {
        SampleResult result = new SampleResult();
        result.sampleStart();
        log.info("userName is:{}",this.userName);
        System.out.println("运行时名称是:" + this.userName);
        result.sampleEnd();
        result.setSuccessful(true);
        result.setSamplerData("this is test!");
        result.setResponseCode("200");
        return result;
    }
}

后续会继续分享关于Jmeter自定义取样器的源码分析，届时会把相关逻辑梳理清楚