JMeter 扩展开发:扩展 TCP 取样器

news2025/1/18 11:52:23

前言

对基于 TCP/IP 协议的套接字应用进行性能测试是非常常见的测试场景。JMeter 提供的“TCP 取样器”大部分情况下可以满足测试的需求,但是也有它的局限性。如果希望实现更灵活的 TCP 套接字测试方式,可以通过对 JMeter 内置的 TCP 取样器进行扩展开发来实现。

JMeter TCP 取样器的实现

1

在使用 JMeter TCP 取样器时,可以指定 TCPClient 接口的扩展类名,以切换不同的实现。如果不指定,JMeter 默认使用的是 org.apache.jmeter.protocol.tcp.sampler.TCPClientImpl。除了 TCPClientImpl,JMeter 还提供了另外两个实现,分别是 BinaryTCPClientImpl 和 LengthPrefixedBinaryTCPClientImpl,用于处理二进制格式的数据。其中:

  • 使用 BinaryTCPClientImpl 时,文本框中应输入十六进制字符内容,该实现将十六进制转换为对应二进制的字节内容后进行发送。
  • 使用 LengthPrefixedBinaryTCPClientImpl 时,使用字节流的前两个或前四个字节存放消息的长度,通过该前缀长度值来确定字节流的结束位置。

实现 TCPClient 接口来增加新的 TCP 取样方式,是扩展 TCP 取样器的一种方法。

但是如果我们需要对 TCP 取样器做一个通用的修改,例如,现在的 TCP 取样器在读取服务器端返回的响应时,会以“行尾 EOL 字节值”中指定的字节作为结束符,来确定读取的结束位置;不过这种设计就不适用于没有明确终止符,只有固定长度的返回响应。仅增加 TCPClient 接口的实现还不足以实现类似的需求,接下来将示例介绍如何进行改造,使得 TCP 取样器除了指定结束符,还能支持指定返回字节流的长度。

实现效果

先看一下修改后的效果。在“行尾EOL字节值”之后增加了一个“响应长度”的字段,举例来说,下图中指定了响应长度为12字节,如果服务器返回的是"Echo: hello\n"(其中"\n"是回车符),那么总长度就是12字节,也就是会读取到回车符之后停止。如果“行尾EOL字节值”和“响应长度”同时设置的话,将优先使用“行尾EOL字节值”的配置。

2

准备开发环境

首先,从 JMeter 官网下载所使用的 JMeter 对应的源代码。

JMeter 对 TCP 协议的支持都放在了 protocol/tcp 目录下,因此本次开发的所做的更改都集中于该目录,如下图所示:

3

整体方法与前文 JMeter 扩展插件实现对自定义协议进行支持 很类似,也需要分别对 Sampler 界面和 Sampler 实现逻辑进行调整。

扩展实现

步骤1:改造 Sampler 界面

需要更改的类为:org.apache.jmeter.protocol.tcp.config.gui.TCPConfigGui.java

主要改动是在类中加入新的“响应长度”的字段。参考代码如下:

...

public class TCPConfigGui extends AbstractConfigGui {

    ...
    private JTextField responseLenth;
    ...

    @Override
    public void configure(TestElement element) {
        ...
        responseLenth.setText(element.getPropertyAsString(TCPSampler.LENGTH, ""));
    }

    @Override
    public void modifyTestElement(TestElement element) {
        ...
        element.setProperty(TCPSampler.LENGTH, responseLenth.getText(), ""); //TCPSampler.LENGTH 稍后定义
    }

    @Override
    public void clearGui() {
        ...
        responseLenth.setText("");
    }

    private JPanel createLengthPanel() {
        JLabel label = new JLabel(JMeterUtils.getResString("response_length")); //$NON-NLS-1$ 

        responseLenth = new JTextField(3); // 3 columns size
        responseLenth.setMaximumSize(new Dimension(responseLenth.getPreferredSize()));
        label.setLabelFor(responseLenth);

        JPanel lengthPanel = new JPanel(new FlowLayout());
        lengthPanel.add(label);
        lengthPanel.add(responseLenth);
        return lengthPanel;
    }

    private void init() {
        ...
        optionsPanel.add(createEolBytePanel());
        optionsPanel.add(createLengthPanel());
        mainPanel.add(optionsPanel);
        ...
    }

