基于LoadRunner平台Socket协议的JavaVuser

阅读更多

/*
* LoadRunner Java script.
* Loadrunner：11.00.0.0
*
* 脚本描述:
* 模拟基于Socket协议的即时消息系统中的客户端行为LR脚本，
* 为模拟真实POS终端，接收消息和发送消息分两个线程工作。
*
* 作者：maosheng
* 制作时间：2013-1-11
* 
*/ 
 
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 OutputStream out; 
    // 输入流 
    private InputStream in; 
    // 客户端是否连接标志 
    //private boolean connFlag = false; 
    private  AtomicBoolean connFlag =new AtomicBoolean(false);
 
 
    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("172.16.4.168", 8888); 
                System.out.println("connect success!!!");
connFlag.set(true);
                out = client.getOutputStream(); 
                in = client.getInputStream();
        } catch (UnknownHostException e) { 
                System.out.println("UnknownHostException=" + e); 
                e.printStackTrace(); 
        } catch (IOException e) { 
                System.out.println("IOException=" + e); 
                e.printStackTrace(); 
        } 
    } 

 
    //发送消息 
    private void sendAction() { 
if(connFlag.get()){  
   
            //收到报文:[0001 1 0 31]0001123456789          e590986a
            //发送报文：
            byte[] byteMsg=new byte[]{
                    Byte.parseByte("0"),
    Byte.parseByte("1"),
    Byte.parseByte("0"),
    Byte.parseByte("1"),
    Byte.parseByte("0"),
    Byte.parseByte("0"),
    Byte.parseByte("31"),
    Byte.parseByte("48"),
    Byte.parseByte("48"),
    Byte.parseByte("48"),
    Byte.parseByte("49"),
    Byte.parseByte("49"),
    Byte.parseByte("50"),
    Byte.parseByte("51"),
    Byte.parseByte("52"),
    Byte.parseByte("53"),
    Byte.parseByte("54"),
    Byte.parseByte("55"),
    Byte.parseByte("56"),
    Byte.parseByte("57"),
    Byte.parseByte("32"),
    Byte.parseByte("32"),
    Byte.parseByte("32"),
    Byte.parseByte("32"),
    Byte.parseByte("32"),
    Byte.parseByte("32"),
    Byte.parseByte("32"),
    Byte.parseByte("32"),
    Byte.parseByte("32"),
    Byte.parseByte("32"),
    Byte.parseByte("101"),
    Byte.parseByte("53"),
    Byte.parseByte("57"),
    Byte.parseByte("48"),
    Byte.parseByte("57"),
    Byte.parseByte("56"),
    Byte.parseByte("54"),
    Byte.parseByte("97")
};

            String msg = new String(byteMsg);

            System.out.println("发送的消息:"+msg);
    sendMsg(msg);
    receiveMsg();
} 
    } 
   
    //关闭客户端 
    private void sendEnd() { 
connFlag.set(false); 
try { 
    client.close(); 
            System.out.println("connect closed!!!");
} catch (IOException e) { 
    e.printStackTrace(); 
} 
    } 
 
    //调用写入流方法 
    private void sendMsg(String msg) { 
//new SendThread(msg).start(); 
writeMsg(msg); 
    } 
 
    //将消息写入流 
    private void writeMsg(String msg) { 
    try { 
        if (out != null) { 
        out.write(msg.getBytes("GBK")); 
        out.flush(); 
        } 
    } catch (Exception e) { 
        System.out.println("Exception=" + e); 
    } 
    } 
   
     //接收消息
    private void receiveMsg() { 
      //new ReceiveThread().start(); 
getMsg();
    } 
 
    //得到返回结果 
    private void getMsg() {
       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);
if(result != null && !"".equals(result)){ 
   System.out.println("接收到的消息:" + result); 
} 
}
            } catch (Exception e) { 
               e.printStackTrace();  
            }   
    } 
 
/******************************************************************/
    /**
     * 接收消息线程类
     * 
     * @author maosheng
     * 
     */ 
    private class ReceiveThread extends Thread { 
    @Override 
    public void run() { 
        while(connFlag.get()) {
    getMsg();
            try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
        } 
      } 
    } 
 
 
   /**
     * 发送消息线程类
     * 
     * @author maosheng
     * 
     */  
    private class SendThread extends Thread {
   
    private String msg; 
     
    public SendThread(String msg) { 
        this.msg = msg; 
    } 
     
    @Override 
    public void run() { 
        while (connFlag.get()) { 
     writeMsg(msg);
             try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
        } 
    } 
  } 
 
}//end for class Actions