mina socket登录性能测试 - 51Testing软件测试论坛

package jmeter;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import net.sf.json.JSONObject;
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;
/**
* socket 测试jmeter脚本
* @author wenhuang
*
*/
public class LoginSocketTestJmeter extends AbstractJavaSamplerClient {
private String Ip;
private int Post;
private String user;
private String password;
private String resultData;
// 这个方法是用来自定义java方法入参的。
//设置可用参数及的默认值；
public Arguments getDefaultParameters() {
Arguments params = new Arguments();
params.addArgument("Ip", "192.168.1.1");
params.addArgument("Post", "1111");
params.addArgument("user", "test");
params.addArgument("password", "1111");
return params;
}
//每个线程测试前执行一次，做一些初始化工作；
public void setupTest(JavaSamplerContext arg0) {
}
//开始测试，从arg0参数可以获得参数值；
public SampleResult runTest(JavaSamplerContext arg0) {
Ip = arg0.getParameter("Ip");
Post = arg0.getIntParameter("Post");
user = arg0.getParameter("user");
password = arg0.getParameter("password");
String json = "{\"test\":0,\"data\":{\"user\":\""+user+"\",\"password\":\""+password+"\"},\"key\":\"0\"}";
SampleResult sr = new SampleResult();
sr.setSampleLabel("登录"); //设置样品标签
sr.setResponseMessage("请求的JSON: "+ json);
try {
sr.sampleStart();// jmeter 开始统计响应时间标记
//1.创建客户端Socket，指定服务器地址和端口
Socket socket=new Socket(Ip, Post);
//2.获取输出流，向服务器端发送信息
OutputStream os=socket.getOutputStream();//字节输出流
os.write(getSendHex(json));
os.flush();
//3.获取输入流，并读取服务器端的响应信息
InputStream is=socket.getInputStream();
resultData = getReceiveJson(is);
//4.关闭资源
if(os != null){
os.close();
}
if(is != null){
is.close();
}
if(socket != null){
socket.close();
}
System.out.println("接收到的json ="+resultData);
JSONObject jsonObj = JSONObject.fromObject(resultData);
String codeNum = jsonObj.get("code").toString();
if (Integer.parseInt(codeNum) == 200){
sr.setSuccessful(true);
System.out.println("登录成功，返回code ="+codeNum);
//sr.setResponseMessage("请求的JSON: "+ json + " 登录成功，返回code ="+codeNum); //输出在查看结果树-的取样器结果中
}else{
sr.setSuccessful(false);
System.out.println("登录失败，返回code ="+codeNum);
//sr.setResponseMessage("请求的JSON: "+ json + " 登录失败，返回code ="+codeNum);
}
// 通过下面的操作就可以将被测方法的响应输出到Jmeter的察看结果树中的响应数据里面了。
if (resultData != null && resultData.length() > 0) {
sr.setResponseData("返回结果是："+resultData, null); //输出在查看结果树-的响应结果中
sr.setDataType(SampleResult.TEXT);
}
// System.out.println(resultData);
} catch (Throwable e) {
sr.setSuccessful(false);
e.printStackTrace();
} finally {
sr.sampleEnd();// jmeter 结束统计响应时间标记
}
return sr;
}
//测试结束时调用；
public void teardownTest(JavaSamplerContext arg0) {
//System.out.println(end);
//System.out.println("The cost is"+(end-start)/1000);
}
// main只是为了调试用，最后打jar包的时候注释掉。
/*
public static void main(String[] args)
{ // TODO Auto-generated method stub
for (int i =0;i<5;i++){
Arguments params = new Arguments();
params.addArgument("Ip", "192.168.1.111");//设置参数，并赋予默认值
params.addArgument("Post", "1111");//设置参数，并赋予默认值
params.addArgument("user", "test");//设置参数，并赋予默认值
params.addArgument("password", "1111");
JavaSamplerContext arg0 = new JavaSamplerContext(params);
LoginSocketTestJmeter test = new LoginSocketTestJmeter();
test.setupTest(arg0);
test.runTest(arg0);
test.teardownTest(arg0);
}
}
*/
}

复制代码

