[skip ci] Update nq (#12596)

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
pull/12601/head
wt 2021-12-02 13:11:38 +08:00 committed by GitHub
parent b89af4fcc9
commit e8442247f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions

View File

@ -64,6 +64,12 @@ def locust_executor(host, port, collection_name, connection_type="single", run_p
MyUser.tasks.update(task)
MyUser.params[op] = value["params"] if "params" in value else None
logger.info(MyUser.tasks)
_nq = nq
if "insert" in MyUser.params and "ni_per" in MyUser.params["insert"]:
ni_per = MyUser.params["insert"]["ni_per"]
_nq = ni_per + 10 if ni_per > nq else _nq
MyUser.values = {
"ids": [random.randint(1000000, 10000000) for _ in range(nb)],
"get_ids": [random.randint(1, 10000000) for _ in range(nb)],