python實(shí)現(xiàn)websocket的客戶端壓力測(cè)試

# -*- coding:utf-8 -*-
# __author__ == 'chenmingle'
 
import websocket
import time
import threading
import json
import multiprocessing
import uuid
from threadpool import ThreadPool, makeRequests
 
# 修改成自己的websocket地址
WS_URL = "xxxx"
# 定義進(jìn)程數(shù)
processes = 4
# 定義線程數(shù)（每個(gè)文件可能限制1024個(gè)，可以修改fs.file等參數(shù)）
thread_num = 700
index = 1
 
 
def on_message(ws, message):
 # print(message)
 pass
 
 
def on_error(ws, error):
 print(error)
 pass
 
 
def on_close(ws):
 # print("### closed ###")
 pass
 
 
def on_open(ws):
 global index
 index = index + 1
 
 def send_thread():
  # 設(shè)置你websocket的內(nèi)容
  # 每隔10秒發(fā)送一下數(shù)據(jù)使鏈接不中斷
  while True:
   ws.send(u'hello服務(wù)器')
   time.sleep(10)
 
 t = threading.Thread(target=send_thread)
 t.start()
 
 
def on_start(num):
 time.sleep(5)
 # websocket.enableTrace(True)
 ws = websocket.WebSocketApp(WS_URL + str(num),
        on_message=on_message,
        on_error=on_error,
        on_close=on_close)
 ws.on_open = on_open
 ws.run_forever()
 
 
def thread_web_socket():
 # 線程池
 pool_list = ThreadPool(thread_num)
 num = list()
 # 設(shè)置開啟線程的數(shù)量
 for ir in range(thread_num):
  num.append(ir)
 requests = makeRequests(on_start, num)
 [pool_list.putRequest(req) for req in requests]
 pool_list.wait()
 
 
if __name__ == "__main__":
 # 進(jìn)程池
 pool = multiprocessing.Pool(processes=processes)
 # 設(shè)置開啟進(jìn)程的數(shù)量
 for i in xrange(processes):
  pool.apply_async(thread_web_socket)
 pool.close()
 pool.join()