luau/tools/heapstat.py
2022-08-18 14:04:33 -07:00

64 lines
2.1 KiB
Python

#!/usr/bin/python
# This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
# Given a heap snapshot, this tool gathers basic statistics about the allocated objects
# To generate a snapshot, use luaC_dump, ideally preceded by luaC_fullgc
import json
import sys
from collections import defaultdict
def updatesize(d, k, s):
oc, os = d.get(k, (0, 0))
d[k] = (oc + 1, os + s)
def sortedsize(p):
return sorted(p, key = lambda s: s[1][1], reverse = True)
def getkey(heap, obj, key):
pairs = obj.get("pairs", [])
for i in range(0, len(pairs), 2):
if pairs[i] and heap[pairs[i]]["type"] == "string" and heap[pairs[i]]["data"] == key:
if pairs[i + 1] and heap[pairs[i + 1]]["type"] == "string":
return heap[pairs[i + 1]]["data"]
else:
return None
return None
with open(sys.argv[1]) as f:
dump = json.load(f)
heap = dump["objects"]
size_type = {}
size_udata = {}
size_category = {}
for addr, obj in heap.items():
updatesize(size_type, obj["type"], obj["size"])
if obj.get("cat") != None:
updatesize(size_category, str(obj["cat"]), obj["size"])
if obj["type"] == "userdata" and "metatable" in obj:
metatable = heap[obj["metatable"]]
typemt = getkey(heap, metatable, "__type") or "unknown"
updatesize(size_udata, typemt, obj["size"])
print("objects by type:")
for type, (count, size) in sortedsize(size_type.items()):
print(type.ljust(10), str(size).rjust(8), "bytes", str(count).rjust(5), "objects")
print()
print("userdata by __type:")
for type, (count, size) in sortedsize(size_udata.items()):
print(type.ljust(20), str(size).rjust(8), "bytes", str(count).rjust(5), "objects")
if len(size_category) != 0:
print()
print("objects by category:")
for type, (count, size) in sortedsize(size_category.items()):
cat = dump["stats"]["categories"][type]
name = cat["name"] if "name" in cat else str(type)
print(name.ljust(30), str(size).rjust(8), "bytes", str(count).rjust(5), "objects")