如何配置Locust/Python脚本循环次数以控制Payload大小？

报错原因解释 #

原来的代码遍历TS_IDs.items()时，每次迭代拿到的是(ts_id, ts_name)格式的元组，代码用两个变量接收。改成range(3)后，迭代得到的是单个整数（0、1、2），但代码仍尝试将其解包为两个变量，因此触发TypeError: cannot unpack non-iterable int object。

可行解决方案 #

根据需求场景，提供两种实现方式：

方案1：重复遍历字典元素，达到指定循环次数 #

适合需要重复使用字典内元素构造Payload的场景，用itertools.cycle实现循环迭代：

from itertools import cycle
from locust import HttpUser, task

# 可配置的循环次数，后续可改为动态参数
PAYLOAD_ITEM_COUNT = 3
# 示例TS_IDs字典
TS_IDs = {"ts_001": "name1", "ts_002": "name2", "ts_003": "name3"}

class MyTestUser(HttpUser):
    @task
    def send_payload(self):
        payload = []
        # 循环迭代字典元素，直到达到指定次数
        for idx, (ts_id, ts_name) in enumerate(cycle(TS_IDs.items())):
            if idx >= PAYLOAD_ITEM_COUNT:
                break
            payload.append({"id": ts_id, "name": ts_name})
        
        self.client.post("/your-api-endpoint", json=payload)

方案2：只取字典的前N个元素构造Payload #

适合仅需使用字典前N个元素的场景，直接截取字典元素列表的前N项：

from locust import HttpUser, task

PAYLOAD_ITEM_COUNT = 3
TS_IDs = {"ts_001": "name1", "ts_002": "name2", "ts_003": "name3"}

class MyTestUser(HttpUser):
    @task
    def send_payload(self):
        payload = []
        # 将字典转成列表后截取前PAYLOAD_ITEM_COUNT个元素
        for ts_id, ts_name in list(TS_IDs.items())[:PAYLOAD_ITEM_COUNT]:
            payload.append({"id": ts_id, "name": ts_name})
        
        self.client.post("/your-api-endpoint", json=payload)

进阶：通过Locust命令行参数动态配置循环次数 #

如果需要运行时灵活调整次数，可添加自定义命令行参数：

from locust import HttpUser, task, events

# 示例TS_IDs字典
TS_IDs = {"ts_001": "name1", "ts_002": "name2", "ts_003": "name3"}

# 添加自定义命令行参数
@events.init_command_line_parser.add_option
def add_payload_count_option(parser):
    parser.add_argument(
        "--payload-count", 
        type=int, 
        default=5, 
        help="Number of items to include in the request payload"
    )

class MyTestUser(HttpUser):
    def on_start(self):
        # 从命令行参数获取配置的次数
        self.payload_count = self.environment.parsed_options.payload_count

    @task
    def send_payload(self):
        payload = []
        for idx, (ts_id, ts_name) in enumerate(cycle(TS_IDs.items())):
            if idx >= self.payload_count:
                break
            payload.append({"id": ts_id, "name": ts_name})
        
        self.client.post("/your-api-endpoint", json=payload)

运行时通过命令行指定次数：

locust -f your_script.py --payload-count=10

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