    ...
}

步骤2:更新 Sampler 逻辑

首先在 org.apache.jmeter.protocol.tcp.sampler.TCPSampler.java 加入“响应长度”字段的定义,该字段值将会被 set 到 TCPClient 中:

...
public class TCPSampler extends AbstractSampler implements ThreadListener, Interruptible {

    ...
    public static final String LENGTH = "TCPSampler.length"; //$NON-NLS-1$
    ...

    private TCPClient getProtocol() {
        TCPClient tcpClient = null;
        Class<?> javaClass = getClass(getClassname());
        if (javaClass == null){
            return null;
        }
        try {
            tcpClient = (TCPClient) javaClass.newInstance();
            if (getPropertyAsString(EOL_BYTE, "").length()>0){
                tcpClient.setEolByte(getEolByte());
                log.info("Using eolByte={}", getEolByte());
            } else if (getPropertyAsString(LENGTH, "").trim().length()>0) { 
                tcpClient.setLength(Integer.parseInt(getPropertyAsString(LENGTH, "").trim()));
                log.info("Using length={}", getPropertyAsString(LENGTH, ""));
            }

            if (log.isDebugEnabled()) {
                log.debug("{} Created: {}@{}", this, getClassname(), Integer.toHexString(tcpClient.hashCode())); //$NON-NLS-1$
            }
        } catch (Exception e) {
            log.error("{} Exception creating: {} ", this, getClassname(), e); //$NON-NLS-1$
        }
        return tcpClient;
    }
    ...
}

在 TCPClient 接口中增加长度字段的支持:

...

public interface TCPClient {

    ...
    /**
     * Get the response length setting
     * 
     * @return
     */
    int getLength();


    /**
     * Set the length of returned response.
     * 
     * @param length
     */
    void setLength(int length);
}

然后对 TCPClient 的各实现类进行改造。

  • AbstractTCPClient 中增加长度字段 get/set 方法的实现:
...

public abstract class AbstractTCPClient implements TCPClient {

    ...
    protected int length = -1;

    ...
    @Override
    public int getLength() {
        if(useEolByte) {
            return -1;
        }
        return length;
    }

    @Override
    public void setLength(int length) {
        this.length = length;
    }
}
  • TCPClientImpl 中对 read 方法进行改造:
...

public class TCPClientImpl extends AbstractTCPClient {

    ...
    @Override
    public String read(InputStream is, SampleResult sampleResult) throws ReadException{
        ByteArrayOutputStream w = new ByteArrayOutputStream();
        try {
            byte[] buffer = new byte[4096];
            int x;
            boolean first = true;
            //如果没有设置响应长度,仍使用行尾EOL字节值来确定响应的结束;否则使用响应长度来进行限制
            if(getLength() == -1) {
                while ((x = is.read(buffer)) > -1) {
                    if (first) {
                        sampleResult.latencyEnd();
                        first = false;
                    }
                    w.write(buffer, 0, x);
                    if (useEolByte && (buffer[x - 1] == eolByte)) {
                        break;
                    }
                }   
            } else {
                buffer = new byte[length];
                if ((x = is.read(buffer, 0, length)) > -1) {
                    sampleResult.latencyEnd();
                    w.write(buffer, 0, x);
                }
            }

            // do we need to close byte array (or flush it?)
            if(log.isDebugEnabled()) {
                log.debug("Read: {}\n{}", w.size(), w.toString());
            }
            return w.toString(CHARSET);
        } catch (IOException e) {
            throw new ReadException("Error reading from server, bytes read: " + w.size(), e, w.toString());
        }
    }
    ...
}

如果需要其他的 TCPClient 实现类也支持响应长度,可以参考 TCPClientImpl 的改造来进行。

步骤3:编译、打包和部署

由于本次扩展直接修改了 JMeter 内置的 TCP 取样器,因此需要对 JMeter 源码部分进行编译和打包。

生成编译好的 jar 包后,替换 $JMETER_HOME/lib/ext/ApacheJMeter_tcp.jar,重启 JMeter 即可生效。

注意:由于替换掉了 JMeter 的内置实现,请先做好原有 ApacheJMeter_tcp.jar 的备份。本文只作为开发扩展的一个参考,如果用于实际的生产测试中,替换前请对扩展的修改进行仔细评估。

