o BDi3@s<ddlmZmZddlmZddlmZmZddlmZddlm Z m Z ddl m Z ddlmZmZmZmZddlmZmZdd lmZmZdd lmZmZdd lmZmZdd lm Z dd l!m"Z"m#Z$ddl%m&Z&ddl'm(Z)erddlm*Z*m+Z+m,Z,ddl-m.Z/ddl0m1Z1m2Z2dZ3dZ4dZ5ddZ6Gddde Z7dS))datetimetimezone)time) TYPE_CHECKINGcast) get_value) SpanProcessor ReadableSpan)SpanAttributes)format_span_idformat_trace_idget_current_spanSpanKind)INVALID_SPAN_IDINVALID_TRACE_ID) get_clientstart_transaction) INSTRUMENTER SPANSTATUS)SENTRY_BAGGAGE_KEYSENTRY_TRACE_KEYadd_global_event_processor) TransactionSpan)Dsn) parse_url)AnyOptionalUnion)context)EventHintotel z auto.otelcCst}|jdtjkr |St|dr|ddkr|St}|s!|S|}|jtks/|j t kr1|S| t |j d}|s>|S| di}| di||S)N instrumentertype transactioncontextstrace)roptionsrOTELhasattrr get_span_contexttrace_idrspan_idrgetr setdefaultupdateget_trace_context)event otel_span_mapclient otel_spanctx sentry_spanr(r:m/var/www/Datamplify/venv/lib/python3.10/site-packages/sentry_sdk/integrations/opentelemetry/span_processor.py!link_trace_context_to_error_event(s" r<cszeZdZdZiZiZfddZddZddZdd d Z d d Z d dZ ddZ ddZ ddZddZddZZS)SentrySpanProcessorzZ Converts OTel spans into Sentry spans so they can be sent to the Sentry backend. cst|ds t||_|jS)Ninstance)r,super__new__r>)cls __class__r:r;r@Ps zSentrySpanProcessor.__new__cstfdd}dS)Ncs t|jSN)r<r5)r4hintselfr:r;global_event_processorYs z.global_event_processorr)rGrHr:rFr;__init__WszSentrySpanProcessor.__init__cCspttd}t|jD]'}|j|tkr|j|q||tkr5|j|D] }|j|dq+qdS)z? Prune spans that have been open for too long. <N) intrlist open_spanskeyssetpopSPAN_MAX_TIME_OPEN_MINUTESr5)rGcurrent_time_minutesspan_start_minutesr/r:r:r;_prune_old_spans^s z$SentrySpanProcessor._prune_old_spansNc CsLt}|jsdSzt|j}Wn tyYdSw|jdtjkr$dS|js+dS| |r2dS| ||}|d}|rD|j |nd}d}|j durWt|j dtj}d} |rj|j|d|j|tjtd} nt|j|d||d|d|tjtd} | |j |d<|j durt|j dd } |j| t|d|dS) Nr%parent_span_ideAr/)r/namestart_timestampr%originr.baggage)rWr/rUr.rZrXr%rYrJ)rdsnr Exceptionr*rr+r-is_valid_is_sentry_span_get_trace_datar5r0 start_timer fromtimestamprutc start_childrW SPAN_ORIGINrrKrMr1rOaddrT) rGr7parent_contextr6_ trace_datarUsentry_parent_spanrXr9span_start_in_minutesr:r:r;on_startpsf         zSentrySpanProcessor.on_startcCs t}|jdtjkr dS|}|jsdSt|j}|j |d}|s&dS|j |_ | ||t |trI|j |_ |t|||||n|||d}|jdur`t|jdtj}|j|d|jdurt|jdd}|j|t|| dS)Nr%rV) end_timestamprJ)!rr*rr+r-r]r r/r5rPrWop_update_span_with_otel_status isinstancer set_contextOPEN_TELEMETRY_CONTEXT_get_otel_context"_update_transaction_with_otel_data_update_span_with_otel_dataend_timerrarrbfinishr`rKrMr1rOdiscardrT)rGr7r6 span_contextr/r9rlrjr:r:r;on_ends>           zSentrySpanProcessor.on_endcCsZd}|jdur|jtj}td|}d}t}|jr!t|jj}|r+|r+||vr+dSdS)zs Break infinite loop: HTTP requests to Sentry are caught by OTel and send again to Sentry. N Optional[str]TF) attributesr0r HTTP_URLrrr[rnetloc)rGr7 otel_span_urldsn_urlr6r:r:r;r^s   z#SentrySpanProcessor._is_sentry_spancCs4i}|jr t|j|d<|jjrt|jj|d<|S)z Returns the OTel context for Sentry. See: https://develop.sentry.dev/sdk/performance/opentelemetry/#step-5-add-opentelemetry-context r{resource)r{dictr)rGr7r8r:r:r;rrs z%SentrySpanProcessor._get_otel_contextc Csi}|}t|j}||d<t|j}||d<|jr!t|jjnd}||d<tt|}td|}|r7|dnd|d<tt |} | |d<|S)z^ Extracts tracing information from one OTel span and its parent OTel context. r/r.NrUz!dict[str, Union[str, bool, None]]parent_sampledrZ) r-r r/r r.parentrrrr) rGr7rfrhrxr/r.rUsentry_trace_datarZr:r:r;r_s      z#SentrySpanProcessor._get_trace_datacCs4|jjrdS|jjr|tjdS|tjdS)z? Set the Sentry span status from the OTel span N)statusis_unsetis_ok set_statusrOKINTERNAL_ERROR)rGr9r7r:r:r;rn$s  z1SentrySpanProcessor._update_span_with_otel_statuscCs|d|j|j}|j}|jdur|jD] \}}|||q|jtj}td|}|jtj }|rd}|jt j krD|d7}n |jt j krN|d7}|}|jtj d} | ra|d| 7}|jtjd} | rr|d| 7}| s| s|jtjd} td| } | rt| } d| j| j| j} |d| 7}|jtjd} td | } | r|| n|rd }|jtjd}td|}|r|}||_||_dS) z Convert OTel span data and update the Sentry span with it. This should eventually happen on the server when ingesting the spans. z otel.kindNrzhttp.server.clientz {}z {}://{}{} Optional[int]db)set_datakindrWr{itemsr0r HTTP_METHODr DB_SYSTEMrSERVERCLIENT NET_PEER_NAMEformat HTTP_TARGETr|urlparseschemer}pathHTTP_STATUS_CODEset_http_status DB_STATEMENTrm description)rGr9r7rmrkeyval http_methoddb_query peer_nametargeturl parsed_url status_code statementr:r:r;rt2s\           z/SentrySpanProcessor._update_span_with_otel_datacCs|jdurdS|jtj}|r?|jtj}td|}|r#||d}|jtj kr0|d7}n |jtj kr:|d7}||_ dSdS)Nrrrr) r{r0r rrrrrrrrrm)rGr9r7rrrmr:r:r;rsss       z6SentrySpanProcessor._update_transaction_with_otel_datarD)__name__ __module__ __qualname____doc__r5rMr@rIrTrkryr^rrr_rnrtrs __classcell__r:r:rBr;r=Es  C.Ar=N)8rrrtypingrropentelemetry.contextropentelemetry.sdk.tracerr OTelSpanopentelemetry.semconv.tracer opentelemetry.tracer r r ropentelemetry.trace.spanrr sentry_sdkrrsentry_sdk.constsrr,sentry_sdk.integrations.opentelemetry.constsrrsentry_sdk.scopersentry_sdk.tracingrr SentrySpansentry_sdk.utilsr urllib3.utilrrrrr opentelemetryr context_apisentry_sdk._typesr!r"rqrQrdr<r=r:r:r:r;s0