o i @sUddlZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z e eZee efZejd dd Zejeeed <Gd d d ejdZGdddeZdS)N)Any)Optional)Union)Context)Span)core) get_loggerdatadog_contextvar)default_DD_CONTEXTVARc@sreZdZdZejdefddZejdee ddfddZ ejdee fd d Z d e d e dee fd dZ dS)BaseContextProviderat A ``ContextProvider`` is an interface that provides the blueprint for a callable class, capable to retrieve the current active ``Context`` instance. Context providers must inherit this class and implement: * the ``active`` method, that returns the current active ``Context`` * the ``activate`` method, that sets the current active ``Context`` returncCdSNselfrrK/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/_trace/provider.py_has_active_context z'BaseContextProvider._has_active_contextctxNcCstd|fdS)Nz!ddtrace.context_provider.activate)rdispatchrrrrractivate$szBaseContextProvider.activatecCrrrrrrractive(rzBaseContextProvider.activeargskwargscOs|S)zMethod available for backward-compatibility. It proxies the call to ``self.active()`` and must not do anything more. )r)rrrrrr__call__,szBaseContextProvider.__call__)__name__ __module__ __qualname____doc__abcabstractmethodboolrr ActiveTracerrrrrrrrr s r ) metaclasscsreZdZdZdfdd ZdefddZdeeddffd d Z deefd d Z d e deefddZ Z S)DefaultContextProviderzContext provider that retrieves contexts from a context variable. It is suitable for synchronous programming and for asynchronous executors that support contextvars. r Ncstt|dSr)superr'__init__r __class__rrr):szDefaultContextProvider.__init__cCst}|duS)zDReturns whether there is an active context in the current execution.N)r getrrrrr=sz*DefaultContextProvider._has_active_contextrcst|tt||dS)z8Makes the given context active in the current execution.N)r setr(r'rrr*rrrBs zDefaultContextProvider.activatecCs"t}t|tur||S|S)z=Returns the active span or context for the current execution.)r r,typer_update_active)ritemrrrrGs  zDefaultContextProvider.activespancCsf|}|r(|jdur(|jdur|jr|jjr||j|jS|j}|r(|jdus ||ur1|||S)zUpdates the active trace in an executor. When a span finishes, the active span becomes its parent. If no parent exists and the context is reactivatable, that context is restored. N) duration_ns_parent_parent_context _reactivater)rr1 new_activerrrr/Qs  z%DefaultContextProvider._update_active)r N)rrr r!r)r$rrr%rrrr/ __classcell__rrr*rr'3s r')r" contextvarstypingrrrddtrace._trace.contextrddtrace._trace.spanrddtrace.internalrddtrace.internal.loggerrrlogr% ContextVarr __annotations__ABCMetar r'rrrrs