package com.test;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;
import net.sf.json.JSONObject;
import org.apache.mina.core.buffer.IoBuffer;
//这两个加密加压的包根据项目修改
import com.tools.Des3;
import com.tools.ZLibUtils;
/**socket 发送消息，和接收消息测试脚本
* @author Administrator
*
*/
public class LoginSocketTest {
public static void main(String[] args) throws Exception {
try {
//1.创建客户端Socket，指定服务器地址和端口
Socket socket=new Socket("192.168.1.1", 11111);
//2.获取输出流，向服务器端发送信息
OutputStream os=socket.getOutputStream();//字节输出流
//PrintWriter pw=new PrintWriter(os);//将输出流包装为打印流
String json = "{\"test\":0,\"data\":{\"user\":\"test\",\"password\":\"test\"},\"key\":\"0\"}";
//将json转成16进制的字节流
os.write(getSendHex(json));
//发送
os.flush();
//socket.shutdownOutput();//关闭输出流
//3.获取输入流，并读取服务器端的响应信息
InputStream is=socket.getInputStream();
//将接收的字节流转成json
String recjson = getReceiveJson(is);
//判断结果是否正确将json转为object对象，取出要判断的参数值code做断言。
JSONObject jsonObj = JSONObject.fromObject(recjson);
String codeNum = jsonObj.get("code").toString();
System.out.println("code="+codeNum);
//4.关闭资源
if(os != null){
os.close();
}
if(is != null){
is.close();
}
if(socket != null){
socket.close();
}
} catch (UnknownHostException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* json转为客户端发送的16进制报文
* @param json
* @return
* @throws Exception
*/
public static byte[] getSendHex(String json) throws Exception {
System.out.println("json ="+json);
IoBuffer buff = IoBuffer.allocate(1024).setAutoExpand(true);
//加密规则，根据项目做修改
json = Des3.encode(json, "test");
//加压
byte[] bytes = ZLibUtils.compress(json.getBytes("UTF-8"));
//装bytes存入buff
buff.putInt(bytes.length);
buff.put(bytes);
buff.flip();
// 将buff转为字节再转为16进制
byte[] b = new byte[buff.limit()];
buff.get(b);
return b;
}
/**
* byte数组转换成16进制字符串
*
* @param src
* @return
*/
public static String bytesToHexString(byte[] src) {
StringBuilder stringBuilder = new StringBuilder();
if (src == null || src.length <= 0) {
return null;
}
for (int i = 0; i < src.length; i++) {
int v = src[i] & 0xFF;
String hv = Integer.toHexString(v);
if (hv.length() < 2) {
stringBuilder.append(0);
}
stringBuilder.append(hv);
stringBuilder.append("\\x");
}
//16进制前面要加\x
return ("\\x"+stringBuilder.substring(0,stringBuilder.length()-2)).toString();
}
/**
* 字节流转int型
* @param b
* @return
*/
public static int b2i(byte[] b) {
int value = 0;
for (int i = 0; i < 4; i++) {
int shift = (4 - 1 - i) * 8;
value += (b[i] & 0x000000FF) << shift;
}
return value;
}
/**客户端得接收服务端返回的iobuffer并转成json
* 接收的iobuffer，前四个字节是表示字节流的长度，先接收前四个字节，并用16进制转10进制int型方法，得到长度的数字
* 再根据长度接收所有的字节流。如果长度错误，拨收是会出错的。
* @param is socket的输入流
* @return json
* @throws Exception
*/
public static String getReceiveJson(InputStream is) throws Exception{
byte[] intByte = new byte[4];
is.read(intByte);
int msgLength = b2i(intByte);
System.out.println("int:"+msgLength);
int readPosition = 0;
byte[] byteArray = new byte[msgLength];
while(readPosition < msgLength){
int value = is.read(byteArray, readPosition, msgLength-readPosition);
if(value == -1){
break;
}
readPosition += value;
}
byte[] data = ZLibUtils.decompress(byteArray); //解压
String code = new String(data, "UTF-8");
String text = Des3.decode(code, "test"); //解密
System.out.println("服务端接收消息内容为：="+text);
return text;
}
}

复制代码