o ҷh<)@s"ddlmZddlmZmZddlmZddlZddlZddl m Z Gddde Z eGd d d Z Gd d d ejZGd ddejZd6ddZd7ddZd8ddZ d9d:ddZ       d;dCz'TraceConfig.to_json..includedCategoriescSr1r r r2r r rr5Er6excludedCategoriescSr1r r r2r r rr5Gr6syntheticDelaysmemoryDumpConfig) r r$r&r'r(r)r*r+r,r)rrr r rr8s$            zTraceConfig.to_jsonc Cs|d|vr t|dndd|vrt|dndd|vr!t|dndd|vr,t|dndd|vr:dd|dDndd|vrHd d|dDndd |vrVd d|d Dndd |vrdt|d d Sdd S)Nr-r.r/r0r7cSg|]}t|qSr strr2r r rr5Sz)TraceConfig.from_json..r8cSr;r r<r2r r rr5Tr>r9cSr;r r<r2r r rr5Ur>r:)r$r&r'r(r)r*r+r,)r=boolrrrr r rrLszTraceConfig.from_json)rrrr$__annotations__r&r'r(r)r*r+r,rr rr r r rr"s        r"c@,eZdZdZdZdZddZeddZdS) StreamFormatz 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. rprotocC|jSr valuer r r rrbzStreamFormat.to_jsoncCrr r rr r rrerzStreamFormat.from_jsonN) rrrrJSONPROTOrr rr r r rrBZsrBc@rA) StreamCompressionzB Compression type to use for traces returned via streams. nonegzipcCrDr rEr r r rrqrGzStreamCompression.to_jsoncCrr r rr r rrtrzStreamCompression.from_jsonN) rrrrNONEGZIPrr rr r r rrJjsrJr 0typing.Generator[T_JSON_DICT, T_JSON_DICT, None]ccsddi}|V}dS)z' Stop trace events collection. methodz Tracing.endNr cmd_dictrr r rendys rSz"get_categories.. categoriesr rQr r rget_categoriess rVsync_idr=ccs$t}||d<d|d}|V}dS)zh Record a clock sync marker in the trace. :param sync_id: The ID of this clock sync marker syncIdzTracing.recordClockSyncMarkerrPparamsNr )rWrZrRrr r rrecord_clock_sync_markers r\ deterministicr%Ctyping.Generator[T_JSON_DICT, T_JSON_DICT, typing.Tuple[str, bool]]ccs@t}|dur ||d<d|d}|V}t|dt|dfS)aI 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. Nr]zTracing.requestMemoryDumprYdumpGuidsuccess)r r=r?)r]rZrRrr r rrequest_memory_dumps   rarUr#optionsbuffer_usage_reporting_intervaltyping.Optional[float] transfer_mode stream_formattyping.Optional[StreamFormat]stream_compression"typing.Optional[StreamCompression] trace_configtyping.Optional[TraceConfig]c cst}|dur ||d<|dur||d<|dur||d<|dur$||d<|dur.||d<|dur8||d<|durB||d<d |d }|V} dS) a7 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)* NrUrbbufferUsageReportingInterval transferMode streamFormatstreamCompression traceConfigz Tracing.startrY)r r) rUrbrcrerfrhrjrZrRrr r rstarts(    rqzTracing.bufferUsagec@s4eZdZUded<ded<ded<ed dd Zd S) BufferUsagerd percent_full event_countrFrrr cCsN|d|vr t|dndd|vrt|dndd|vr#t|ddSddS)N percentFull eventCountrF)rsrtrF)floatrr r rrszBufferUsage.from_jsonN)rrr rr)rrrr@r rr r r rrrs rrzTracing.dataCollectedc@s(eZdZUdZded<ed ddZd S) DataCollectedz 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. ztyping.List[dict]rFrrr cCs|dd|dDdS)NcSr;r r[r2r r rr5r>z+DataCollected.from_json..rFrEr rr r rrszDataCollected.from_jsonN)rrr rxrrrrr@r rr r r rrxs rxzTracing.tracingCompletec@s@eZdZUdZded<ded<ded<ded <edd dZdS)TracingCompletez Signals that tracing is stopped and there is no trace buffers pending flush, all data were delivered via dataCollected events. r?data_loss_occurredz typing.Optional[io.StreamHandle]streamrg trace_formatrirhrrr cCs`|t|dd|vrtj|dndd|vrt|dndd|vr,t|ddSddS)NdataLossOccurredr| traceFormatro)r{r|r}rh)r?r StreamHandlerrBrJrr r rrs zTracingComplete.from_jsonN)rrr rzryr r r rrz s rz)r rO)r rT)rWr=r rOr )r]r%r r^)NNNNNNN)rUr#rbr#rcrdrer#rfrgrhrirjrkr rO) __future__rutilrr dataclassesrenumtypingrr rr"EnumrBrJrSrVr\rarqrrrxrzr r r rsB   ;    *