관리-도구
편집 파일: propagator.cpython-38.pyc
U �Mg� � @ s� d dl mZ d dlmZmZmZ d dlmZmZm Z m Z m Z m Z d dl mZmZmZ d dlmZmZ d dlmZ d dlmZmZ d dlmZmZ d d l m Z e r�d d l m Z m Z G d d � d e �Z d S )� )�trace)�Context� get_current� set_value)�CarrierT�Getter�Setter�TextMapPropagator�default_getter�default_setter)�NonRecordingSpan� SpanContext� TraceFlags)�SENTRY_BAGGAGE_KEY�SENTRY_TRACE_KEY)�SentrySpanProcessor)�BAGGAGE_HEADER_NAME�SENTRY_TRACE_HEADER_NAME)�Baggage�extract_sentrytrace_data)� TYPE_CHECKING)�Optional�Setc @ s8 e Zd ZdZdefdd�Zdefdd�Zedd� �Z dS ) �SentryPropagatorz[ Propagates tracing headers for Sentry's tracing system in a way OTel understands. Nc C s� |d krt � }|�|t�}|s"|S t|d �}|s6|S tt||�}|d |d }}tt|d�t|d�ttj �dd�}|�|t �} | r�t � | d �} n t i d�} | � � tt| |�}t|�} t�| |�} | S )Nr �trace_idZparent_span_id� T)r �span_idZ trace_flagsZ is_remote)Z sentry_items)r �getr r r r r �intr ZSAMPLEDr r Zfrom_incoming_header�freezer r r Zset_span_in_context) �self�carrier�context�getterZ sentry_traceZ sentrytracer r Z span_contextZbaggage_header�baggage�spanZmodified_context� r& �^/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/opentelemetry/propagator.py�extract- s2 � zSentryPropagator.extractc C s� |d krt � }t�|�}|�� }|js*d S t�|j�}t� j}|� |d �}|sRd S |� |t |� � � |j r�|j �� } | r�| �� } | r�|� |t| � d S �N)r r Zget_current_spanZget_span_contextZis_validZformat_span_idr r Z otel_span_mapr �setr Zto_traceparentZcontaining_transactionZ get_baggage� serializer ) r r! r" �setterZ current_spanZcurrent_span_contextr Zspan_mapZ sentry_spanr$ Z baggage_datar& r& r' �injectW s$ zSentryPropagator.injectc C s t thS r) )r r )r r& r& r' �fieldsr s zSentryPropagator.fields) �__name__� __module__� __qualname__�__doc__r r( r r- �propertyr. r& r& r& r' r ( s *r N)!Z opentelemetryr Zopentelemetry.contextr r r Z!opentelemetry.propagators.textmapr r r r r r Zopentelemetry.tracer r r Z,sentry_sdk.integrations.opentelemetry.constsr r Z4sentry_sdk.integrations.opentelemetry.span_processorr Zsentry_sdk.tracingr r Zsentry_sdk.tracing_utilsr r �typingr r r r r& r& r& r' �<module> s