Locust——对其他协议(非http协议)的脚本开发和使用

　　Locust是一款可扩展的，分布式的，性能测试的，开源的，用Python编写的性能测试工具。对于测试HTTP协议的接口是比较方便的，但是它也支持测试别的协议的接口，不过需要重写Locust类。

　　接下来以websocket协议为例，记录一下Locust脚本的写法。

　　1.要想在python中使用websocket协议，首先需要安装python的websocket模块，Ubuntu系统中可以通过"pip install websocket-client"来安装。

　　2.在python中，我们可以通过这样的方式来连接服务器和发送消息：

　　　ws = websocket.WebSocket()

　　　ws.connect("ws://10.23.132.178:8807")

　　　ws.send('{"url":"Game/Tool/test","data":{"fd":1}}')

　　　result = ws.recv()

　　当然，locust脚本中也可以这样参考着写。

　　3.对locust类重写:

class WebSocketClient():    def __init__(self, host, port):        self.host = host        self.port = portclass WebSocketLocust(Locust):    def __init__(self, *args, **kwargs):        self.client = WebSocketClient("10.38.74.123", 9501)...　...
class ApiUser(WebSocketLocust):    min_wait = 1000    max_wait = 2000    task_set = UserBehavior

　　4.对发送消息相关信息按照如下方式进行重新设置，这样locust才能捕捉到消息数以及响应时间等消息，才能在监控主页(127.0.0.1:8089)上显示出正确的时间和RPS。

        try:            start_time = time.time()            self.ws.send('{"url":"/buy","data":{"id":"123","issue":"20170822","doubled_num":2}}')            result = self.ws.recv()        except xmlrpclib.Fault as e:            total_time = int((time.time() - start_time) * 1000)            events.request_failure.fire(request_type="tcp", name="buy", response_time=total_time, exception=e)        else:            total_time = int((time.time() - start_time) * 1000)            events.request_success.fire(request_type="tcp", name="buy", response_time=total_time,                                        response_length=0)

　　5.最后贴出完整代码:

　　

from locust import Locust, events, task, TaskSetimport websocketclass WebSocketClient():    def __init__(self, host, port):        self.host = host        self.port = portclass WebSocketLocust(Locust):    def __init__(self, *args, **kwargs):        self.client = WebSocketClient("10.98.64.101", 9503)class UserBehavior(TaskSet):    self.ws = websocket.WebSocket()    self.ws.connect("ws://10.98.64.103:8807")    @task(1)    def buy(self):        try:            start_time = time.time()            self.ws.send('{"url":"/buy","data":{"id":"123","issue":"20170822","doubled_num":2}}')            result = self.ws.recv()        except xmlrpclib.Fault as e:            total_time = int((time.time() - start_time) * 1000)            events.request_failure.fire(request_type="tcp", name="buy", response_time=total_time, exception=e)        else:            total_time = int((time.time() - start_time) * 1000)            events.request_success.fire(request_type="tcp", name="buy", response_time=total_time,                                        response_length=0)class ApiUser(TcpLocust):    min_wait = 1000    max_wait = 2000    task_set = UserBehavior　　5.注：ws要作为self的一个对象来定义，这样可以防止之后的task出现找不到ws的情况。