1. 线程组,在我们测试方案里面,每个线程模拟一个用户,执行用户的登录、等等等一系列的操作。由于我们的项目是长连接的,如何能实现多个sample公用一个长连接客户端,考虑了很久,最后实现方法如下:
1 package tea.client.network;
2 /**
3 * @author Teaey
4 * @creation 2012-8-25
5 */
6 public class NetworkClientHolder
7 {
8 /**
9 * 这里使用ThradLocal存储BaseClient
10 * 方便一轮测试的每个sample都是由同一个socketChannel发送
11 * 更真实的模拟用户
12 */
13 private static ThreadLocal<BaseClient> clientHolder = new ThreadLocal<BaseClient>();
14 public static BaseClient getClient(String ip, String port)
15 {
16 BaseClient client = clientHolder.get();
17 if (null == client)
18 {
19 client = new BaseClient(ip, port);
20 client.connect();
21 clientHolder.set(client);
22 }
23 return client;
24 }
25 }
26
代码中使用thread_local保存Socket客户端,这样每个sample中发送数据的客户端都是从这里拿的,就可以保证长连接的情况下,socket不会重复创建,很好的模拟了用户。
当然不单单是链接可以保存,所有需要在线程中共享的数据都可以通过这种方法来实现。
2. 接下来是如何封装发送请求的客户端,这里用的netty,具体可以根据项目情况使用mina或者nio都可以。代码直接明了^_^:
1 package tea.client.network;
2
3 import java.net.InetSocketAddress;
4 import java.util.concurrent.Executors;
5 import org.jboss.netty.bootstrap.ClientBootstrap;
6 import org.jboss.netty.channel.Channel;
7 import org.jboss.netty.channel.ChannelFuture;
8 import org.jboss.netty.channel.ChannelHandlerContext;
9 import org.jboss.netty.channel.ChannelPipeline;
10 import org.jboss.netty.channel.ChannelPipelineFactory;
11 import org.jboss.netty.channel.ChannelStateEvent;
12 import org.jboss.netty.channel.Channels;
13 import org.jboss.netty.channel.ExceptionEvent;
14 import org.jboss.netty.channel.MessageEvent;
15 import org.jboss.netty.channel.SimpleChannelHandler;
16 import org.jboss.netty.channel.socket.nio.NioClientSocketChannelFactory;
17 import tea.common.network.ClientDecoder;
18 import tea.common.network.ClientEncoder;
19 import tea.common.network.ClientMessage;
20
21 /**
22 * @author Teaey
23 * @creation 2012-8-25
24 */
25 public class BaseClient
26 {
27 public BaseClient(String ip, String port)
28 {
29 this.ip = ip;
30 this.port = port;
31 }
32 private String ip;
33 private String port;
34 private Channel channel;
35 private ClientBootstrap bootstrap;
36 private Object syn = new Object();
37 private static final int Receive_Timeout = 10000; //ms
38 private ClientMessage response = null;
39 public void connect()
40 {
41 bootstrap = new ClientBootstrap(new NioClientSocketChannelFactory(Executors.newCachedThreadPool(), Executors.newCachedThreadPool()));
42 bootstrap.setOption("tcpNoDelay", true);
43 bootstrap.setPipelineFactory(new ClientPipelineFactory());
44 while (true)
45 {
46 ChannelFuture future = bootstrap.connect(new InetSocketAddress(ip, Integer.parseInt(port)));
47 future.awaitUninterruptibly(5000);
48 if (future.isDone())
49 {
50 channel = future.getChannel();
51 if (channel != null && channel.isConnected())
52 {
53 break;
54 }
55 }
56 }
57 }
58 public void disconnect()
59 {
60 if (channel.isConnected())
61 {
62 channel.disconnect();
63 }
64 }
65 public boolean isConnected()
66 {
67 return channel.isConnected();
68 }
69 public void close()
70 {
71 if (this.channel.isOpen())
72 {
73 this.channel.close();
74 }
75 bootstrap.releaseExternalResources();
76 }
77 /**
78 * 发送消息,无需返回
79 */
80 public void send(ClientMessage message)
81 {
82 channel.write(message);
83 }
84 /**
85 * 发送消息,等待返回
86 */
87 public ClientMessage sendWaitBack(ClientMessage message)
88 {
89 response = null;
90 try
91 {
92 channel.write(message);
93 synchronized (syn)
94 {
95 try
96 {
97 syn.wait(Receive_Timeout);
98 } catch (InterruptedException e)
99 {
100 e.printStackTrace();
101 }
102 }
103 if (null == response)
104 {
105 System.err.println("Receive response timeout");
106 }
107 } catch (Exception e)
108 {
109 e.printStackTrace();
110 }
111 return response;
112 }
113 class ClientPipelineFactory implements ChannelPipelineFactory
114 {
115 public ChannelPipeline getPipeline() throws Exception
116 {
117 ChannelPipeline p = Channels.pipeline();
118 p.addLast("frameDecoder", new ClientDecoder());
119 p.addLast("fremeEncoder", new ClientEncoder());
120 p.addLast("logicHandler", new ClientMsgHandler());
121 return p;
122 }
123 }
124 class ClientMsgHandler extends SimpleChannelHandler
125 {
126 public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception
127 {
128 Object obj = e.getMessage();
129 if (obj instanceof ClientMessage)
130 {
131 ClientMessage msg = (ClientMessage) obj;
132 response = msg;
133 synchronized (syn)
134 {
135 syn.notifyAll();
136 }
137 }
138 }
139 public void channelConnected(ChannelHandlerContext ctx, ChannelStateEvent e) throws Exception
140 {
141 System.out.println("connected server:" + ctx.getChannel());
142 }
143 public void channelDisconnected(ChannelHandlerContext ctx, ChannelStateEvent e) throws Exception
144 {
145 System.out.println("disconnected server:" + ctx.getChannel());
146 }
147 public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e) throws Exception
148 {
149 System.out.println("Error in exceptionCaught:" + e.getCause());
150 }
151 }
152 }
153
这段代码展示了我们的客户端,这里所有的请求有两种发送模式,一种是发送并阻塞等待返回(sendWaitBack),第二种就是直接发送(send)。
3. 有了发送请求的客户端,那如何能够更简单的实现一个协议好让客户端发送,再贴一段代码^_^:
1 package tea.client.network;
2
3 import org.apache.jmeter.config.Arguments;
4 import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient;
5 import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
6 import org.apache.jmeter.samplers.SampleResult;
7 import com.google.protobuf.InvalidProtocolBufferException;
8 import com.google.protobuf.MessageLite;
9
10 /**
11 * @author Teaey
12 * @creation 2012-8-25
13 */
14 public abstract class BaseSample extends AbstractJavaSamplerClient
15 {
16 public static final String PARAM_IP = "ip";
17 public static final String PARAM_PORT = "port";
18 public static final String VAR_IP = "${ip}";
19 public static final String VAR_PORT = "${port}";
20 protected BaseClient client;
21 public void addParameter(Arguments params)
22 {
23 }
24 /**
25 * Jmeter获取消息参数,默认配置ip和port两个参数
26 * 如果子类有更多参数,调用super.getDefaultParameters()获取Arguments后,继续设置其他方法
27 */
28 @Override
29 public Arguments getDefaultParameters()
30 {
31 System.out.println("1.getDefaultParameters");
32 Arguments params = new Arguments();
33 params.addArgument(PARAM_IP, VAR_IP);
34 params.addArgument(PARAM_PORT, VAR_PORT);
35 addParameter(params);
36 return params;
37 }
38 /**
39 * runTest的前置方法
40 */
41 @Override
42 public void setupTest(JavaSamplerContext context)
43 {
44 System.out.println("2.setupTest:" + context.containsParameter(PARAM_IP));
45 String ip = context.getParameter(PARAM_IP);
46 String port = context.getParameter(PARAM_PORT);
47 this.client = NetworkClientHolder.getClient(ip, port);
48 System.out.println("thread--->" + Thread.currentThread().getId() + " client--->" + client);
49 }
50 /**
51 * Jmeter调用,用于实际的测试
52 */
53 @Override
54 public SampleResult runTest(JavaSamplerContext context)
55 {
56 SampleResult sample = getSample();
57 sample.sampleStart();
58 try
59 {
60 MessageLite response = doTest();
61 String msg = response == null ? "" : response.toString();
62 sample.setResponseMessage(msg);
63 sample.setSuccessful(true);
64 } catch (Exception e)
65 {
66 sample.setSuccessful(false);
67 e.printStackTrace();
68 } finally
69 {
70 sample.sampleEnd();
71 }
72 return sample;
73 }
74 /**
75 * 获取本Sample的标签,子类实现
76 */
77 public abstract String getLabel();
78 /**
79 * 获取一个带标签的Sample
80 */
81 public SampleResult getSample()
82 {
83 SampleResult sample = new SampleResult();
84 sample.setSampleLabel(getLabel());
85 return sample;
86 }
87 /**
88 * Jmeter调用,用于
89 */
90 @Override
91 public void teardownTest(JavaSamplerContext context)
92 {
93 System.out.println("4.teardownTest");
94 }
95 /**
96 * 需实现,具体测试的方法,调用client的send/sendWithBack发送请求
97 * 如无返回,放回null即可
98 */
99 public abstract MessageLite doTest() throws InvalidProtocolBufferException;
100 }
好的,这里封装了下AbstractJavaSamplerClient,每个消息默认包含ip和port参数,这可以再jmeter的用户变量中定义好。为了方便大家添加消息的参数,这里实现了空的
addParameter(Arguments params)方法,这样在具体消息中直接重写这个方法,来添加具体的参数。是不是很方便?^_^,具体协议还需要实现的两个方法分别是:getLabel和doTest。第一个方法时用于报告显示的请求名字,一般定义为消息名字+“Label”就OKay。第二个方法就是我们重点重写的方法,这里再贴段代码,是一个具体消息的实现:
1 package tea.client;
2
3 import com.google.protobuf.InvalidProtocolBufferException;
4 import com.google.protobuf.MessageLite;
5 import tea.client.network.BaseSample;
6 import tea.common.network.ClientMessage;
7 import tea.common.network.RPC.HeartBeat_C2S;
8 import tea.common.network.RPC.HeartBeat_S2C;
9
10 /**
11 * @author Teaey
12 * @creation 2012-8-24
13 */
14 public class HeartBeatSample extends BaseSample
15 {
16 @Override
17 public MessageLite doTest() throws InvalidProtocolBufferException
18 {
19 HeartBeat_C2S.Builder request = HeartBeat_C2S.newBuilder();
20 request.setTimestamp(System.currentTimeMillis());
21 ClientMessage cm = new ClientMessage();
22 cm.setContent(request.build().toByteArray());
23 cm.setName("HeartBeat");
24 ClientMessage sm = client.sendWaitBack(cm);
25 HeartBeat_S2C response = HeartBeat_S2C.parseFrom(sm.getContent());
26 return response;
27 }
28 @Override
29 public String getLabel()
30 {
31 return "HeartBeatSample";
32 }
33 }
34
可以看到doTest的工作就是封装请求,并拿到父类的client发送,然后返回响应(send方式返回null),Okay,大功告成。
要说的就是这么多,写得不好但属原创,体量作者转载请标明出处。提意见请留言或者163邮箱:masfay,感谢。