测试Websocket建立通信，使用protobuf格式交换数据

　　接到一个应用测试，应用实现主要使用websocket保持长链接，使用protobuf格式交换数据，用途为发送消息，需要我们测试评估性能，初步评估需要测试长链接数、峰值消息数以及长期运行稳定性

　　整体需求不难，主要困难是第一次接触websocket和protobuf，同时也没有现成的脚本可以使用，jmeter原生也不支持websocket协议，需要安装插件支持，protobuf格式也需要重新编译大包使用

　　万里长征第一步，先开发测试脚本，由于目前一直在使用python开发，评估项目的性能要求，不需要严格性能指标，所以直接使用python作为开发语言编写脚本，主要实现脚本如下

　　入口主文件，mulitprocess.py，多进程发起建立链接

 # -*- coding: utf-8 -*-
 # @Author  : monleylu
 # @Time    : 2018/7/5 12:22 PM

 from multiprocessing import Process,Pool
 import os
 from data import user
 from protobufcon import ProtoService
 import tnlog,logging

 def consumer(host,port,username):
     logging.error('启动子进程： %s...' % (os.getpid()))
     kwargs = {"host": host, "port": port, "username": username }
     pro=ProtoService(**kwargs)
     pro.startWebsocket()

 def consumer2(*args,**kwargs):
     logging.error('启动子进程： %s...' % (os.getpid()))
     pro = ProtoService(**kwargs)
     pro.startWebsocket()

 if  __name__ =="__main__":
     logging.error('启动主进程 %s.' % os.getpid())

     p = Pool(450)
     for i in range(450):
         #第一种传参数方法
         # p.apply_async(consumer, args=("11.31.60.10",8880,user.pop(),))

         #第二种传参数方法
         kw={"host": "11.31.60.10", "port": 8880, "username": user.pop()}
         p.apply_async(consumer, args=(),kwds=kw)

     logging.error('Waiting for all subprocesses done...')

     p.close()
     p.join()
     logging.error('All subprocesses done.')

　　实现代码，protobuffun.py，主要负责发送、接收消息

 # -*- coding: utf-8 -*-
 # @Author  : monleylu
 # @Time    : 2018/7/5 12:15 PM

 import protoinfo.msg_pb2 as msgpb2
 import websocket,json,datetime
 import json
 try:
     import thread
 except ImportError:
     import _thread as thread
 import tnlog,logging
 from data import user
 class ProtoService:

     def __init__(self,*,host="10.30.66.11",port=8886,username="monleylu"):
         self.host=host
         self.port=port
         self.username=username

     def on_message(self,ws, message):
         msg2 = msgpb2.Msg()
         msg2.ParseFromString(message)
         logging.error("用户:{} 收到消息，类型:{},内容:{}".format(self.username,msg2.method,msg2.data))
         if msg2.method == "login":
             b=msg2.data
             c=json.loads(b)
             self.sendprotobuf(ws,"serve_get",{'uc_id':c["uc_id"],'boxReadFlag':'','boxArriveFlag':'','boxCollectFlag':''})

     def on_error(self,ws, error):
         logging.error("发生异常:{}".format(error))

     def on_close(self,ws):
         ws.close()
         logging.error("### closed ###")

     # def wrapp(username):
     #     def w(*args,**kwargs):

     def on_open(self,ws):
         self.sendprotobuf(ws, "connect", {'uc_name': self.username})

     def sendprotobuf(self,ws,method,data):
         msg = msgpb2.Msg()
         msg.method = method
         msg.data = json.dumps(data)
         ws.send(msg.SerializeToString())
         logging.error("发送消息，类型:{},内容:{}".format(msg.method, msg.data))

     def startWebsocket(self):
         # websocket.enableTrace(True)
         logging.error("连接服务器:{},端口：{},用户名:{}".format(self.host,self.port,self.username))
         ws = websocket.WebSocketApp("ws://{}:{}".format(self.host,self.port),
                                     on_message=self.on_message,
                                     on_error=self.on_error,
                                     on_close=self.on_close)
         ws.on_open = self.on_open

         # ws.run_forever(ping_interval=5)
         logging.error(datetime.datetime.now())
         ws.run_forever(ping_interval=5)
         # ws.run_forever()
         logging.error(datetime.datetime.now())

 #10.30.66.11

 if __name__ == "__main__":
     kwargs = {"host": "10.30.66.11", "port": 8886, "username": "liangyuanfang", }
     p=ProtoService(**kwargs)
     p.startWebsocket()

查询运行过程中，会把日志写到对应的文件中，通过统计文件内容分析测试结果

这里可能各位也注意到写日志，代码使用了error级别，因为之前在调试时，info或者debug级别也会打印出一些引用组建的调试代码，通过这种方式规避下