Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add script to merge the JSON files produced by the FastTimerService #44836

Merged
merged 1 commit into from
Apr 24, 2024
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
57 changes: 57 additions & 0 deletions HLTrigger/Timer/scripts/mergeResourcesJson.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,57 @@
#! /usr/bin/env python3

import sys
import json

usage = """Usage: mergeResourceJson.py FILE [FILE ...]

Merge the content of multiple "resources.json" files produced by the FastTimerService,
and print the result to standard output.

Example:
mergeResourceJson.py step*/pid*/resources.json > resources.json
"""

def merge_into(metrics, data, dest):
dest["events"] += data["events"]
for metric in metrics:
dest[metric] += data[metric]


if len(sys.argv) == 1:
print(usage)
sys.exit(1)

if '-h' in sys.argv[1:] or '--help' in sys.argv[1:]:
print(usage)
sys.exit(0)

with open(sys.argv[1]) as f:
output = json.load(f)

metrics = [ label for resource in output["resources"] for label in resource ]

datamap = { module["type"] + '|' + module["label"] : module for module in output["modules"] }

for arg in sys.argv[2:]:
with open(arg) as f:
input = json.load(f)

if output["resources"] != input["resources"]:
print("Error: input files describe different metrics")
sys.exit(1)

if output["total"]["label"] != input["total"]["label"]:
print("Warning: input files describe different process names")
merge_into(metrics, input["total"], output["total"])

for module in input["modules"]:
key = module["type"] + '|' + module["label"]
if key in datamap:
merge_into(metrics, module, datamap[key])
else:
datamap[key] = module
output["modules"].append(datamap[key])

json.dump(output, sys.stdout, indent = 2 )
sys.stdout.write('\n')