对基于 TCP/IP 协议的套接字应用进行性能测试是非常常见的测试场景。JMeter 提供的“TCP 取样器”大部分情况下可以满足测试的需求,但是也有它的局限性。如果希望实现更灵活的 TCP 套接字测试方式,可以通过对 JMeter 内置的 TCP 取样器进行扩展开发来实现。
在使用 JMeter TCP 取样器时,可以指定 TCPClient 接口的扩展类名,以切换不同的实现。如果不指定,JMeter 默认使用的是 org.apache.jmeter.protocol.tcp.sampler.TCPClientImpl。除了 TCPClientImpl,JMeter 还提供了另外两个实现,分别是 BinaryTCPClientImpl 和 LengthPrefixedBinaryTCPClientImpl,用于处理二进制格式的数据。其中:
实现 TCPClient 接口来增加新的 TCP 取样方式,是扩展 TCP 取样器的一种方法。
但是如果我们需要对 TCP 取样器做一个通用的修改,例如,现在的 TCP 取样器在读取服务器端返回的响应时,会以“行尾 EOL 字节值”中指定的字节作为结束符,来确定读取的结束位置;不过这种设计就不适用于没有明确终止符,只有固定长度的返回响应。仅增加 TCPClient 接口的实现还不足以实现类似的需求,接下来将示例介绍如何进行改造,使得 TCP 取样器除了指定结束符,还能支持指定返回字节流的长度。
先看一下修改后的效果。在“行尾EOL字节值”之后增加了一个“响应长度”的字段,举例来说,下图中指定了响应长度为12字节,如果服务器返回的是"Echo: hello\n"(其中"\n"是回车符),那么总长度就是12字节,也就是会读取到回车符之后停止。如果“行尾EOL字节值”和“响应长度”同时设置的话,将优先使用“行尾EOL字节值”的配置。
首先,从 JMeter 官网下载所使用的 JMeter 对应的源代码。
JMeter 对 TCP 协议的支持都放在了 protocol/tcp 目录下,因此本次开发的所做的更改都集中于该目录,如下图所示:
整体方法与前文 JMeter 扩展插件实现对自定义协议进行支持 很类似,也需要分别对 Sampler 界面和 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);
...
}
...
}
首先在 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 的各实现类进行改造。
...
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;
}
}
...
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 的改造来进行。
由于本次扩展直接修改了 JMeter 内置的 TCP 取样器,因此需要对 JMeter 源码部分进行编译和打包。
生成编译好的 jar 包后,替换 $JMETER_HOME/lib/ext/ApacheJMeter_tcp.jar,重启 JMeter 即可生效。
注意:由于替换掉了 JMeter 的内置实现,请先做好原有 ApacheJMeter_tcp.jar 的备份。本文只作为开发扩展的一个参考,如果用于实际的生产测试中,替换前请对扩展的修改进行仔细评估。
本文是 JMeter 扩展开发的一次应用,在对 JMeter 内置的 TCP 取样器本身有所了解的情况下,对它的功能进行了拓展。
版权声明: 本文为 EMQ 原创,转载请注明出处。