总结

本文是 JMeter 扩展开发的一次应用,在对 JMeter 内置的 TCP 取样器本身有所了解的情况下,对它的功能进行了拓展。

版权声明: 本文为 EMQ 原创,转载请注明出处。

原文链接:https://www.emqx.com/zh/blog/jmeter-tcp

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/44471.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

在数据工厂中刷新PowerBI数据集

一开始因为部门使用的是坚果云来同步资料&#xff0c;而坚果云同步SSIS工程总是会报错&#xff0c;一气之下就把所有的SSIS迁移到了Azure云上&#xff0c;本来部门使用的就是Azure SQL&#xff0c;所以迁到Data Factory&#xff08;数据工厂&#xff09;也不需要过多的配置。 …

传输线理论基础01——相关定义、信号速率、分布参数与电报方程

前言一直以来都对高频信号、信号完整性、传输线、分布参数这些概念似懂非懂&#xff0c;上学时没学过相关课程&#xff0c;这导致我对高频电路和PCB理解较差&#xff0c;这里新开一个专栏&#xff0c;补齐这方面知识。 一. 传输线相关定义1.1 传输线定义 传输线指的是传输信号…

PyTorch学习笔记-神经网络模型搭建小实战

1. torch.nn.Sequential torch.nn.Sequential 是一个Sequential 容器&#xff0c;能够在容器中嵌套各种实现神经网络中具体功能相关的类&#xff0c;来完成对神经网络模型的搭建。模块的加入一般有两种方式&#xff0c;一种是直接嵌套&#xff0c;另一种是以 OrderedDict 有序…

LabVIEW创建类 1

LabVIEW创建类 1 通过创建LabVIEW类&#xff0c;可在LabVIEW中创建用户定义的数据类型。LabVIEW类定义了对象相关的数据和可对数据执行的操作&#xff08;即方法&#xff09;。通过封装和继承可创建模块化的代码&#xff0c;使代码更易修改而不影响应用程序中的其它代码。 在…

Terraform 华为云最佳实践

目录划分如下&#xff1a;首先是环境&#xff0c;分为网络和service。global是全局的配置&#xff0c;也就是backend的配置&#xff0c;这次使用s3的存储作为backend的存储。最后就是模块做了一些封装。 在global里面的backend里面的main.tf去创建s3的存储。华为云支持s3存储&a…

[附源码]Python计算机毕业设计Django病房管理系统

项目运行 环境配置&#xff1a; Pychram社区版 python3.7.7 Mysql5.7 HBuilderXlist pipNavicat11Djangonodejs。 项目技术&#xff1a; django python Vue 等等组成&#xff0c;B/S模式 pychram管理等等。 环境需要 1.运行环境&#xff1a;最好是python3.7.7&#xff0c;…

RK3588平台开发系列讲解(USB篇)USB 外设 CONFIG

平台内核版本安卓版本RK3588Linux 5.10Android 12文章目录 一、 Mass Storage Class CONFIG二、USB Serial Converter CONFIG三、USB HID CONFIG四、USB Net CONFIG五、USB Camera CONFIG六、USB Audio CONFIG七、 USB HUB CONFIG沉淀、分享、成长,让自己和他人都能有所收获!…

PG::Seppuku

nmap -Pn -p- -T4 --min-rate1000 192.168.81.90 nmap -Pn -p 21,22,80,139,445,7080,7601,8088 -sCV 192.168.81.90 查看7601端口的页面 对路径进行爆破 在/secret路径下得到了用户名和一个密码字典 尝试ssh爆破 得到密码 eeyoree ssh登录 这里使用sudo -l&#xff0…

FineReport表格软件- 计算操作符说明

1. 概述 FineReport 中使用函数需要用到很多的操作符。 操作符不仅包含很多运算符&#xff0c;还包括一些报表特有的操作符。 FineReport 11.0 优化了公式 2. 运算符类型 运算符用于指定要对公式中的元素执行的计算类型。有默认计算顺序&#xff0c;但可以使用括号更改此顺序…

企业表格软件-FineReport 数组函数概述

1. ADD2ARRAY ADD2ARRAY(array, insertArray, start)&#xff1a;在数组 array 的第 start 个位置插入 insertArray 中的所有元素&#xff0c;再返回该数组。 示例&#xff1a; ADD2ARRAY([3, 4, 1, 5, 7], [23, 43, 22], 3)返回[3, 4, 23, 43, 22, 1, 5, 7]。 ADD2ARRAY([…

