基於Loadrunner平臺Socket協議的JavaVuser（多線程）

原創 captain_gbt 2018-09-12 04:10

 

  1. /* 
  2.  * LoadRunner Java script. (Build: 15) 
  3.  * Script Description:  
  4.  * 
  5.  * 作者：谷博濤 
  6.  * 製作時間：2012-1-18 
  7.  * E-mail：[email protected] 
  8.  * Loadrunner：11.00 
  9.  *  
  10.  * 內容概要： 
  11.  * 模擬基於Socket協議的即時消息系統中的客戶端行爲LR腳本， 
  12.  * 爲模擬真實IM客戶端，接受消息和發送消息分兩個線程工作。 
  13.  *  
  14.  */ 
  15.  
  16. import lrapi.lr; 
  17. import java.io.IOException; 
  18. import java.io.InputStream; 
  19. import java.io.OutputStream; 
  20. import java.net.Socket; 
  21. import java.net.UnknownHostException; 
  22. import java.util.Iterator; 
  23. import java.util.concurrent.atomic.AtomicBoolean; 
  24.  
  25. public class Actions 
  26. { 
  27.     // 客戶端 
  28.     private Socket client; 
  29.     // 房間號 
  30.     private String roomId; 
  31.     // 用戶ID 
  32.     private String id; 
  33.     // 輸出流 
  34.     private OutputStream out; 
  35.     // 輸入流 
  36.     private InputStream in; 
  37.     // 連接標誌 
  38.     //private boolean connFlag = false; 
  39.     private  AtomicBoolean connFlag =new AtomicBoolean(false); 
  40.  
  41.     // 客戶端是否結束標誌 
  42.     //private boolean endFlag = true; 
  43.     private  AtomicBoolean endFlag =new AtomicBoolean(true); 
  44.  
  45.  
  46.     public int init() throws Throwable { 
  47.         connect();  
  48.     //lr.think_time(10); 
  49.         return 0; 
  50.     }//end of init 
  51.  
  52.  
  53.     public int action() throws Throwable { 
  54.         sendAction(); 
  55.         return 0; 
  56.     }//end of action 
  57.  
  58.  
  59.     public int end() throws Throwable { 
  60.         sendEnd(); 
  61.         return 0; 
  62.     }//end of end 
  63.  
  64. //====主題代碼==================// 
  65.  
  66.     //連接服務器 
  67.     private void connect() { 
  68.         try { 
  69.                 client = new Socket("127.0.0.1", 5222); 
  70.                 System.out.println("connect success!!!"); 
  71.                 out = client.getOutputStream(); 
  72.                 in = client.getInputStream(); 
  73.                 receiveAction(); 
  74.                 login(); 
  75.         } catch (UnknownHostException e) { 
  76.                 System.out.println("UnknownHostException=" + e); 
  77.                 e.printStackTrace(); 
  78.         } catch (IOException e) { 
  79.                 System.out.println("IOException=" + e); 
  80.                 e.printStackTrace(); 
  81.         } 
  82.     } 
  83.  
  84.     //登錄服務器 
  85.     private void login() { 
  86.         String loginMsg = "<msg type=\"login\" channel=\"CCTV1\"></msg>"; 
  87.         sendMsg(loginMsg); 
  88.     } 
  89.  
  90.     //啓動接受線程 
  91.     private void receiveAction() { 
  92.     new ReceiveThread().start(); 
  93.     } 
  94.  
  95.     //得到房間號碼和用戶ID 
  96.     private void getEleVal(String msg) { 
  97.     int index =0; 
  98.  
  99.     index = msg.indexOf("to"); 
  100.     msg = msg.substring(index + 4, msg.length()); 
  101.     index = msg.indexOf("'"); 
  102.     id = msg.substring(0, index); 
  103.     System.out.println(roomId); 
  104.  
  105.     index = msg.indexOf("roomId"); 
  106.     msg = msg.substring(index + 8, msg.length()); 
  107.     index = msg.indexOf("'"); 
  108.     roomId = msg.substring(0, index); 
  109.     System.out.println(id); 
  110.  
  111.     connFlag.set(true); 
  112.     } 
  113.  
  114.     //發送消息 
  115.     private void sendAction() { 
  116.     if(connFlag.get()){ 
  117.         System.out.println("發送消息----->"); 
  118.         String msg = "<msg type=\"groupchat\" channel=\"CCTV1\" from=\"" + id 
  119.         + "\" to=\"" + roomId + "\"><body>test</body></msg>"; 
  120.      
  121.         sendMsg(msg); 
  122.     } 
  123.     } 
  124.  
  125.     //調用寫入流方法 
  126.     private void sendMsg(String msg) { 
  127.     //new SendThread(msg).start(); 
  128.     writeMsg(msg); 
  129.     } 
  130.  
  131.     //將消息寫入流 
  132.     private void writeMsg(String msg) { 
  133.     try { 
  134.         if (out != null) { 
  135.         out.write(msg.getBytes("UTF-8")); 
  136.         out.flush(); 
  137.         } 
  138.     } catch (Exception e) { 
  139.         System.out.println("Exception=" + e); 
  140.     } 
  141.     } 
  142.  
  143.     //關閉客戶端 
  144.     private void sendEnd() { 
  145.     endFlag.set(false); 
  146.     try { 
  147.         client.close(); 
  148.     } catch (IOException e) { 
  149.         // TODO Auto-generated catch block 
  150.         e.printStackTrace(); 
  151.     } 
  152.     } 
  153.     /** 
  154.      * 接收消息線程類 
  155.      *  
  156.      * @author Administrator 
  157.      *  
  158.      */ 
  159.     private class ReceiveThread extends Thread { 
  160.     @Override 
  161.     public void run() { 
  162.         while (endFlag.get()) {// 循環接收消息 
  163.             try { 
  164.             int len = in.available(); 
  165.  
  166.             if(len>0){ 
  167.                 System.out.println(len); 
  168.                 byte[] bytes = new byte[len]; 
  169.                 in.read(bytes); 
  170.                 String result = new String(bytes); 
  171.                 System.out.println("接收到的消息:" + result); 
  172.  
  173.                 if(result != null && !"".equals(result)&& result.contains("res_bind")){ 
  174.                 getEleVal(result); 
  175.                 } 
  176.             } 
  177.             } catch (Exception e) { 
  178.             // TODO: handle exception 
  179.             } 
  180.         } 
  181.     } 
  182.     } 
  183.  
  184.  
  185. //======發送消息線程類（本代碼未用到）========= 
  186.     private class SendThread extends Thread { 
  187.     private String msg; 
  188.      
  189.     public SendThread(String msg) { 
  190.         this.msg = msg; 
  191.     } 
  192.      
  193.     @Override 
  194.     public void run() { 
  195.         if (connFlag.get()) { 
  196.         writeMsg(msg); 
  197.         } 
  198.     } 
  199.     } 
  200.  
  201. }//end for class Actions 

 

