Compare commits
2 Commits
2fb7c40497
...
a461791ce8
Author | SHA1 | Date |
---|---|---|
Joshua Boniface | a461791ce8 | |
Joshua Boniface | 9fdb6d8708 |
|
@ -115,9 +115,10 @@ class BenchmarkError(Exception):
|
||||||
#
|
#
|
||||||
|
|
||||||
|
|
||||||
def cleanup(job_name, db_conn=None, db_cur=None, zkhandler=None):
|
def cleanup(job_name, db_conn=None, db_cur=None, zkhandler=None, final=False):
|
||||||
if db_conn is not None and db_cur is not None:
|
if db_conn is not None and db_cur is not None:
|
||||||
# Clean up our dangling result
|
if not final:
|
||||||
|
# Clean up our dangling result (non-final runs only)
|
||||||
query = "DELETE FROM storage_benchmarks WHERE job = %s;"
|
query = "DELETE FROM storage_benchmarks WHERE job = %s;"
|
||||||
args = (job_name,)
|
args = (job_name,)
|
||||||
db_cur.execute(query, args)
|
db_cur.execute(query, args)
|
||||||
|
@ -410,6 +411,7 @@ def worker_run_benchmark(zkhandler, celery, config, pool):
|
||||||
db_conn=db_conn,
|
db_conn=db_conn,
|
||||||
db_cur=db_cur,
|
db_cur=db_cur,
|
||||||
zkhandler=zkhandler,
|
zkhandler=zkhandler,
|
||||||
|
final=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
current_stage += 1
|
current_stage += 1
|
||||||
|
|
|
@ -158,9 +158,9 @@ def getNodeInformation(zkhandler, node_name):
|
||||||
zkhandler, node_name, node_health_plugins
|
zkhandler, node_name, node_health_plugins
|
||||||
)
|
)
|
||||||
|
|
||||||
if _node_network_stats is not None:
|
try:
|
||||||
node_network_stats = json.loads(_node_network_stats)
|
node_network_stats = json.loads(_node_network_stats)
|
||||||
else:
|
except Exception:
|
||||||
node_network_stats = dict()
|
node_network_stats = dict()
|
||||||
|
|
||||||
# Construct a data structure to represent the data
|
# Construct a data structure to represent the data
|
||||||
|
|
Loading…
Reference in New Issue