多进程分布式运行locust 示例代码

以下是一个简单的示例代码，用于在多进程分布式环境下运行 Locust：

from locust import HttpUser, task, between
from locust.contrib.fasthttp import FastHttpUser
from locust import LoadTestShape
from locust import events
from locust.runners import MasterRunner, WorkerRunner
import gevent
from gevent import Greenlet

class MyUser(HttpUser):
    wait_time = between(1, 5)

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

class MyShape(LoadTestShape):
    def tick(self):
        run_time = self.get_run_time()
        if run_time < 60:
            return (1, 5)
        else:
            return None

def run_locust(user_class, host, num_users, spawn_rate, run_time):
    runner = MasterRunner(host=host)
    greenlets = []
    for _ in range(num_users):
        worker = WorkerRunner()
        greenlet = Greenlet(worker.start, user_class, spawn_rate, None, runner, {}, None, None)
        greenlet.start()
        greenlet.link_exception(lambda g: runner.quit())
        greenlets.append(greenlet)

    def spawn_users():
        for i in range(num_users):
            runner.hatch(user_class, spawn_rate=spawn_rate)

    def stop_users():
        runner.stop()

    events.spawn_users += spawn_users
    events.stop_users += stop_users

    runner.start(MyShape(run_time))

    for g in greenlets:
        g.join()

if __name__ == "__main__":
    run_locust(MyUser, "http://localhost:8080", 10, 5, 60)

这个示例代码演示了如何在多进程分布式环境下运行 Locust。在这个示例代码中，`run_locust` 函数接受 `user_class`、`host`、`num_users`、`spawn_rate` 和 `run_time` 这些参数。`user_class` 是 Locust 用户类，`host` 是要测试的主机地址，`num_users` 是要启动的用户数，`spawn_rate` 是每秒启动的用户数，`run_time` 是测试运行的时间。

在 `run_locust` 函数中，我们创建了一个 `MasterRunner` 对象，并启动了多个 `WorkerRunner` 对象。然后，我们使用 Greenlet 来启动每个 `WorkerRunner` 对象，并将它们添加到 `MasterRunner` 中。最后，我们使用 `events.spawn_users` 和 `events.stop_users` 事件来控制用户生成和停止。

要运行这个示例代码，只需在控制台中输入以下命令：

locust -f locustfile.py --no-web --master

然后在另一个控制台中输入以下命令：

locust -f locustfile.py --no-web --worker --master-host=localhost

其中，`--no-web` 参数表示不启动 Locust Web UI，`--master` 参数表示本机是 Master 节点，`--worker` 参数表示本机是 Worker 节点，`--master-host` 参数表示 Master 节点的主机地址。