yadunut/clan-core
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

clan_vm_manager: Add CUMTIME to profiler output

Browse Source
This commit is contained in:
Qubasa
2024-03-04 16:00:04 +07:00
parent b77ffac4d4
commit cde72f3710
3 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pkgs/clan-vm-manager/clan_vm_manager/components/profiler.py
View File

@@ -30,10 +30,10 @@ cProfile Output Columns Explanation:
"""
def print_profile(profiler: cProfile.Profile) -> None:
def print_profile(profiler: cProfile.Profile, sortkey: pstats.SortKey) -> None:
s = io.StringIO()
ps = pstats.Stats(profiler, stream=s)
ps.sort_stats(pstats.SortKey.CUMULATIVE)
ps.sort_stats(sortkey)
ps.print_stats(12)
# Process the output to trim file paths
@@ -60,6 +60,7 @@ def print_profile(profiler: cProfile.Profile) -> None:
print(new_line)
# TODO: Add an RLock for every profiler, currently not thread safe
class ProfilerStore:
profilers: dict[str, cProfile.Profile]
@@ -76,7 +77,8 @@ class ProfilerStore:
def on_exit(self) -> None:
for key, profiler in self.profilers.items():
print("=" * 7 + key + "=" * 7)
print_profile(profiler)
print_profile(profiler, pstats.SortKey.TIME)
print_profile(profiler, pstats.SortKey.CUMULATIVE)
print(explanation)