Locust 执行阶梯式压测 接口案例

以下是一个使用 Locust API 接口实现阶梯式压测的示例代码：

from locust import HttpUser, task, between
import random

class MyUser(HttpUser):
    wait_time = between(5, 9)
    min_users = 10
    max_users = 100
    step_users = 10
    step_time = 5

    @task
    def my_task(self):
        self.client.get("/")

    def on_start(self):
        self.user_count = self.min_users
        self.set_user_count()

    def set_user_count(self):
        self.environment.runner.user_count = self.user_count

    def on_stop(self):
        self.user_count = 0
        self.set_user_count()

    def on_step(self, step):
        if (step + 1) % self.step_time == 0 and self.user_count < self.max_users:
            self.user_count += self.step_users
            self.set_user_count()

这个示例代码定义了一个名为 MyUser 的用户类，其中定义了压测任务 my_task，用于模拟用户访问网站的行为。在 on_start 方法中，初始化用户数为最小用户数 min_users，并调用 set_user_count 方法设置 Locust 运行时的用户数。在 on_stop 方法中，停止用户数，并将 Locust 运行时的用户数设置为 0。

在 on_step 方法中，每隔一定时间（由 step_time 参数控制），将用户数增加 step_users（由 step_users 参数控制），并调用 set_user_count 方法，重新设置 Locust 运行时的用户数。

需要注意的是，这个示例代码仅作为演示使用，实际使用时需要根据测试场景和网站实际情况进行调整。同时，需要使用命令行启动 Locust 进行压测，并传递相关参数。具体命令行参数可以参考 Locust 的官方文档。