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类啦，填写上相关的参数就可以啦。

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