Locust测试数据参数化

目录

全局数据

非全局数据

---

常见场景：

全局数据

同一份数据，随机取数

同一份数据，顺序取数

每个worker节点拥有独立的一份数据 & 不重复

        通过init事件注册监听，可结合init_command_line_parser事件配置全局数据，数据针对worker独立唯一。

def work_id_data():
    """队列获取作品ID数据"""
    # work_ids = queue.Queue()
    with open('work_id.txt', 'r', encoding='utf-8') as f:
        for i in f.readlines()[int(os.environ.get('first_index')):int(os.environ.get('last_index'))]:
            work_ids.put_nowait(int(i.strip('\n')))
    return work_ids


@events.init_command_line_parser.add_listener
def add_user_parser(parser, **kwargs):
    """自定义locust命令参数"""
    parser.add_argument(
        "-fi",
        "--first_index",
        help="获取截取列表首索引"
             "[fc:]",