谷博濤

二零一二年 農曆臘月廿五 下午

http://blog.csdn.net/captain_gbt/article/details/7209570

 

發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章

java性能優化筆記（二）設計優化

kid2412 2019-02-24 13:34:37

Linux基本操作命令

wbzjacky 2019-02-24 13:12:38

444句英語

wbzjacky 2019-02-24 13:12:37

真實的模擬***綜合實驗

wbzjacky 2019-02-24 13:12:37

三層交換機的HSRP、vlan、端口聚合

wbzjacky 2019-02-24 13:12:37

HSRP和二層交換機的端口聚合、vlan

wbzjacky 2019-02-24 13:12:37

服務器雙機熱備解決方案

wbzjacky 2019-02-24 13:12:37

各種光模塊的傳輸距離

realzfd 2019-02-24 13:10:25

ESX4 通過VMware vSphere CLI 開啓snmp

realzfd 2019-02-24 13:09:43

SOLARWINDS服務器更改計算機名導致的一次奇怪故障

realzfd 2019-02-24 13:09:42

Dlink交換機常用命令

千年小道 2019-02-24 13:06:21

Sametime的安裝細節問題

god9394 2019-02-24 13:04:19

3G愛你沒商量

god9394 2019-02-24 13:04:19

郵件自動歸檔

god9394 2019-02-24 13:04:19

郵件分類管理

god9394 2019-02-24 13:04:19