Jmeter自定义编写Java代码调用socket通信

一、前言

最近需要测试一款手机游戏的性能,找不到啥录制脚本的工具,然后,另外想办法。性能测试实际上就是对服务器的承载能力的测试,和各种类型的手机客户端没有啥多大关系,手机再好,服务器负载不了,也不能够运行,依照这个思路,测试该手游的性能就化为对服务器的性能测试,也就是服务器和手机客户端所联系的接口的测试。

如果是可以用HTTP调用的,那样很好解决,直接带上地址参数就OK,开始一两个很顺利,设计参数化了一下数据就很顺利的解决,然后,遇到了一个切换场景的接口,用的是socket通信,再用上述方法貌似不可行,然后偶很懒,所以折腾了一天,结果,还是没有找到好的办法,无奈,想到了Jmeter可以调用java请求,只好到开发那边拿到协议,通过自定义编写java代码来实现socket通信,连接成功,返回数据成功。顺便说一句,Jmeter真是居家旅行必备~~~~

首先,Jmeter使用自定义编写代码的步骤:http://www.cnblogs.com/linglingyuese/articles/linglingyuese-two.html

二、代码

SocketClient.java类

package app;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.net.Socket; public class SocketClient { public boolean sendMessage(String ipAddress,int port,String data){
try {
Socket socket=new Socket(ipAddress,port);
DataInputStream input=new DataInputStream(socket.getInputStream());
DataOutputStream output = new DataOutputStream(socket.getOutputStream()); byte[] message=data.getBytes("UTF-8"); //获得长度
int length=message.length; message[0] = (byte) 0x00;
message[length-1] = (byte) 0xff; output.write(message);
output.flush();
System.out.println(message);
Thread.sleep(1000);
if (!socket.isClosed() && socket.isConnected()) {
byte[] cbuf=new byte[8096];
input.read(cbuf);
System.out.println("读取到的是:"+cbuf);
String responseLine=new String(cbuf);
System.out.println(responseLine+"===========");
}
output.close();
input.close(); socket.close(); return true;
} catch (Exception e) {
return false;
}
}
}

TestScence.java类

package app;

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; public class TestScence extends AbstractJavaSamplerClient { private SampleResult results; private String ipAddress; private String port; private String data; /**
* 初始化方法
*/
public void setupTest(JavaSamplerContext jsc) {
results = new SampleResult(); ipAddress = jsc.getParameter("ipAddress", "");
port = jsc.getParameter("port", "");
data = jsc.getParameter("data", ""); if (ipAddress != null && ipAddress.length() > 0){
results.setSamplerData(ipAddress);
} if (port != null && port.length() > 0){
results.setSamplerData(port);
} if (data != null && data.length() > 0){
results.setSamplerData(data);
} } /**
* 设置传入参数
*/
public Arguments getDefaultParameters() {
Arguments params = new Arguments(); params.addArgument("ipAddress", "");
params.addArgument("port", "");
params.addArgument("data", ""); return params;
} /**
* 线程运行体
*/
public SampleResult runTest(JavaSamplerContext arg0) {
SocketClient sc = new SocketClient(); int port2 = 50001; if (port != null && port.length() > 0){
port2 = Integer.parseInt(port);
} results.sampleStart(); boolean flag = sc.sendMessage(ipAddress, port2, data); results.sampleEnd(); if (flag){
results.setSuccessful(true);
}else{
results.setSuccessful(false);
} return results;
} /**
* 结束方法
*/
public void teardownTest(JavaSamplerContext arg0) { } }

三、导出为可以使Jmeter读取的jar包

在项目的根节点,右击选择【Export】,选择java下面的【JAR file】

然后一直点击下一步即可。

将导出的jar包放到如下目录:\apache-jmeter-2.9\lib\ext,然后重新启动Jmeter啊,新建java请求,就可以看到你刚刚写的java类啦,填写上相关的参数就可以啦。

测试不到一年,对性能啥的理解还不是太深,各位大神不要较真~~~~

上一篇:【3y】从零单排学Redis【青铜】


下一篇:如何更规范化编写Java 代码