Allow specifying job names for benchmarks

This commit is contained in:
2024-09-18 14:55:12 -04:00
parent 736762901c
commit 41cd34ba4d
5 changed files with 39 additions and 13 deletions

View File

@ -410,11 +410,14 @@ def run_benchmark_job(
return resource_data, jstdout
def worker_run_benchmark(zkhandler, celery, config, pool):
def worker_run_benchmark(zkhandler, celery, config, pool, name):
# Phase 0 - connect to databases
cur_time = datetime.now().isoformat(timespec="seconds")
cur_primary = zkhandler.read("base.config.primary_node")
job_name = f"{cur_time}_{cur_primary}"
if not name:
cur_time = datetime.now().isoformat(timespec="seconds")
cur_primary = zkhandler.read("base.config.primary_node")
job_name = f"{cur_time}_{cur_primary}"
else:
job_name = name
current_stage = 0
total_stages = 13