测试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':'0','boxArriveFlag':'0','boxCollectFlag':'0'})

    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级别也会打印出一些引用组建的调试代码，通过这种方式规避下