python实现websocket的客户端压力测试

 # -*- 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" # 定义进程数 processes = 4 # 定义线程数（每个文件可能限制1024个，可以修改fs.file等参数） 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(): # 设置你websocket的内容 # 每隔10秒发送一下数据使链接不中断 while True: ws.send(u'hello服务器') time.sleep(10) t <span style="color:transparent">来源gaodai#ma#com搞*!代#%^码网</span>= 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() # 设置开启线程的数量 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__": # 进程池 pool = multiprocessing.Pool(processes=processes) # 设置开启进程的数量 for i in xrange(processes): pool.apply_async(thread_web_socket) pool.close() pool.join()