o i@sddlmZddlZddlZddlmZddlmZddlZddlZddlm Z ddl m Z ddl m Z ddl mZdd l mZdd l mZdd l mZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z!ddl"m#Z#ddl"m$Z$ddl"m%Z%ddl"m&Z&ddl'm(Z(ddl'm)Z)ddl'm*Z*ddl'm+Z+ddl'm,Z,ddl-m.Z.dd l-m/Z/dd!l-m0Z0dd"l-m1Z1dd#l-m2Z2dd$l-m3Z3dd%l-m4Z4dd&l-m5Z5dd'l-m6Z6dd(l7m8Z8dd)l9m:Z:dd*l;mm?Z?dd-l@mAZAdd.lBmCZCdd/lDmEZEdd0lFmGZGdd0lHmGZIdd1lJmKZKdd2lLmMZMdd3lNmOZOdd4lPmQZQdd5lRmSZSdd6lRmTZTdd7lUmVZVdd8lWmXZXe:eYZZed9e d:Z[d;eAdZ]Gd?d@d@e^Z_dS)A)contextmanagerN)iscoroutinefunction)chain)getpid)Lock)Callable)Optional)TypeVar)Union)cast)Context)SpanAggregator) SpanProcessor)TopLevelSpanProcessor)TraceProcessor)ResourceRenamingProcessor)BaseContextProvider)DefaultContextProvider)Span)APPSEC) _HOSTNAME_KEY)ENV_KEY)PID) VERSION_KEY)atexit)core)debug)forksafe)hostname) LOG_ATTR_ENV)LOG_ATTR_SERVICE)LOG_ATTR_SPAN_ID)LOG_ATTR_TRACE_ID)LOG_ATTR_VALUE_EMPTY)LOG_ATTR_VALUE_ZERO)LOG_ATTR_VERSION)SAMPLING_DECISION_TRACE_TAG_KEY)SPAN_API_DATADOG) get_hostname) get_logger)PyTracerMetadata)store_metadata)PeerServiceProcessor)EndpointCallCounterProcessor)get_runtime_id)BaseServiceProcessor)config) _ps_config)_get_metas_to_propagate)DDTraceDeprecationWarning)format_trace_id)AgentWriterInterface) HTTPWriter) deprecate) __version__ AnyCallable)boundprofiling_span_processorreturncCsTg}|tg7}tjr|ttjr#tjs#ddlm}|||||S)z5Construct the default list of span processors to use.r)SpanStatsProcessorV06) rr0 _trace_resource_renaming_enabledappendr_trace_compute_stats_trace_writer_native ddtrace.internal.processor.statsr=)r;span_processorsr=rDI/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/_trace/tracer.py _default_span_processors_factoryFs     rFc@seZdZdZdZdZd]ddZedee fddZ e j d e fd dZ d]d d Z d dZ d]ddZedefddZeddZej ddZeddZej ddZeddZdeefddZd^deeeefdee e ffdd Z    !   d_d"eed#eed$eed%ee d&eed'eed(eeeddfd)d*Zd+d,Zd-d.Z     /d`d(eeed#eed0eed$eed1eddf d2d3Z!ddddd4e"fd5e d6eeeefd7ee d8ee d9ee d:ed;e defdd?Z%d@eddfdAdBZ&dCdDZ'ddde"fd5e d7ee d8ee d9ee d;e def dEdFZ(deefdGdHZ)deefdIdJZ*edee fdKdLZ+dMdNZ,   dadOe-dPe d7ee d8ee d9ee de-f dQdRZ.   dadOe-dPe d7ee d8ee d9ee de-f dSdTZ/    dbd5ee d7ee d8ee d9ee de0e-ge-ff dUdVZ1dWee e fddfdXdYZ2d^dZee3ddfd[d\Z4dS)cTracera_ Tracer is used to create, sample and submit spans that measure the execution time of sections of code. If you're running an application that will serve a single trace per thread, you can use the global tracer instance:: from ddtrace.trace import tracer trace = tracer.trace('app.request', 'web-server').finish() Nr<c CsNt|turtjdur|t_ntdtj|_d|_ t |_ tj |_ t|_tjr2d|_ dt_t|_t|j|_ttjtjtttgd|_t|j t!"|j#t!$|j#t"|j%t&|_'d|_(t)t*t+t,tj-ptdtj.pxdtj/p|dddd}zt0||_1WdSt2y}ztj3dt4|ddWYd}~dSd}~ww) z Create a new ``Tracer`` instance. A global tracer is already initialized for common usage, so there is no need to initialize your own ``Tracer``. Nz^Initializing multiple Tracer instances is not supported. Use ``ddtrace.trace.tracer`` instead.F)partial_flush_enabledpartial_flush_min_spans dd_processors) runtime_idtracer_versionr service_name service_envservice_version process_tags container_idz)Failed to store the configuration on disk)error)extra)5typerG _instancelogrSr0tagscopy_tags_wrap_executorr_pid_tracing_enabledenabledrcontext_provider asm_config _apm_opt_outr@r-%_endpoint_call_counter_span_processorrF_span_processorsr _partial_flush_enabled_partial_flush_min_spansr,r1r/_span_aggregatorrregister_before_fork_sample_before_forkrregister_atexitregister_on_exit_signal_child_after_forkr_shutdown_lock _new_processr*r.r8r(serviceenvversionr+_config_on_disk Exceptionrdict)selfmetadataerDrDrE__init__nsV          "zTracer.__init__cCs t|jjttfr|jjjSdSN isinstancerfwriterr6r5 intake_urlrurDrDrE _agent_urls zTracer._agent_urlvaluecCs$t|jjttfr||jj_dSdSryrzrurrDrDrErscCs4tjdkrdnd}td|j||j|jddS)Nntz ctrl-breakzctrl-cz8Waiting %d seconds for tracer to finish. Hit %s to quit.)timeout)osnamerWrSHUTDOWN_TIMEOUTshutdown)rukeyrDrDrErjszTracer._atexitcCs|j|dSry)_samplersampleruspanrDrDrErsz Tracer.samplecCs2|}|dur|jjdur||dSdSdSry)current_root_spancontextsampling_priorityrrrDrDrErhszTracer._sample_before_forkrc csR|j}d|_|j|zdVWd|_|j|dSd|_|j|w)NTF)r_active _reactivateactivate)rur prev_activerDrDrE_activate_contexts  zTracer._activate_contextcCs"ddlm}tdddtd|S)Nr)data_streams_processorz+Tracer.data_streams_processor is deprecatedz:Use ddtrace.data_streams.data_streams_processor() instead.5.0.0)prefixmessageremoval_versioncategory)ddtrace.internal.datastreamsrr7r3)rurrDrDrErs zTracer.data_streams_processorcCstddtddS)Nz4Setting Tracer.data_streams_processor is unsupportedr)rrr)r7r3rrDrDrErs  cCs |jjjSryrfsampling_processorsamplerr~rDrDrErs zTracer._samplercCs||jj_dSryrrrDrDrErscCs ttjSry)rW isEnabledForloggingDEBUGr~rDrDrE debug_loggings zTracer.debug_loggingcOs,|j}t|tr |St|tr|jSdS)znReturn the context for the current trace. If there is no active trace then None is returned. N)r_rr{r rr)ruargskwargsrrDrDrEcurrent_trace_contexts   zTracer.current_trace_contextrc Cs||dur|js tjr|j}t}}|r)|jrt|jn|}|jr't |jn|}t |t |t t jp2ttt jp7ttt jpGs  z#Tracer.configure.. reset_buffer)r` _asm_enabled_asm_enabled_origin _iast_enabled_apm_tracing_enabledrar0r]r^r@rWrany _recreater_)rur_rrrrrrrDrDrE configures:     zTracer.configurec CstjstjrRzt}Wn ty,}zd|}|tjd|WYd}~dSd}~wwt tj r>d|}|tj || d}|rTd|}|tj|dSdSdS)Nz#Failed to collect start-up logs: %sz - DATADOG TRACER DIAGNOSTIC - %sz#- DATADOG TRACER CONFIGURATION - %s agent_error) r0 _debug_mode_startup_logs_enabledrcollectrs _log_compatrWARNINGrWrINFOget)ruinforwmsgrrDrDrE_generate_diagnostic_logsWs"     z Tracer._generate_diagnostic_logscCst|_|jddd|_dS)NTr)rr\rrnr~rDrDrErljs  zTracer._child_after_forkT apm_opt_outrcCs&|jj|||||dt|j|_dS)z6Re-initialize the tracer's processors and trace writer)user_processors compute_statsrrrN)rfresetrFrbrc)rurrrrrrDrDrEros  zTracer._recreateFrchild_ofroresource span_typerspan_apic Cs^|jrd|_t|tr|j}|j|ur|j||}d} d} d} d} |dur>> span = tracer.trace("web.request") try: # do something finally: span.finish() >>> with tracer.trace("web.request") as span: # do something Example of the automatic parenting:: parent = tracer.trace("parent") # has no parent span assert tracer.current_span() is parent child = tracer.trace("child") assert child.parent_id == parent.span_id assert tracer.current_span() is child child.finish() # parent is now the active span again assert tracer.current_span() is parent parent.finish() assert tracer.current_span() is None parent2 = tracer.trace("parent2") assert parent2.parent_id is None parent2.finish() T)rrorrrr)rr_r)rurrorrrrDrDrEtrace[s0z Tracer.tracecCs|}|dur dS|jS)aReturns the local root span of the current execution/process. Note: This cannot be used to access the true root span of the trace in a distributed tracing setup if the actual root span occurred in another execution/process. This is useful for attaching information to the local root span of the current execution/process, which is often also service entry span. For example:: # get the local root span local_root_span = tracer.current_root_span() # set the host just once on the root span if local_root_span: local_root_span.set_tag('host', '127.0.0.1') N)rrrrDrDrErszTracer.current_root_spancCs|j}t|tr |SdS)zReturn the active span in the current execution context. Note that there may be an active span represented by a context object (like from a distributed trace) which will not be returned by this method. N)r_rr{r)rurrDrDrErs zTracer.current_spancCst|jjtr |jjjSdS)zTrace agent urlN)r{rfr|r5r}r~rDrDrEagent_trace_urls zTracer.agent_trace_urlcCs|jjdS)z^Flush the buffer of the trace writer. This does nothing if an unbuffered trace writer is used.N)rfr| flush_queuer~rDrDrEflushsz Tracer.flushf span_namec*tfdd}tt|S)'Wrap a generator function with tracing.c ?stjdurj||dSjd|i|EdH}|WdS1s3wYdSNrorrr[r)rr return_valuerrrurorrrDrE func_wrappers   $z,Tracer._wrap_generator..func_wrapper functoolswrapsr r9rurrrorrrrDrrE_wrap_generators  zTracer._wrap_generatorcr)rcs^jd|i|2z 3dHW}|Vq6WddS1s(wYdSrr)rrrrrDrErs "z2Tracer._wrap_generator_async..func_wrapperrrrDrrE_wrap_generator_asyncs  zTracer._wrap_generator_asynccs"dtdtffdd }|S)a# A decorator used to trace an entire function. If the traced function is a coroutine, it traces the coroutine execution when is awaited. :param str name: the name of the operation being traced. If not set, defaults to the fully qualified function name. :param str service: the name of the service being traced. If not set, it will inherit the service from its parent. :param str resource: an optional name of the resource being tracked. :param str span_type: an optional operation type. >>> @tracer.wrap('my.wrapped.function', service='my.service') def run(): return 'run' >>> # name will default to 'execute' if unset @tracer.wrap() def execute(): return 'executed' >>> # or use it in asyncio coroutines @tracer.wrap() async def coroutine(): return 'executed' >>> @tracer.wrap() @asyncio.coroutine def coroutine(): return 'executed' >>> # or use it on generators @tracer.wrap() def gen(): yield 'executed' >>> @tracer.wrap() async def gen(): yield 'executed' You can access the current span using `tracer.current_span()` to set tags: >>> @tracer.wrap() def execute(): span = tracer.current_span() span.set_tag('a', 'b') rr<csrndjjftrjd}|Str.jd}|StrDt fdd}|St fdd}|S)Nz%s.%srcsLjd|i|IdHWdS1swYdSrrrrrrDrErCs$z9Tracer.wrap..wrap_decorator..func_wrapperc shjdurj||dSjd|i|WdS1s-wYdSrrrrrDrErJs  $) __module____name__inspectisgeneratorfunctionrisasyncgenfunctionrrrr)rrrrruror)rrrEwrap_decorator*s4 , $z#Tracer.wrap..wrap_decorator)r9)rurrorrr rDr rEwraps76z Tracer.wraprXcCs|j|dS)zSet some tags at the tracer level. This will append those tags to each span created by the tracer. :param dict tags: dict of tags to set at tracer level N)rZupdate)rurXrDrDrErbszTracer.set_tagsrcCs|jjdds dSzLt|jtj|jgD] }|r||qd|_d|_ |j d|j |j krNt|jt|jt|j|j |_ W|jdSW|jdS|jw)apShutdown the tracer and flush finished traces. Avoid calling shutdown multiple times. :param timeout: How long in seconds to wait for the background worker to flush traces before exiting or :obj:`None` to block until flushing has successfully completed (default: :obj:`None`) :type timeout: :obj:`int` | :obj:`float` | :obj:`None` F)blockingN)rmacquirerrcrrrfrr^r[r_rrrrunregister_before_forkrhr unregisterrjrlrelease)rur processorrDrDrErjs$       zTracer.shutdown)r<Nry)NNNrNNN)NNNNT)NNN)NNNN)5rr __qualname____doc__rrVrxpropertyrrrsetterrjrrhrr rrrrrr rrtrrboollistrrrrlrr'rrrrrrrrrrr9rrrrrfloatrrDrDrDrErG_s~  C       (   <   )  :  "   orG)` contextlibrrr r itertoolsrrrr threadingrtypingrrr r r ddtrace._trace.contextr ddtrace._trace.processorr rrr*ddtrace._trace.processor.resource_renamingrddtrace._trace.providerrrddtrace._trace.spanrddtrace.appsec._constantsrddtrace.constantsrrrrddtrace.internalrrrrrddtrace.internal.constantsrr r!r"r#r$r%r&r'ddtrace.internal.hostnamer(ddtrace.internal.loggerr)ddtrace.internal.nativer*r+'ddtrace.internal.peer_service.processorr,0ddtrace.internal.processor.endpoint_call_counterr-ddtrace.internal.runtimer.!ddtrace.internal.schema.processorr/!ddtrace.internal.settings._configr0ddtrace.internal.settings.asmr`&ddtrace.internal.settings.peer_servicer1ddtrace.internal.utilsr2#ddtrace.internal.utils.deprecationsr3ddtrace.internal.utils.formatsr4ddtrace.internal.writerr5r6ddtrace.vendor.debtcollectorr7ddtrace.versionr8rrWr9rrFobjectrGrDrDrDrEs