如何通过代码向Locust传递locust-plugins扩展配置参数

方法1：模拟命令行参数（最简单） #

locust-plugins的命令行参数是通过扩展Locust的参数解析器实现的，直接在代码启动前修改sys.argv，把需要的参数加进去，Locust初始化时会自动识别这些参数：

import sys
from locust import Environment, HttpUser, task, run_single_user
from locust_plugins import TimescaleListener

# 添加locust-plugins需要的命令行参数
sys.argv.extend(["--timescale", "--pguser", "postgres", "--host", "你的主机地址"])

# 定义测试用户类
class MyUser(HttpUser):
    @task
    def my_task(self):
        self.client.get("/")

# 初始化环境并运行测试
env = Environment(user_classes=[MyUser])
run_single_user(env)

方法2：手动初始化插件（更灵活） #

如果不想依赖命令行参数解析，可以直接手动实例化locust-plugins提供的监听器，按需传入参数：

from locust import Environment, HttpUser, task, run_single_user
from locust_plugins.listeners import TimescaleListener

# 定义测试用户类
class MyUser(HttpUser):
    @task
    def my_task(self):
        self.client.get("/")

# 初始化Locust环境
env = Environment(user_classes=[MyUser], host="你的主机地址")

# 手动添加Timescale监听器并传入参数
TimescaleListener(
    env,
    pguser="postgres",
    # 按需添加其他参数，比如pgpassword、pgdatabase、pgport等
)

# 启动测试
run_single_user(env)

注意事项 #

• 确保已安装locust-plugins：pip install locust-plugins
• 若需其他参数（如--pgpassword），可在sys.argv中追加对应项，或在手动初始化时传入同名参数
• 分布式运行时，主节点和工作节点都需要做对应的参数配置

内容的提问来源于stack exchange，提问作者Sukhbir