locust-实现集合点及随机数参数化

Locust_gather_****.py

from locust import events
from gevent._semaphore import Semaphore
all_locusts_spawned = Semaphore()
all_locusts_spawned.acquire()

def on_hatch_complete(**kwargs):
    all_locusts_spawned.release() ##创建钩子方法


# 挂在到locust钩子函数（所有的Locust实例产生完成时触发）
# events.hatch_complete += on_hatch_complete  # 1.0之前的写法
events.spawning_complete.add_listener(on_hatch_complete)  # 1.0之后的写法

user_portrait.py

# -*- coding: utf-8 -*-
# Author : Ethan
# Time : 2023/3/30 21:20
from locust import HttpUser,task,between,constant,constant_pacing,tag,User
import urllib3
import requests,uuid,time
import requests
urllib3.disable_warnings()
import random,os,queue
from Locust_gather_**** import *

class user_por(HttpUser):
    wait_time =between(0,0)
    @task
    def get_user_labels(self):
        self.user_id = str(int(time.time() * 10000000)) + str(random.randint(1, 1000))
        self.url='/feign/user/labels'
        self.data= [{"userId":self.user_id,"firstDimension":"TAKEOUT","thirdDimension":"","secondDimension":"STOREID:18353"}]
        self.header={
  'Content-Type': 'application/json'
    }
        all_locusts_spawned.wait()
        with self.client.post(url=self.url,headers=self.header,json=self.data,verify=False ,catch_response=True) as response:
            try:
                userLabels =response.json()[0]['userLabels']
            except Exception as e:
                response.failure(response.content)
                print(self.user_id)
            else:
                    if "TAKEOUT_STORE_" in str(userLabels):
                        response.success()
                    else:
                        response.failure(response.json())

    def on_start(self):
        pass
    def on_stop(self):
        pass

if __name__ == '__main__':
    os.system('locust  -f user_portrait.py  --host=http://*****.cf93fe1dde8584346a2d81575c79aff9e.cn-shenzhen.alicontainer.com')