常见场景：

全局数据

同一份数据，随机取数

同一份数据，顺序取数

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

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

  1. def work_id_data():
  2.     """队列获取作品ID数据"""
  3.     # work_ids = queue.Queue()
  4.     with open('work_id.txt', 'r', encoding='utf-8') as f:
  5.         for i in f.readlines()[int(os.environ.get('first_index')):int(os.environ.get('last_index'))]:
  6.             work_ids.put_nowait(int(i.strip('\n')))
  7.     return work_ids
  10. @events.init_command_line_parser.add_listener
  11. def add_user_parser(parser, **kwargs):
  12.     """自定义locust命令参数"""
  13.     parser.add_argument(
  14.         "-fi",
  15.         "--first_index",
  16.         help="获取截取列表首索引"
  17.              "[fc:]",
  18.         default=0,
  19.         type=str
  20.     )
  21.     parser.add_argument(
  22.         "-li",
  23.         "--last_index",
  24.         help="获取截取列表末索引"
  25.              "[:li]",
  26.         default=0,
  27.         type=str
  28.     )
  29.     args = parser.parse_args()
  30.     if args.first_index:
  31.         os.environ['first_index'] = args.first_index
  32.     if args.last_index:
  33.         os.environ['last_index'] = args.last_index
  36. @events.init.add_listener
  37. def on_locust_init(environment, **kwargs):
  38.     print("当前环境变量first_index为： %s" % os.environ.get('first_index'))
  39.     print("当前环境变量last_index为： %s" % os.environ.get('last_index'))
  40.     work_id_data()
  41.     print('初始化测试数据完毕')
  42.     if isinstance(environment.runner, MasterRunner):
  43.         print("master节点执行")
  44.     else:
  45.         print("worker节点执行")
  47. if __name__ == "__main__":
  48.     file_name = os.path.abspath(__file__)
  49.     os.system(f'locust -f {file_name} --master --first_index 0 --last_index 1000')

非全局数据

动态入参，与上下文无联系（例如时间戳、uuid）

动态入参，与上下文联系（请求数据递增版本号）

通过函数方法调用，以字典方式存储，结合类属性定义索引，动态取数，数据针对模拟用户独立唯一。

  2. def version_id_data():
  3.     """生成version_ids字典"""
  4.     version_id = []
  5.     for i in range(1000000):
  6.         version_id.append(i)
  7.     version_ids = dict(zip(version_id, version_id))
  8.     return version_ids
  11. class ApiUser(TaskSet):
  12.     """压测op操作"""
  13.     index = 1
  15.     @task(15)
  16.     def test_op(self):
  17.         """op操作"""
  18.         all_locusts_spawned.wait()  # 限制在所有用户准备完成前处于等待状态
  19.         self.client.send('42["operation",{'
  20.                          f'"revision":{self.user.version_id.get(self.index)}'
  21.                          ',"op":[[["extra","user_state","275127340","current_stage_actor",{"r":true,"i":'
  22.                          f'"{uuid.uuid1()}"'
  23.                          '}]]]}]')
  24.         self.index += 1