diff --git a/simlib/blendnet-sims/scripts/monitor.py b/simlib/blendnet-sims/scripts/monitor.py new file mode 100644 index 0000000..a9b6341 --- /dev/null +++ b/simlib/blendnet-sims/scripts/monitor.py @@ -0,0 +1,48 @@ +import argparse +import json +from collections.abc import Iterable +from typing import Any + +import pandas as pd + +import mixlog + + +def analyze_monitors(input_stream: Iterable[str]) -> None: + df = pd.DataFrame(monitor_records(input_stream)) + + result = { + "min": df["min"].min(), + "avg": (df["num_conns"] * df["avg"]).sum() / df["num_conns"].sum(), + "max": df["max"].max(), + } + print(result) + + +def monitor_records(input_stream: Iterable[str]) -> list[Any]: + records = [] + + for line in input_stream: + try: + record = json.loads(line) + except json.JSONDecodeError: + continue + + if "message_type" in record and "num_conns" in record: + records.append(record) + + return records + + +if __name__ == "__main__": + parser = argparse.ArgumentParser(description="Mix connection monitor analysis") + parser.add_argument( + "--log-path", + nargs="?", + type=str, + help="An input log file path. If not provided, input will be read from stdin.", + ) + args = parser.parse_args() + + input = mixlog.get_input_stream(args.log_path) + analyze_monitors(input)