o i9 @slddlmZddlZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl mZdd l mZdd l mZdd l mZdd l mZdd l mZddl mZddl mZddl mZddl mZddl mZddl mZddl mZddl mZddl mZddl mZddl mZddl mZddl m Z ddl m!Z!ddl m"Z"ddl m#Z#ddl m$Z$ddl m%Z%dd l m&Z&dd!l m'Z'dd"l m(Z(dd#l m)Z)dd$l*m+Z+dd%l,m-Z-dd&l.m/Z/dd'l0m1Z1dd(l2m3Z3dd)l2m4Z4e1e5Z6d*e7e8efd+e7e8e8fd,e7e8effd-d.Z9d/e7d,e&fd0d1Z:d2ed,e;fd3d4ZZ?Gd;d<dd>e?ZAGd?d@d@e?ZBGdAdBdBe?ZCGdCdDdDe?ZDGdEdFdFeEZFejGe@ejHeAejIeBejJeCejKeDiZLdGe7e8efd,efdHdIZMe=dJe7dKe-d,eefdLdMZNGdNdOdOeOeZPGdPdQdQe4ZQdS)R)EnumN)Any)Callable)Iterable)Optional)config)DEFAULT_CAPTURE_LIMITS)"DEFAULT_PROBE_CONDITION_ERROR_RATE)DEFAULT_PROBE_RATE)DEFAULT_SNAPSHOT_PROBE_RATE)DEFAULT_TRIGGER_PROBE_RATECaptureExpression) CaptureLimits)ExpressionTemplateSegment) FunctionProbe) LineProbe)LiteralTemplateSegment)LogFunctionProbe) LogLineProbe)MetricFunctionProbe)MetricLineProbe)Probe)ProbeEvalTiming) ProbeType)SpanDecoration)SpanDecorationFunctionProbe)SpanDecorationLineProbe)SpanDecorationTag)SpanFunctionProbe)StringTemplate)TemplateSegment) TimingMixin)TriggerFunctionProbe)TriggerLineProbe) ProbeRegistry)ProbeStatusLogger)DDRedactedExpression) get_logger)Payload) RCCallbackdmappingreturncsfdd|DS)Ncsi|] \}}|||qS)get).0kvr,r.Y/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/debugging/_probe/remoteconfig.py 3szxlate_keys..)items)r+r,r.r3r4 xlate_keys2sr7segmentcCs@d|vr t|ddSd|vrtt|dSd|}t|)Nstr) str_valuejson)exprzInvalid template segment: )rrr'compile ValueError)r8msgr.r.r4_compile_segment6s  r@probecCs@|jdd}|jdd}|dus|tjko|dup|tjkS)Nversionenv)tagsr/ tracer_configrBrC)rA probe_version probe_envr.r.r4_match_env_and_versionAs rHf.cs"dtdtdttffdd }|S)Nargskwargsr-csdd|i|DS)NcSsg|]}t|r|qSr.)rHr0_r.r.r4 Lsz@_filter_by_env_and_version.._wrapper..r.)rJrKrIr.r4_wrapperKsz,_filter_by_env_and_version.._wrapper)rrr)rIrPr.rOr4_filter_by_env_and_versionJsrQc@seZdZUdZeeeed<dZeee ed<e de e e fde e e fddfddZe de e e fde e e fde fd d ZdS) ProbeFactoryN__line_class____function_class__rJattribsr-cCst)N)NotImplementedErrorclsrJrUr.r.r4 update_argsUszProbeFactory.update_argscCs||||d}|dddur3|jdurtd|d|d<t|dd|d<|jdi|S|jdureZdZeZeZedee e fdee e fddfddZ dS)LogProbeFactoryrJrUr-Nc Cs|dd}|r tnt}|d}|dur|d|}|jd|vr(t|dnd|d|vr?tdit|dddd d d ntt |d d |dgD|ddd |dgDddS)NcaptureSnapshotFsamplingsnapshotsPerSecondwhencapture max_levelmax_sizemax_len max_fields)maxReferenceDepthmaxCollectionSize maxLength maxFieldCountcSsg|] }tdi|qS)r.r rLr.r.r4rNsz/LogProbeFactory.update_args..captureExpressionstemplatecSg|]}t|qSr.r@r0r8r.r.r4rNsegments) conditionratelimitscondition_error_rate take_snapshotcapture_expressionsrrr.) r/r r updater'r=rr7rr )rXrJrUrrrur.r.r4rYvs4       zLogProbeFactory.update_args) rlrmrnrrSrrTrprqr9rrYr.r.r.r4rsr ,rsc@rr)MetricProbeFactoryrJrUr-Nc Cs\|d|dd<|jd|vrt|dnd|d|dtd|vr(t|dndddS) Nprobe_idrDzdebugger.probeidrw metricNamekindvalue)rnamerrrrr'r=r rWr.r.r4rYs zMetricProbeFactory.update_args) rlrmrnrrSrrTrprqr9rrYr.r.r.r4rrrc@s:eZdZeZedeeefdeeefddfddZ dS)SpanProbeFactoryrJrUr-NcCs(|jd|vr t|dndtddS)Nrw)rrrrWr.r.r4rYs zSpanProbeFactory.update_args) rlrmrnrrTrprqr9rrYr.r.r.r4rs,rc@rr)SpanDecorationProbeFactoryrJrUr-NcCs$|j|ddd|dDddS)N targetSpanc Ss@g|]}td|vrt|dnddd|dgDdqS)rwNc SsDg|]}t|dt|dddd|ddgDddqS)rrrcSrr.rrr.r.r4rNrzPSpanDecorationProbeFactory.update_args....r)rr)rr)rr r/)r0tr.r.r4rNs zESpanDecorationProbeFactory.update_args...rD)rwrD)rr'r=r/)r0r+r.r.r4rNs z:SpanDecorationProbeFactory.update_args.. decorations) target_spanr)rrWr.r.r4rYs  z&SpanDecorationProbeFactory.update_args) rlrmrnrrSrrTrprqr9rrYr.r.r.r4rrrc@rr)TriggerProbeFactoryrJrUr-NcCsJ|j|didt|dt|dd|vrt|dndtddS)NrucooldownInSeconds session_idlevelrw)rrrrr)rr/r rir'r=r rWr.r.r4rYs  zTriggerProbeFactory.update_args) rlrmrnr$rSr#rTrprqr9rrYr.r.r.r4rrrc@s eZdZdS)InvalidProbeConfigurationN)rlrmrnr.r.r.r4rsrrUc Csz |d}|d}Wnty}ztd|d}~wwt||ddtdd|d gDd }t|}|durB|||Std |) z& Create a new Probe instance. r_idzInvalid probe attributes: %sNrBrcss|] }|ddVqdS):N)splitrLr.r.r4 szbuild_probe..rDrrBrDzUnsupported probe type: %s)KeyErrorrrqr/ PROBE_FACTORYrk)rU_type_iderJfactoryr.r.r4 build_probes       rr status_loggerc Csvzt|gWStyty:}z!|jt|d|ddidt|jt|fdgWYd}~Sd}~ww)NrrBrr)rAerror) rr Exceptionrrr/r_rlr9)rrrr.r.r4 get_probess  rc@seZdZdZdZdZdZdS)ProbePollerEventrrN)rlrmrn NEW_PROBESDELETED_PROBESMODIFIED_PROBES STATUS_UPDATEr.r.r.r4rs rc @seZdZdZdeeeegdfdeddde ddf d d Z d e e efd e e efddfd dZ de deddfddZdddZdddZdejeddfddZdS)DebuggerRCCallbackzRemote config callback for dynamic instrumentation probes. The periodic() method emits probe status at every polling operation (with time-based throttling). The __call__() method processes dynamic instrumentation probe payloads when present. callbackNrregistryr%diagnostics_intervalr-cCs(||_||_||_||_d|_i|_dS)aEInitialize the debugger callback. Args: callback: Function to call when probe events occur status_logger: Logger for probe status messages registry: Probe registry for emitting status messages diagnostics_interval: Interval in seconds between status emissions gN) _callback_status_logger _registry_diagnostics_interval_last_status_emission_configs)selfrrrrr.r.r4__init__s  zDebuggerRCCallback.__init__ prev_probes next_probescsfddD}fddD}fddD}|r*|tj||r3|tj||r>|tj|dSdS)z6Dispatch events for new, deleted, and modified probes.cg|] \}}|vr|qSr.r.r0rMprr.r4rN6z=DebuggerRCCallback._dispatch_probe_events..crr.r.r)rr.r4rN7rcs(g|]\}}|vr||kr|qSr.r.rrr.r4rN8s(N)r6rrrrr)rrr new_probesdeleted_probesmodified_probesr.)rrr4_dispatch_probe_events4sz)DebuggerRCCallback._dispatch_probe_events config_idrcCsv|j|i}|dvrddt||jDni}tdtt| |||r2||j|<dS|j |ddS)z+Update probes for a specific configuration.)NFcSsi|]}|j|qSr.)r)r0rAr.r.r4r5Erz@DebuggerRCCallback._update_probes_for_config..z:[%s][P: %s] Dynamic Instrumentation, dispatch probe eventsN) rr/rrlogdebugosgetpidgetppidrpop)rrrrrr.r.r4_update_probes_for_configAs z,DebuggerRCCallback._update_probes_for_configcCs*|jD]}||iq|jdS)zDelete all registered probes.N)rvaluesrclear)rrr.r.r4delete_all_probesQsz$DebuggerRCCallback.delete_all_probescCsLt}||j|jkrtdtt|j ||_|j dS)zPeriodic method called at every polling operation. Checks if it's time to emit probe status messages based on the diagnostics interval. zG[%s][P: %s] Dynamic Instrumentation: emitting probe status log messagesN) timerrrrrrrrlog_probes_statusrflush)r current_timer.r.r4periodicWs zDebuggerRCCallback.periodicpayloadscCsbtdtt|D]}|jdur tdttq ||jj|jq |j dS)zProcess dynamic instrumentation probe configuration payloads. Args: payloads: Sequence of configuration payloads to process z+[%s][P: %s] Dynamic Instrumentation UpdatedNzP[%s][P: %s] Dynamic Instrumentation: no RCM metadata for configuration; skipping) rrrrrmetadatarrcontentrr)rrpayloadr.r.r4__call__is zDebuggerRCCallback.__call__)r-N)rlrmrn__doc__rrlistrr&floatrrqr9rrrrrrSequencer)rr.r.r.r4rs$ &  r)RenumrrrtypingrrrrrddtracerrEddtrace.debugging._probe.modelrr r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$!ddtrace.debugging._probe.registryr%ddtrace.debugging._probe.statusr&ddtrace.debugging._redactionr'ddtrace.internal.loggerr(ddtrace.internal.remoteconfigr)r*rlrrqr9r7r@boolrHrQobjectrRrsrrrrr>r LOG_PROBE METRIC_PROBE SPAN_PROBESPAN_DECORATION_PROBE TRIGGER_PROBErrrrirrr.r.r.r4s                                        . * !%