由于公司系统传送报文格式的特殊性 ，公司环境要求报文格式转换加上报文长度转换成byte[] 数组的格式，拼在报文里面。 导致websocket插件不好用。所以用java自定义了一套代码，用来压测接口。
1，首先创建一个java工程，在里面新建一个测试类，名称随便起。 我这里项目名称叫做socketTest，测试类Test。
2，从Jmeter的安装目录lib/ext中拷贝两个文件ApacheJMeter_core.jar和ApacheJMeter_java.jar到TestNumber项目中，然后引入这两个JAR包。
3，在类中继承AbstractJavaSamplerClient 类，并且实现以下几个方法。
4，方法如下：
setupTest()：初始化方法，用于初始化性能测试时的每个线程；
getDefaultParameters()：主要用于设置传入的参数；
runTest()：为性能测试时的线程运行体；
teardownTest()：测试结束方法，用于结束性能测试中的每个线程。
具体代码试下如下。
这段代码发送的报文，只使用于公司环境下。并不适应所有工程。其他工程需要自己替换sendMsg()方法。其中在代码里初始化中设置了三个字段，这些字段分别都在jmeter中可调，ip（ip地址），port(端口) ，request（请求报文）。
import com.kakasys.common.VO;
import com.kakasys.socket.tool.Tool;
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 java.io.*;
import java.net.*;
import java.util.HashMap;
import java.util.Map;
/**
* @author howsky
*/
public class Test extends AbstractJavaSamplerClient {
// Sock begin----------------------------------------------
private Socket sid;
private PrintWriter out;
private BufferedReader in;
private String ip;
private String port;
private String request;
private static String label = "howsky.net";
// 测试结果
private SampleResult sr;
/**
* 初始化
*/
public void setupTest(JavaSamplerContext arg0) {
System.out.println("setupTest");
request = arg0.getParameter("request","");
if (request==null){
System.out.println("没有传参");
}
}
/**
* 设置请求的参数
*/
public Arguments getDefaultParameters() {
Arguments params = new Arguments();
params.addArgument("ip", "10.10.15.245");
params.addArgument("port", "19088");
params.addArgument("request",request);
return params;
}
/**
* 运行过程
*/
public SampleResult runTest(JavaSamplerContext arg) {
String message = request;
//log.debug("====开始调用=====" + message);
byte[] msg = new byte[0];
try {
msg = message.getBytes("GBK");
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
}
try {
msg = message.getBytes("GBK");
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
}
byte[] rst = sendMsg(msg, "10.10.15.245", 19088);
SampleResult result = new SampleResult();
// String res = new String(rst, "UTF-8");
System.out.println("===========");
System.out.println(rst.length);
result.setEncodingAndType("GB2312");
result.setResponseData(rst);
result.setSuccessful(true);
return result;
}
/**
* 结束
*/
public void teardownTest(JavaSamplerContext arg0) {
System.out.println("end");
}
/**
* 发送消息
*
* @param ip
* @param port
* @param
* @throws Exception
*/
public static byte[] sendMsg(byte[] allbyte, String ip, int port) {
int headLen = 6;
int timeout = 1800000;
byte[] buff = null;
// 发送信息
Socket st = null;
OutputStream outputStream = null;
try {
//log.info("请求socket：" + new String(allbyte,"GBK"));
// 1.新建socket对象
//log.info("请求ip：" + ip + "请求port：" + port);
st = new Socket(ip, port);
// 2.报文长度的处理
int len = allbyte.length;
//log.debug("request pack length :" + len);
byte[] lengthDomain = Tool.intAsciiTobyte(len, +headLen);
//log.debug("request pack length :" + lengthDomain);
byte[] xmlByte = Tool.memmerge(lengthDomain, allbyte);
//log.debug("request XML length::" + xmlByte.length);
// 3.开始发送
outputStream = st.getOutputStream();
OutputStream dataOutputStream = new DataOutputStream(new BufferedOutputStream(outputStream));
dataOutputStream.write(xmlByte);
// log.debug("request XML 内容 :" + new String(xmlByte,"GBK"));
dataOutputStream.flush();
// 1.接收同步回执，调用组件进行解析处理
st.setSoTimeout(timeout);
InputStream is = st.getInputStream();
byte[] buf = new byte[headLen];
int l = is.read(buf, 0, headLen);
//log.debug("============length:"+l);
if (l < headLen) {
// log.debug("MsgControlClient header len is short ");
VO vo = new VO();
vo.setAttr("msg", "message format error header short");
}
// 2.如果是字符串数字
String bodyLen = new String(buf);
//log.debug("body length :" + bodyLen);
int lenDomain = Integer.parseInt(bodyLen);
// log.debug("lenDomain : " + lenDomain);
// 3.返回报文
buff = new byte[lenDomain];
int size = 1024;
int totalSize = 0;
while (totalSize < lenDomain) {
byte[] buffer = new byte[size];
size = is.read(buffer, 0, size);
//log.debug("====body actual size:" + size);
memcpy(buff, totalSize, buffer, 0, size);
totalSize += size;
}
// log.debug("request XML 内容"+(new String(buff,"GBK")));
} catch (Exception e) {
// log.debug("exception ", e);
return null;
}
// log.info("响应socket：" + new String(buff));
return buff;
}
public static void memcpy(byte[] dest, int dest_offset, byte[] src, int src_offset, int len) {
int j = dest_offset;
for (int i = src_offset; i < src_offset + len; i++) {
dest[j] = src[i];
j++;
}
}
public static byte[] intAsciiTobyte(int num, int len) {
String src = "" + num;
byte[] result = new byte[len];
String leftZero = "";
for (int i = 0; i < len - src.length(); i++) {
leftZero = "0" + leftZero;
}
src = leftZero + src;
try {
//删除了里面的GBK
result = src.getBytes();
} catch (Exception e) {
//log.error("Tool.intAsciiTobyte exe throw UnsupportedEncodingException", e);
}
return result;
}
// 将两个参数按照先后顺序合并到一起，返回一个字节数组
public static byte[] memmerge(byte[] buffer1, byte[] buffer2) {
byte[] result = new byte[buffer1.length + buffer2.length];
System.arraycopy(buffer1, 0, result, 0, buffer1.length);
System.arraycopy(buffer2, 0, result, buffer1.length, buffer2.length);
return result;
}
}
在Jmeter中引用的过程
1.在jmeter安装目录下引入当前工程打的jar包，将jar包放到lib/ext目录下。
2.在线程组下面新建java请求。
3.配置界面如下：
Catch(11-22-16-0(11-25-11-10-56).jpg
4.调用结果如下：
CatchE5F5(11-22-(11-25-11-10-56).jpg
5.后来查看返回结果是返回报文是乱码，经过查证是windows1521格式 需要转成GB2312格式才可以正常显示。这个在jmeter工具里很方便，直接设置返回报文字符集就可以了。
直接修改sampleresult.default.encoding=GB2312。（记住去掉#，不要还是注释状态哦）
# The encoding to be used if none is provided (default ISO-8859-1)
#sampleresult.default.encoding=GB2312
网友评论