python socket 长连接_python_websocket长连接压测

from threadpool import ThreadPool, makeRequests

import websocket

import threading

def on_message(ws, message):

print(message)

def on_error(ws, error):

print(error)

def on_close(ws):

print("### closed ###")

def on_open(ws):

def send_trhead():

while True:

time.sleep(5)

ws.send("test")

t = threading.Thread(target=send_trhead)

t.start()

def connect(a):

time.sleep(a % 1)

websocket.enableTrace(True)

ws = websocket.WebSocketApp("压测地址",

on_message=on_message,

on_error=on_error,

on_close=on_close)

ws.on_open = on_open

ws.run_forever(ping_interval=60, ping_timeout=5)

return ws

if __name__ == "__main__":

pool = ThreadPool(100)

test = list()

for i in range(100):

test.append(i)

result = makeRequests(connect, test)

[pool.putRequest(req) for req in result]

pool.wait()