logos-blockchain/logos-blockchain-simulations
1
0
Fork 0
mirror of https://github.com/logos-blockchain/logos-blockchain-simulations.git synced 2026-01-02 13:13:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

update batch.py for log topic

Browse Source
This commit is contained in:
Youngjoon Lee 2024-12-20 16:50:05 +09:00
parent 8d227dac75
commit 97fd438dca
No known key found for this signature in database
GPG Key ID: 303963A54A81DD4D
1 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
simlib/blendnet-sims/scripts/batch.py
View File

@ -11,33 +11,35 @@ STEP_DURATION_MS = 50
def bandwidth_result(log_path: str) -> dict[str, float]:
step_id = 0
max_step_id = 0
for _, json_msg in mixlog.get_input_stream(log_path):
if (step_id := json_msg.get("step_id")) is not None:
max_step_id = max(max_step_id, step_id)
for line in mixlog.get_input_stream(log_path):
if '"payload_id"' in line and '"step_id"' in line:
step_id = max(step_id, json.loads(line)["step_id"])
with open(log_path, "r") as file:
for line in file:
if "total_outbound_bandwidth" in line:
line = line[line.find("{") :]
line = line.replace("{ ", '{"')
line = line.replace(": ", '": ')
line = line.replace(", ", ', "')
record = json.loads(line)
if "total_outbound_bandwidth" in line:
line = line.replace("{ ", '{"')
line = line.replace(": ", '": ')
line = line.replace(", ", ', "')
record = json.loads(line)
elapsed = (step_id * STEP_DURATION_MS) / 1000.0
return {
"min": float(record["min_node_total_bandwidth"]) / elapsed,
"avg": float(record["avg_node_total_bandwidth"]) / elapsed,
"max": float(record["max_node_total_bandwidth"]) / elapsed,
}
elapsed = (max_step_id * STEP_DURATION_MS) / 1000.0
return {
"min": float(record["min_node_total_bandwidth"]) / elapsed,
"avg": float(record["avg_node_total_bandwidth"]) / elapsed,
"max": float(record["max_node_total_bandwidth"]) / elapsed,
}
raise Exception("No bandwidth data found in log file")
def topology_result(log_path: str) -> dict[str, int]:
for line in mixlog.get_input_stream(log_path):
if "diameter" in line:
return json.loads(line)
raise Exception("No bandwidth data found in log file")
for topic, json_msg in mixlog.get_input_stream(log_path):
if topic == "Topology":
return json_msg
raise Exception("No topology found in log file")
# Read the CSV data