浅谈locust 性能压测使用

1. 基本介绍

Locust是一个开源的负载测试工具，用于模拟大量用户并发访问一个系统或服务，以评估其性能和稳定性。编写语言为Python，可通过Python来自定义构建性能压测场景脚本。Locust支持分

布式负载测试，可以通过多个机器协同工作来模拟大量用户并发访问。并提供了一个Web界面，可以实时监控并查看测试的性能指标和统计数据，并支持常见的协议，如HTTP、HTTPS、

FTP、WebSocket等，更重要的是具有良好的扩展性，用户可以编写自定义的扩展来满足特定需求。它还提供了多个钩子函数，可以在测试运行过程中插入自定义代码，高度灵活，随心创造。

2. 基本组成

User 类（User class）：
　　在Locust脚本中，必须定义一个继承自HttpUser或User类的用户类。这个用户类代表了模拟的用户，并定义了用户的行为和任务。用户类中的每个方法都代表一个用户行为，例如发送请求、处理响应等。

Task 类（Task class）：
　　用户类中的方法通常被称为任务（task），每个任务定义了用户的一个具体行为。任务可以是一个函数、一个装饰器或一个类方法。任务通常用于发送请求、处理响应等操作。

TaskSet 类（TaskSet class）：
　　TaskSet类是用于组织任务的一种方式。它可以包含多个任务，并定义了用户在执行这些任务时的行为和逻辑。通过TaskSet，可以更好地组织和管理任务，使脚本的结构更加清晰。

Weight 属性（Weight attribute）：
　　在用户类中，可以为每个任务指定一个权重（weight）。权重用于控制每个任务在负载测试中的执行频率。较高权重的任务将更频繁地执行，而较低权重的任务将相对较少执行。

on_start() 方法：
　　用户类中的on_start()方法定义了每个用户在开始执行任务之前要执行的初始化操作。例如，可以在on_start()方法中进行用户登录或准备测试数据等操作。

on_stop() 方法：
用户类中的on_stop()方法定义了每个用户在完成所有任务后要执行的清理操作。例如，可以在on_stop()方法中进行用户退出或清理资源等操作。

Host 属性：
在用户类中，可以定义一个host属性，指定要测试的目标主机地址。这个属性告诉Locust在执行请求时要发送到哪个主机。

TaskSet 属性：
用户类中可以定义一个或多个TaskSet属性，用于组织和管理任务。通过TaskSet属性，可以将任务分组并按组执行，使脚本更加模块化和可维护。

3. 基本使用步骤

定义用户行为和任务：
首先，需要定义用户的行为和任务，即编写Locust脚本。在脚本中，定义一个继承自HttpUser或User类的用户类，并在其中定义任务和行为。例如，发送请求、处理响应等。

from locust import HttpUser, task, between

class MyUser(HttpUser):
    wait_time = between(2, 5) # 定义用户等待时间范围

    @task
    def my_task(self):
        self.client.get("/path") # 发送GET请求

    @task(3)  # 指定任务权重
    def weighted_task(self):
        response = self.client.post("/path", json={"key": "value"})  # 发送POST请求
        if response.status_code == 200:
            self.log_success("POST request succeeded", response.elapsed.total_seconds())
        else:
            self.log_failure("POST request failed", response.elapsed.total_seconds())

    def log_success(self, message, response_time):  # 定义请求成功和失败处理
        self.environment.events.request_success.fire(
            request_type="my_task",
            name=message,
            response_time=response_time,
            response_length=0,
        )

    def log_failure(self, message, response_time):
        self.environment.events.request_failure.fire(
            request_type="my_task",
            name=message,
            response_time=response_time,
            exception=None,
            response_length=0,
        )

　

设置负载测试参数：
在脚本中，可以设置一些负载测试的参数，如用户的并发数量、每秒的请求数等。

class MyUser(HttpUser):

    tasks = [MyUser]  # 定义要执行的用户类
    wait_time = constant(1)  # 设置每个用户的等待时间
    host = "https://example.com"  # 设置目标主机地址
    min_wait = 5000  # 设置用户最小等待时间
    max_wait = 10000  # 设置用户最大等待时间
    users = 100  # 设置并发用户数
    spawn_rate = 10  # 设置每秒新增用户数

启动 Locust　

locust -f my_script.py　

访问 Locust Web 界面：
启动Locust后，可以通过访问Locust的Web界面进行监控和控制。默认情况下，Web界面监听在http://localhost:8089。

总之，Locust是一个功能强大、易用且灵活的负载测试工具，可用于模拟大规模用户并发访问，并评估系统的性能和稳定性。