top objects api

This commit is contained in:
Victor Shyba 2020-03-10 21:39:42 -03:00
parent 56c8ad1221
commit 8169bf6b97
2 changed files with 56 additions and 2 deletions

View file

@ -1,3 +1,4 @@
import linecache
import os
import re
import asyncio
@ -4541,10 +4542,10 @@ class Daemon(metaclass=JSONRPCServerType):
Enable/disable tracemalloc memory tracing
Usage:
jsonrpc_tracemalloc_set (<enable>)
jsonrpc_tracemalloc_set (<enable> | --enable=<enable>)
Options:
--enable : (bool) True enables, False disables
--enable=<enable> : (bool) True enables, False disables
Returns:
(bool) is it tracing?
@ -4555,6 +4556,49 @@ class Daemon(metaclass=JSONRPCServerType):
tracemalloc.stop()
return tracemalloc.is_tracing()
def jsonrpc_tracemalloc_top(self, items: int = 10):
"""
Show most common objects, the place that created them and their size.
Usage:
jsonrpc_tracemalloc_top [(<items> | --items=<items>)]
Options:
--items=<items> : (int) maximum items to return, from the most common
Returns:
(dict) dictionary containing most common objects in memory
{
"line": (str) filename and line number where it was created,
"code": (str) code that created it,
"size": (int) size in bytes, for each "memory block",
"count" (int) number of memory blocks
}
"""
if not tracemalloc.is_tracing():
raise Exception("Enable tracemalloc first! See 'tracemalloc set' command.")
stats = tracemalloc.take_snapshot().filter_traces((
tracemalloc.Filter(False, "<frozen importlib._bootstrap>"),
tracemalloc.Filter(False, "<unknown>"),
# tracemalloc and linecache here use some memory, but thats not relevant
tracemalloc.Filter(False, tracemalloc.__file__),
tracemalloc.Filter(False, linecache.__file__),
)).statistics('lineno', True)
results = []
for stat in stats:
frame = stat.traceback[0]
filename = os.sep.join(frame.filename.split(os.sep)[-2:])
line = linecache.getline(frame.filename, frame.lineno).strip()
results.append({
"line": f"{filename}:{frame.lineno}",
"code": line,
"size": stat.size,
"count": stat.count
})
if len(results) == items:
break
return results
COMMENT_DOC = """
View, create and abandon comments.
"""

View file

@ -44,3 +44,13 @@ class TroubleshootingCommands(CommandTestCase):
async def test_tracemalloc_commands(self):
self.assertFalse(self.daemon.jsonrpc_tracemalloc_set(False))
self.assertTrue(self.daemon.jsonrpc_tracemalloc_set(True))
class WeirdObject():
pass
hold_em = [WeirdObject() for _ in range(500)]
self.assertEqual(
[{'code': 'hold_em = [WeirdObject() for _ in range(500)]',
'count': 502,
'line': 'other/test_other_commands.py:50',
'size': 36656}], self.daemon.jsonrpc_tracemalloc_top(items=1)
)