o i@sddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZd Zd Zd ZdZddZddZddZdddZdS))web) SystemRoute)config) SpanTypes)http)core)schematize_url_operation) SpanDirection)DDTraceDeprecationWarning) deprecate datadog_tracedatadog_context__datadog_trace_config__datadog_request_spancsfdd}|S)a} ``aiohttp`` middleware that traces the handler execution. Because handlers are run in different tasks for each request, we attach the Context instance both to the Task and to the Request objects. In this way: * the Task is used by the internal automatic instrumentation * the ``Context`` attached to the request can be freely used in the application code c std}tjdtddtjdtj|ijt j dtddd V}|j }| d|t d |t j f|jt<|t<tt<z'IdHt j d scttjrcjfd d WWdStyx|w1s|wYdS) Nservicezaiohttp.requestr)protocol directionTdistributed_tracing_enabled) span_name span_typertagsdistributed_headersintegration_configactivate_distributed_headers#distributed_headers_config_overrideheaders_case_sensitivereq_spanzweb.request.start"disable_stream_timing_for_mem_leakcs tS)N)finish_request_span)_requestresponse`/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/aiohttp/middlewares.py=s z:trace_middleware..attach_context..) CONFIG_KEYrcontext_with_datarr INBOUNDrWEBheadersraiohttpspanset_itemdispatchcontextREQUEST_CONTEXT_KEYREQUEST_SPAN_KEYREQUEST_CONFIG_KEY isinstancerStreamResponsetaskadd_done_callback Exception set_traceback)r!rctxrapphandlerr r$attach_contextsB         z(trace_middleware..attach_contextr#)r;r<r=r#r:r$trace_middlewares %r>c Cs|td}|s dSt|j}|jjjrB|jjj}|dr&|d}n|dr1|d}n |dr;|d}d|j |}||_|t d}|durTt j j }|r^|tj|jd}t|jdrs|jj}t|tss|jj}td|t j|j t|j|jd|j|j|df dS) Npath formatterprefixz{} {}trace_query_stringroutezweb.request.finishT)getr1strstatus match_inforCresourceget_infoformatmethodr2r_httprB _set_tag_strr QUERY_STRING query_stringhasattrr3r canonicalrr.r+urlr*)r!r" request_spanrHres_inforBrC aiohttp_router#r#r$rFsL           rcs6tjdst|tjur|jsdSt||dS)z| The on_prepare signal is used to close the request span that is created during the trace middleware execution. rN)rr+typerr4r5donerr r#r#r$ on_prepares  rXN aiohttp-webcCsbt|ddrdSd|_|durtddtddtj|d dd |t<|jd t |j t dS) z Tracing function that patches the ``aiohttp`` application so that it will be traced using the global tracer. :param app: aiohttp application to trace :param service: service name of tracer __datadog_traceFNTz"The tracer parameter is deprecatedz'The global tracer will be used instead.z5.0.0)messagecategoryremoval_version)default)rrr) getattrrZr r r _get_servicer& middlewaresinsertr>on_response_prepareappendrX)r;tracerrr#r#r$ trace_apps   rf)NrY)r+raiohttp.web_urldispatcherrddtracer ddtrace.extrrddtrace.internalrddtrace.internal.schemar-ddtrace.internal.schema.span_attribute_schemar #ddtrace.internal.utils.deprecationsr ddtrace.vendor.debtcollectorr r&r0r2r1r>rrXrfr#r#r#r$s$         2;