291 lines
10 KiB
Python
Executable File
291 lines
10 KiB
Python
Executable File
# DO NOT EDIT THIS FILE!
|
|
#
|
|
# This file is generated from the CDP specification. If you need to make
|
|
# changes, edit the generator and regenerate all of the modules.
|
|
#
|
|
# CDP domain: Tracing (experimental)
|
|
from __future__ import annotations
|
|
from .util import event_class, T_JSON_DICT
|
|
from dataclasses import dataclass
|
|
import enum
|
|
import typing
|
|
from . import io
|
|
|
|
|
|
class MemoryDumpConfig(dict):
|
|
'''
|
|
Configuration for memory dump. Used only when "memory-infra" category is enabled.
|
|
'''
|
|
def to_json(self) -> dict:
|
|
return self
|
|
|
|
@classmethod
|
|
def from_json(cls, json: dict) -> MemoryDumpConfig:
|
|
return cls(json)
|
|
|
|
def __repr__(self):
|
|
return 'MemoryDumpConfig({})'.format(super().__repr__())
|
|
|
|
|
|
@dataclass
|
|
class TraceConfig:
|
|
#: Controls how the trace buffer stores data.
|
|
record_mode: typing.Optional[str] = None
|
|
|
|
#: Turns on JavaScript stack sampling.
|
|
enable_sampling: typing.Optional[bool] = None
|
|
|
|
#: Turns on system tracing.
|
|
enable_systrace: typing.Optional[bool] = None
|
|
|
|
#: Turns on argument filter.
|
|
enable_argument_filter: typing.Optional[bool] = None
|
|
|
|
#: Included category filters.
|
|
included_categories: typing.Optional[typing.List[str]] = None
|
|
|
|
#: Excluded category filters.
|
|
excluded_categories: typing.Optional[typing.List[str]] = None
|
|
|
|
#: Configuration to synthesize the delays in tracing.
|
|
synthetic_delays: typing.Optional[typing.List[str]] = None
|
|
|
|
#: Configuration for memory dump triggers. Used only when "memory-infra" category is enabled.
|
|
memory_dump_config: typing.Optional[MemoryDumpConfig] = None
|
|
|
|
def to_json(self):
|
|
json = dict()
|
|
if self.record_mode is not None:
|
|
json['recordMode'] = self.record_mode
|
|
if self.enable_sampling is not None:
|
|
json['enableSampling'] = self.enable_sampling
|
|
if self.enable_systrace is not None:
|
|
json['enableSystrace'] = self.enable_systrace
|
|
if self.enable_argument_filter is not None:
|
|
json['enableArgumentFilter'] = self.enable_argument_filter
|
|
if self.included_categories is not None:
|
|
json['includedCategories'] = [i for i in self.included_categories]
|
|
if self.excluded_categories is not None:
|
|
json['excludedCategories'] = [i for i in self.excluded_categories]
|
|
if self.synthetic_delays is not None:
|
|
json['syntheticDelays'] = [i for i in self.synthetic_delays]
|
|
if self.memory_dump_config is not None:
|
|
json['memoryDumpConfig'] = self.memory_dump_config.to_json()
|
|
return json
|
|
|
|
@classmethod
|
|
def from_json(cls, json):
|
|
return cls(
|
|
record_mode=str(json['recordMode']) if 'recordMode' in json else None,
|
|
enable_sampling=bool(json['enableSampling']) if 'enableSampling' in json else None,
|
|
enable_systrace=bool(json['enableSystrace']) if 'enableSystrace' in json else None,
|
|
enable_argument_filter=bool(json['enableArgumentFilter']) if 'enableArgumentFilter' in json else None,
|
|
included_categories=[str(i) for i in json['includedCategories']] if 'includedCategories' in json else None,
|
|
excluded_categories=[str(i) for i in json['excludedCategories']] if 'excludedCategories' in json else None,
|
|
synthetic_delays=[str(i) for i in json['syntheticDelays']] if 'syntheticDelays' in json else None,
|
|
memory_dump_config=MemoryDumpConfig.from_json(json['memoryDumpConfig']) if 'memoryDumpConfig' in json else None,
|
|
)
|
|
|
|
|
|
class StreamFormat(enum.Enum):
|
|
'''
|
|
Data format of a trace. Can be either the legacy JSON format or the
|
|
protocol buffer format. Note that the JSON format will be deprecated soon.
|
|
'''
|
|
JSON = "json"
|
|
PROTO = "proto"
|
|
|
|
def to_json(self):
|
|
return self.value
|
|
|
|
@classmethod
|
|
def from_json(cls, json):
|
|
return cls(json)
|
|
|
|
|
|
class StreamCompression(enum.Enum):
|
|
'''
|
|
Compression type to use for traces returned via streams.
|
|
'''
|
|
NONE = "none"
|
|
GZIP = "gzip"
|
|
|
|
def to_json(self):
|
|
return self.value
|
|
|
|
@classmethod
|
|
def from_json(cls, json):
|
|
return cls(json)
|
|
|
|
|
|
def end() -> typing.Generator[T_JSON_DICT,T_JSON_DICT,None]:
|
|
'''
|
|
Stop trace events collection.
|
|
'''
|
|
cmd_dict: T_JSON_DICT = {
|
|
'method': 'Tracing.end',
|
|
}
|
|
json = yield cmd_dict
|
|
|
|
|
|
def get_categories() -> typing.Generator[T_JSON_DICT,T_JSON_DICT,typing.List[str]]:
|
|
'''
|
|
Gets supported tracing categories.
|
|
|
|
:returns: A list of supported tracing categories.
|
|
'''
|
|
cmd_dict: T_JSON_DICT = {
|
|
'method': 'Tracing.getCategories',
|
|
}
|
|
json = yield cmd_dict
|
|
return [str(i) for i in json['categories']]
|
|
|
|
|
|
def record_clock_sync_marker(
|
|
sync_id: str
|
|
) -> typing.Generator[T_JSON_DICT,T_JSON_DICT,None]:
|
|
'''
|
|
Record a clock sync marker in the trace.
|
|
|
|
:param sync_id: The ID of this clock sync marker
|
|
'''
|
|
params: T_JSON_DICT = dict()
|
|
params['syncId'] = sync_id
|
|
cmd_dict: T_JSON_DICT = {
|
|
'method': 'Tracing.recordClockSyncMarker',
|
|
'params': params,
|
|
}
|
|
json = yield cmd_dict
|
|
|
|
|
|
def request_memory_dump(
|
|
deterministic: typing.Optional[bool] = None
|
|
) -> typing.Generator[T_JSON_DICT,T_JSON_DICT,typing.Tuple[str, bool]]:
|
|
'''
|
|
Request a global memory dump.
|
|
|
|
:param deterministic: *(Optional)* Enables more deterministic results by forcing garbage collection
|
|
:returns: A tuple with the following items:
|
|
|
|
0. **dumpGuid** - GUID of the resulting global memory dump.
|
|
1. **success** - True iff the global memory dump succeeded.
|
|
'''
|
|
params: T_JSON_DICT = dict()
|
|
if deterministic is not None:
|
|
params['deterministic'] = deterministic
|
|
cmd_dict: T_JSON_DICT = {
|
|
'method': 'Tracing.requestMemoryDump',
|
|
'params': params,
|
|
}
|
|
json = yield cmd_dict
|
|
return (
|
|
str(json['dumpGuid']),
|
|
bool(json['success'])
|
|
)
|
|
|
|
|
|
def start(
|
|
categories: typing.Optional[str] = None,
|
|
options: typing.Optional[str] = None,
|
|
buffer_usage_reporting_interval: typing.Optional[float] = None,
|
|
transfer_mode: typing.Optional[str] = None,
|
|
stream_format: typing.Optional[StreamFormat] = None,
|
|
stream_compression: typing.Optional[StreamCompression] = None,
|
|
trace_config: typing.Optional[TraceConfig] = None
|
|
) -> typing.Generator[T_JSON_DICT,T_JSON_DICT,None]:
|
|
'''
|
|
Start trace events collection.
|
|
|
|
:param categories: *(Optional)* Category/tag filter
|
|
:param options: *(Optional)* Tracing options
|
|
:param buffer_usage_reporting_interval: *(Optional)* If set, the agent will issue bufferUsage events at this interval, specified in milliseconds
|
|
:param transfer_mode: *(Optional)* Whether to report trace events as series of dataCollected events or to save trace to a stream (defaults to ```ReportEvents````).
|
|
:param stream_format: *(Optional)* Trace data format to use. This only applies when using ````ReturnAsStream```` transfer mode (defaults to ````json````).
|
|
:param stream_compression: *(Optional)* Compression format to use. This only applies when using ````ReturnAsStream```` transfer mode (defaults to ````none```)
|
|
:param trace_config: *(Optional)*
|
|
'''
|
|
params: T_JSON_DICT = dict()
|
|
if categories is not None:
|
|
params['categories'] = categories
|
|
if options is not None:
|
|
params['options'] = options
|
|
if buffer_usage_reporting_interval is not None:
|
|
params['bufferUsageReportingInterval'] = buffer_usage_reporting_interval
|
|
if transfer_mode is not None:
|
|
params['transferMode'] = transfer_mode
|
|
if stream_format is not None:
|
|
params['streamFormat'] = stream_format.to_json()
|
|
if stream_compression is not None:
|
|
params['streamCompression'] = stream_compression.to_json()
|
|
if trace_config is not None:
|
|
params['traceConfig'] = trace_config.to_json()
|
|
cmd_dict: T_JSON_DICT = {
|
|
'method': 'Tracing.start',
|
|
'params': params,
|
|
}
|
|
json = yield cmd_dict
|
|
|
|
|
|
@event_class('Tracing.bufferUsage')
|
|
@dataclass
|
|
class BufferUsage:
|
|
#: A number in range [0..1] that indicates the used size of event buffer as a fraction of its
|
|
#: total size.
|
|
percent_full: typing.Optional[float]
|
|
#: An approximate number of events in the trace log.
|
|
event_count: typing.Optional[float]
|
|
#: A number in range [0..1] that indicates the used size of event buffer as a fraction of its
|
|
#: total size.
|
|
value: typing.Optional[float]
|
|
|
|
@classmethod
|
|
def from_json(cls, json: T_JSON_DICT) -> BufferUsage:
|
|
return cls(
|
|
percent_full=float(json['percentFull']) if 'percentFull' in json else None,
|
|
event_count=float(json['eventCount']) if 'eventCount' in json else None,
|
|
value=float(json['value']) if 'value' in json else None
|
|
)
|
|
|
|
|
|
@event_class('Tracing.dataCollected')
|
|
@dataclass
|
|
class DataCollected:
|
|
'''
|
|
Contains an bucket of collected trace events. When tracing is stopped collected events will be
|
|
send as a sequence of dataCollected events followed by tracingComplete event.
|
|
'''
|
|
value: typing.List[dict]
|
|
|
|
@classmethod
|
|
def from_json(cls, json: T_JSON_DICT) -> DataCollected:
|
|
return cls(
|
|
value=[dict(i) for i in json['value']]
|
|
)
|
|
|
|
|
|
@event_class('Tracing.tracingComplete')
|
|
@dataclass
|
|
class TracingComplete:
|
|
'''
|
|
Signals that tracing is stopped and there is no trace buffers pending flush, all data were
|
|
delivered via dataCollected events.
|
|
'''
|
|
#: Indicates whether some trace data is known to have been lost, e.g. because the trace ring
|
|
#: buffer wrapped around.
|
|
data_loss_occurred: bool
|
|
#: A handle of the stream that holds resulting trace data.
|
|
stream: typing.Optional[io.StreamHandle]
|
|
#: Trace data format of returned stream.
|
|
trace_format: typing.Optional[StreamFormat]
|
|
#: Compression format of returned stream.
|
|
stream_compression: typing.Optional[StreamCompression]
|
|
|
|
@classmethod
|
|
def from_json(cls, json: T_JSON_DICT) -> TracingComplete:
|
|
return cls(
|
|
data_loss_occurred=bool(json['dataLossOccurred']),
|
|
stream=io.StreamHandle.from_json(json['stream']) if 'stream' in json else None,
|
|
trace_format=StreamFormat.from_json(json['traceFormat']) if 'traceFormat' in json else None,
|
|
stream_compression=StreamCompression.from_json(json['streamCompression']) if 'streamCompression' in json else None
|
|
)
|