基于Loadrunner平台Socket协议的JavaVuser（多线程）

本文由谷博涛撰写，讲述了在二零一二年农历腊月廿五的一个下午，如何在Loadrunner平台上利用JavaVuser进行Socket协议的多线程测试，探讨了涉及socket通信、线程管理和异常处理等方面的技术细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/*
 * LoadRunner Java script. (Build: 15)
 * Script Description: 
 *
 * 作者：谷博涛
 * 制作时间：2012-1-18
 * E-mail：captain_gbt@foxmail.com
 * Loadrunner：11.00
 * 
 * 内容概要：
 * 模拟基于Socket协议的即时消息系统中的客户端行为LR脚本，
 * 为模拟真实IM客户端，接收消息和发送消息分两个线程工作。
 * 
 */

import lrapi.lr;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Iterator;
import java.util.concurrent.atomic.AtomicBoolean;

public class Actions
{
    // 客户端
    private Socket client;
    // 房间号
    private String roomId;
    // 用户ID
    private String id;
    // 输出流
    private OutputStream out;
    // 输入流
    private InputStream in;
    // 连接标志
    //private boolean connFlag = false;
    private  AtomicBoolean connFlag =new AtomicBoolean(false);

    // 客户端是否结束标志
    //private boolean endFlag = true;
    private  AtomicBoolean endFlag =new AtomicBoolean(true);


    public int init() throws Throwable {
        connect(); 
	//lr.think_time(10);
        return 0;
    }//end of init


    public int action() throws Throwable {
        sendAction();
        return 0;
    }//end of action


    public int end() throws Throwable {
        sendEnd();
        return 0;
    }//end of end

//====主题代码==================//

    //连接服务器
    private void connect() {
        try {
                client = new Socket("127.0.0.1", 5222);
                System.out.println("connect success!!!");
                out = client.getOutputStream();
                in = client.getInputStream();
                receiveAction();
                login();
        } catch (UnknownHostException e) {
                System.out.println("UnknownHostException=" + e);
                e.printStackTrace();
        } catch (IOException e) {
                System.out.println("IOException=" + e);
                e.printStackTrace();
        }
    }

    //登录服务器
    private void login() {
	    String loginMsg = "<msg type=\"login\" channel=\"CCTV1\"></msg>";
	    sendMsg(loginMsg);
    }

    //启动接收线程
    private void receiveAction() {
	new ReceiveThread().start();
    }

    //得到房间号码和用户ID
    private void getEleVal(String msg) {
	int index =0;

	index = msg.indexOf("to");
	msg = msg.substring(index + 4, msg.length());
	index = msg.indexOf("'");
	id = msg.substring(0, index);
	System.out.println(roomId);

	index = msg.indexOf("roomId");
	msg = msg.substring(index + 8, msg.length());
	index = msg.indexOf("'");
	roomId = msg.substring(0, index);
	System.out.println(id);

	connFlag.set(true);
    }

    //发送消息
    private void sendAction() {
	if(connFlag.get()){
	    System.out.println("发送消息----->");
	    String msg = "<msg type=\"groupchat\" channel=\"CCTV1\" from=\"" + id
	    + "\" to=\"" + roomId + "\"><body>test</body></msg>";
    
	    sendMsg(msg);
	}
    }

    //调用写入流方法
    private void sendMsg(String msg) {
	//new SendThread(msg).start();
	writeMsg(msg);
    }

    //将消息写入流
    private void writeMsg(String msg) {
	try {
	    if (out != null) {
		out.write(msg.getBytes("UTF-8"));
		out.flush();
	    }
	} catch (Exception e) {
	    System.out.println("Exception=" + e);
	}
    }

    //关闭客户端
    private void sendEnd() {
	endFlag.set(false);
	try {
	    client.close();
	} catch (IOException e) {
	    // TODO Auto-generated catch block
	    e.printStackTrace();
	}
    }
    /**
     * 接收消息线程类
     * 
     * @author Administrator
     * 
     */
    private class ReceiveThread extends Thread {
	@Override
	public void run() {
	    while (endFlag.get()) {// 循环接收消息
		    try {
			int len = in.available();

			if(len>0){
			    System.out.println(len);
			    byte[] bytes = new byte[len];
			    in.read(bytes);
			    String result = new String(bytes);
			    System.out.println("接收到的消息:" + result);

			    if(result != null && !"".equals(result)&& result.contains("res_bind")){
				getEleVal(result);
			    }
			}
		    } catch (Exception e) {
			// TODO: handle exception
		    }
	    }
	}
    }


//======发送消息线程类（本代码未用到）=========
    private class SendThread extends Thread {
	private String msg;
    
	public SendThread(String msg) {
	    this.msg = msg;
	}
    
	@Override
	public void run() {
	    if (connFlag.get()) {
		writeMsg(msg);
	    }
	}
    }

}//end for class Actions

谷博涛

二零一二年农历腊月廿五下午

http://blog.youkuaiyun.com/captain_gbt/article/details/7209570