locust如何多个worker共享同一批用户, worker 1 有vu 张三 李四, worker 2 有用户 王五 赵六

要让多个worker共享同一批用户，可以使用分布式任务队列来实现。

1. 首先，需要在locustfile.py中定义一个任务队列，例如：

from locust import HttpUser, task, between, TaskSet

class MyTaskSet(TaskSet):
    @task
    def my_task(self):
      # 执行任务内容
      
shared_tasks = MyTaskSet.tasks

2. 然后，在启动locust时，使用命令行参数“--master”来启动一个主节点，使用命令行参数“--worker”来启动多个工作节点。例如：

locust -f locustfile.py --master
locust -f locustfile.py --worker
locust -f locustfile.py --worker

3. 接下来，在主节点上，使用以下代码将任务队列发送到任务队列RabbitMQ：

from locust import HttpUser, task, between, TaskSet
from locust.runners import MasterRunner
from locust.exception import StopUser

shared_tasks = ...

class MyUser(HttpUser):
    tasks = shared_tasks
    wait_time = between(0.5, 5)

    # 在测试结束时停止所有用户
    def stop(self, *args, **kwargs):
        raise StopUser()

if __name__ == "__main__":
    runner = MasterRunner([MyUser])
    shared_tasks.setup()
    runner.send_task("my_task", shared_tasks)

4. 最后，在工作节点上，使用以下代码接收任务队列并执行：

from locust import HttpUser, task, between, TaskSet
from locust.runners import WorkerRunner
from locust.rpc import rpc

class MyUser(HttpUser):
    tasks = []
    wait_time = between(0.5, 5)

    @task
    def my_task(self):
        # 执行任务内容

if __name__ == "__main__":
    worker = WorkerRunner([MyUser])
    rpc.Connection(worker).run()

这样，多个工作节点就可以共享同一批用户了。注意，如果需要在运行过程中修改任务队列，可以在共享代码的位置修改，并在主节点上重新发送任务队列。工作节点会在下一次接收任务队列时生效。