一個高併發的測試websocket腳本
根據最近公司上線一個直播項目,但是有一天測試那邊一臉苦逼的來找到我,問我有什麼好工具可以實現高併發的測試,他使用jmeter限制了400個併發數(個人沒去研究過不知道是否真的限制),後來我第一時間想到python寫一個腳本,也沒想多小就答應幫他搞一個測試腳本。
首先整理思:
- 要實現高並發不到兩點
- 多進程
- 多線程
首先安裝先要環境:
pip install websocket
pip install threadpool
pip install websocket-client
pip install multiprocessing
直接上腳本:
#!/usr/bin/python
#-*- coding:utf-8 -*-
#__author__ == 'chenmingle'
import websocket
import time
import threading
import json
import multiprocessing
from threadpool import ThreadPool, makeRequests
#修改成自己的websocket地址
WS_URL = "wss://ws.test.com/"
#定義進程數
processes=5
#定義線程數(每個文件可能限制1024個,可以修改fs.file等參數)
thread_num=1000
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):
def send_trhead():
#設置你websocket的內容
send_info = {"cmd": "refresh", "data": {"room_id": "58", "wx_user_id": 56431}}
#每隔10秒發送一下數據使鏈接不中斷
while True:
time.sleep(10)
ws.send(json.dumps(send_info))
t = threading.Thread(target=send_trhead)
t.start()
def on_start(num):
time.sleep(num%20)
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 = ThreadPool(thread_num)
num = list()
#設置開啓線程的數量
for ir in range(thread_num):
num.append(ir)
requests = makeRequests(on_start, num)
[pool.putRequest(req) for req in requests]
pool.wait()
if __name__ == "__main__":
#進程池
pool = multiprocessing.Pool(processes=processes)
#設置開啓進程的數量
for i in xrange(processes):
pool.apply_async(thread_web_socket)
pool.close()
pool.join()
這樣就完成一個高併發的websocket測試腳本,如果覺得這文章不錯的,請在本人Github上點個star,感謝!!
參考資料: