CuteEngineering/cutekit
2
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Add support for heap profiling

Browse source
This commit is contained in:
Sleepy Monax 2024-01-22 14:32:02 +01:00
parent 3019b1807f
commit c4ae611ec3
1 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
cutekit/shell.py
View file

@ -13,7 +13,7 @@ import tempfile
import dataclasses as dt import dataclasses as dt
from pathlib import Path from pathlib import Path
from typing import Optional from typing import Literal, Optional
from . import const, cli from . import const, cli
_logger = logging.getLogger(__name__) _logger = logging.getLogger(__name__)
@ -207,7 +207,7 @@ def debug(cmd: list[str], debugger: str = "lldb", wait: bool = False):
raise RuntimeError(f"Unknown debugger {debugger}") raise RuntimeError(f"Unknown debugger {debugger}")
def profile(cmd: list[str], rate=1000): def _profileCpu(cmd: list[str], rate=1000):
mkdir(const.TMP_DIR) mkdir(const.TMP_DIR)
perfFile = f"{const.TMP_DIR}/perf.data" perfFile = f"{const.TMP_DIR}/perf.data"
try: try:
@ -240,6 +240,19 @@ def profile(cmd: list[str], rate=1000):
rmrf(perfFile) rmrf(perfFile)
def _profileMem(cmd: list[str]):
exec("heaptrack", *cmd)
def profile(cmd: list[str], rate=1000, what: Literal["cpu", "mem"] = "cpu"):
if what == "cpu":
_profileCpu(cmd, rate)
elif what == "mem":
_profileMem(cmd)
else:
raise RuntimeError(f"Unknown profile type {what}")
def readdir(path: str) -> list[str]: def readdir(path: str) -> list[str]:
_logger.debug(f"Reading directory {path}") _logger.debug(f"Reading directory {path}")
@ -406,7 +419,7 @@ def _(args: cli.Args):
wait = args.consumeOpt("wait", False) is True wait = args.consumeOpt("wait", False) is True
debugger = args.consumeOpt("debugger", "lldb") debugger = args.consumeOpt("debugger", "lldb")
command = [str(args.consumeArg()), *args.extra] command = [str(args.consumeArg()), *args.extra]
debug(command, debugger=debugger, wait=wait) debug(command, debugger=str(debugger), wait=wait)
@cli.command("p", "profile", "Profile a program") @cli.command("p", "profile", "Profile a program")