mirror of
https://github.com/python/cpython.git
synced 2025-10-25 18:54:53 +00:00
33 lines
1,006 B
Python
33 lines
1,006 B
Python
from abc import ABC, abstractmethod
|
|
|
|
# Enums are slow
|
|
THREAD_STATE_RUNNING = 0
|
|
THREAD_STATE_IDLE = 1
|
|
THREAD_STATE_GIL_WAIT = 2
|
|
THREAD_STATE_UNKNOWN = 3
|
|
|
|
STATUS = {
|
|
THREAD_STATE_RUNNING: "running",
|
|
THREAD_STATE_IDLE: "idle",
|
|
THREAD_STATE_GIL_WAIT: "gil_wait",
|
|
THREAD_STATE_UNKNOWN: "unknown",
|
|
}
|
|
|
|
class Collector(ABC):
|
|
@abstractmethod
|
|
def collect(self, stack_frames):
|
|
"""Collect profiling data from stack frames."""
|
|
|
|
@abstractmethod
|
|
def export(self, filename):
|
|
"""Export collected data to a file."""
|
|
|
|
def _iter_all_frames(self, stack_frames, skip_idle=False):
|
|
"""Iterate over all frame stacks from all interpreters and threads."""
|
|
for interpreter_info in stack_frames:
|
|
for thread_info in interpreter_info.threads:
|
|
if skip_idle and thread_info.status != THREAD_STATE_RUNNING:
|
|
continue
|
|
frames = thread_info.frame_info
|
|
if frames:
|
|
yield frames
|