如何在Docker Compose中为Locust分布式集群的每个Worker设自定义参数

直接用docker-compose up --scale worker=3默认会启动3个完全相同的Worker实例，没法直接给每个实例分配不同参数，但可以通过以下几种方式实现需求：

方法1：为每个Worker定义独立的服务配置 #

这种方式最直观，无需依赖scale命令，直接在docker-compose.yml里为每个Worker单独编写服务块，各自指定不同参数：

version: '3'
services:
  master:
    image: locustio/locust
    ports:
      - "8089:8089"
    volumes:
      - ./locustfile.py:/mnt/locust/locustfile.py
    command: -f /mnt/locust/locustfile.py --master

  worker1:
    image: locustio/locust
    volumes:
      - ./locustfile.py:/mnt/locust/locustfile.py
    command: -f /mnt/locust/locustfile.py --worker --master-host master --my-custom-param value1

  worker2:
    image: locustio/locust
    volumes:
      - ./locustfile.py:/mnt/locust/locustfile.py
    command: -f /mnt/locust/locustfile.py --worker --master-host master --my-custom-param value2

  worker3:
    image: locustio/locust
    volumes:
      - ./locustfile.py:/mnt/locust/locustfile.py
    command: -f /mnt/locust/locustfile.py --worker --master-host master --my-custom-param value3

启动时直接运行docker-compose up -d，每个Worker都会加载对应的自定义参数。

方法2：利用环境变量+启动脚本动态传递参数 #

如果想保留灵活增减Worker的能力，可以通过启动脚本结合环境变量实现动态参数配置：

步骤1：编写启动脚本start-worker.sh #

#!/bin/sh
# 从环境变量读取自定义参数，可设置默认值
CUSTOM_PARAM=${CUSTOM_PARAM:-default-value}
# 启动Locust Worker并传入参数
locust -f /mnt/locust/locustfile.py --worker --master-host master --my-custom-param $CUSTOM_PARAM

给脚本添加执行权限：chmod +x start-worker.sh

步骤2：修改docker-compose.yml #

version: '3'
services:
  master:
    image: locustio/locust
    ports:
      - "8089:8089"
    volumes:
      - ./locustfile.py:/mnt/locust/locustfile.py
    command: -f /mnt/locust/locustfile.py --master

  worker:
    image: locustio/locust
    volumes:
      - ./locustfile.py:/mnt/locust/locustfile.py
      - ./start-worker.sh:/mnt/locust/start-worker.sh
    entrypoint: /mnt/locust/start-worker.sh

步骤3：启动多参数Worker #

通过docker-compose run分别启动每个Worker，传递不同环境变量：

# 启动第一个Worker，参数为value1
docker-compose run -d -e CUSTOM_PARAM=value1 worker
# 启动第二个Worker，参数为value2
docker-compose run -d -e CUSTOM_PARAM=value2 worker
# 启动第三个Worker，参数为value3
docker-compose run -d -e CUSTOM_PARAM=value3 worker

这种方式既可以灵活控制Worker数量，又能为每个实例分配独立参数，比scale命令更适配你的需求。

注意事项 #

• 确保你的Locust测试代码已经通过argparse或其他方式正确接收并处理--my-custom-param这类自定义参数。
• 确认所有Worker能通过Docker Compose默认网络访问Master节点（直接用服务名master即可）。

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