python多线程连接websocket的测试脚本

本人作为测试人员，对现有服务端的websocket协议进行测试
当前脚本保持100个websocket连接，从而发现测试服务器的瓶颈在网络
测试服务器的带宽也确实不高才1M
同时连接多个WK，，网络资源占用消耗比较高

#coding=utf-8

from ws4py.client.threadedclient import WebSocketClient
from ws4py import websocket
from framework.testCoin import loginByMobile,getTable,wsHost
from framework.logger2 import Logger
import json,threading,time

logger = Logger(logger='zhajinhua').getlog()

class Ws(WebSocketClient):

    def __init__(self,url,userId,token,tableId):
        super().__init__(url)
        self.userId = userId
        self.token = token
        self.tableId = tableId

    def opened(self):                   		#socket连接后调用
        ws101 = json.dumps({
            "cammand":101,
            "data":{
                "userId":self.userId,
                "token":self.token,
                "tableId":self.tableId,
                "gameType":"zhajinhua"
            }
        })

        self.send(ws101)

    def closed(self, code, reason=None):        	#wk关闭时调用
        print(code,time.strftime('%Y-%m-%d %H:%M:%S'))

    def received_message(self, message):   		 #长连接时接收socket的所有消息,可根据消息做对应处理
        ms = json.loads(str(message))

        #加入桌子
        ws99 = json.dumps({
            "cammand": 99,
            "gameType": "zhajinhua"
        })

        if ms['cammand'] == 101:
            self.send(ws99)
            
        '''省略部分'''

def start(mobile):

    loginInfo = loginByMobile(mobile)                                   #登录
    tableId = getTable(loginInfo['token'],11)                           #获取桌子ID

    ws = Ws(wsHost,loginInfo['userId'],loginInfo['token'],tableId)      #创建websocket
    ws.connect()                                                        #连接wk
    websocket.Heartbeat(ws).run()                                       #发送心跳
    ws.run_forever()                                                    #运行


if __name__ == '__main__':
    mobile = 13600000001
    for i in range(100):
        th = threading.Thread(target=start,args=(mobile,))              #创建线程
        th.start()                                                      #启动线程
        time.sleep(1)
        mobile += 1