将 AWS IAM Identity Center (SSO) SAML 与 Amazon OpenSearch Dashboard集成

Amazon OpenSearch Amazon OpenSearch Service 是一项 AWS 托管服务&#xff0c;可以让您运行和扩展 OpenSearch 集群&#xff0c;而不必担心管理、监控和维护您的基础设施&#xff0c;或者不必在操作 OpenSearch 集群方面积累深入的专业知识。 基于 SAML 的 OpenSearch Dash…

Json用法总结

1、忽略json JsonIgnoreProperties(value{“addressId”}) JSONField(serializefalse) JsonIgnore 2、 JsonFiled JsonProperty XStreamAlias Builder.Default 网上可以查询下相关资料 3、 JSON.parseObject(response, ***Response.class) JSONObject.parseObject(response, **…

LockSupport的使用

参考链接&#xff1a; LockSupport使用场景及原理详解 AQS的引入 LockSupport的使用 LockSupport是一个工具类&#xff0c;提供了基本的线程阻塞和唤醒功能&#xff0c;它是创建锁和其他同步组件的基础工具&#xff0c;内部是使用sun.misc.Unsafe类实现的。LockSupport和使用…

android分区概述

Android 设备包括几个分区&#xff0c;它们在启动过程中提供不同的功能。 1、 标准隔断 注意&#xff1a;支持无缝更新的设备每个分区需要一个插槽用于boot 、 system 、 vendor和radio 。 boot分区。此分区包含内核映像&#xff0c;并使用mkbootimg创建。您可以使用虚拟分区…

idea搭建ssm项目全过程详解:

1&#xff0c;创建maven项目&#xff1a; 然后&#xff0c;点击next 其次 2&#xff0c;在pom.xml导入相关依赖&#xff1a;&#xff08;如果idea没有集成maven需要先集成maven&#xff09; <dependencies><dependency><groupId>org.springframework</gr…

【LeetCode】接雨水 II [H](堆)

407. 接雨水 II - 力扣&#xff08;LeetCode&#xff09; 一、题目 给你一个 m x n 的矩阵&#xff0c;其中的值均为非负整数&#xff0c;代表二维高度图每个单元的高度&#xff0c;请计算图中形状最多能接多少体积的雨水。 示例 1&#xff1a; 输入: heightMap [[1,4,3,1,3…

Wireshark TS | 三谈 TCP 握手异常问题

前言 继续以一个实际案例来说下 TCP 握手问题&#xff0c;该数据包仍然来自于 Wireshark sharkfest 2017&#xff0c;一些简短但有趣的 TCP 跟踪文件中的又一个&#xff0c;或者说是最后一个了。可以说这些都是和 TCP 握手相关的连接问题&#xff0c;有兴趣的朋友可以私信&…

Mybatis-Plus开发提速器mybatis-plus-generator-ui

前言 在基于Mybatis的开发模式中&#xff0c;很多开发者还会选择Mybatis-Plus来辅助功能开发&#xff0c;以此提高开发的效率。虽然Mybatis也有代码生成的工具&#xff0c;但Mybatis-Plus由于在Mybatis基础上做了一些调整&#xff0c;因此&#xff0c;常规的生成工具生成的代码…

【一文秒懂——SLF4j日志】

目录 1. SLF4j日志 2. 日志输出 1. SLF4j日志 在添加了spring-boot-starter的项目中&#xff0c;已经包含了SLF4j日志的相关依赖项。 在添加了lombok的项目中&#xff0c;可以在类上添加Slf4j注解&#xff0c;则lombok框架会在编译期在类中声明名为log的变量&#xff0c;通…

农民歌唱家大衣哥喜迎贵客,这三位明星一般人还真请不动

都知道农民歌唱家大衣哥家里热闹&#xff0c;不过大部分都是蹭流量拍视频的&#xff0c;真正的好朋友绝对没有几个。虽然说没有几个好朋友&#xff0c;但是也不代表一个没有&#xff0c;看看在大衣哥家里吃饭的三位&#xff0c;每一个都不是一般人物。 如今的大衣哥&#xff0c…