1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
import glob
import json
import os
from pathlib import Path
from typing import Any, Dict
REPO_ROOT = Path(__file__).resolve().parent.parent.parent
def flatten_data(d: Dict[str, Any]) -> Dict[str, Any]:
# Flatten the sccache stats data from a possibly nested dictionary to a flat
# dictionary. For example, the input:
# {
# "cache": {
# "hit": 1,
# "miss": 2,
# },
# }
# will be transformed to:
# {
# "cache_hit": 1,
# "cache_miss": 2,
# }
flat_data = {}
for key, value in d.items():
if isinstance(value, dict):
for k, v in flatten_data(value).items():
flat_data[f"{key}_{k}"] = v
else:
flat_data[key] = value
return flat_data
def main() -> None:
records = []
for file in glob.glob(str(REPO_ROOT / "sccache-stats-*.json")):
with open(file) as f:
data = json.load(f)
# I don't know what sccache info will be most useful yet, and the
# sccache json has a decent number of keys, so just flatten the data
# and store all of it
records.append(
{
"benchmark": {
"name": "sccache_stats",
},
"metric": {
"name": "sccache_stats",
"benchmark_values": [int(os.environ.get("BUILD_TIME") or 0)],
"extra_info": flatten_data(data),
},
}
)
output_file = REPO_ROOT / "test" / "test-reports" / "sccache-stats.json"
os.makedirs(output_file.parent, exist_ok=True)
with open(output_file, "w") as f:
json.dump(records, f)
if __name__ == "__main__":
main()
|
