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

最新推荐文章于 2024-09-24 03:47:05 发布

原创最新推荐文章于 2024-09-24 03:47:05 发布 · 5.6k 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#locust #python #websocket #性能测试

性能测试工具专栏收录该内容

2 篇文章

订阅专栏

本文介绍如何利用Locust工具进行WebSocket协议的性能测试，包括安装必要的Python库、自定义Locust类、设置发送消息及接收响应的时间记录等步骤。

　　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 = port

class 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, TaskSet
import websocket

class WebSocketClient():
    def __init__(self, host, port):
        self.host = host
        self.port = port

class WebSocketLocust(Locust):
    def __init__(self, *args, **kwargs):
        self.client = WebSocketClient("10.98.64.103", 8807)

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()
            # websocket请求
            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的情况。