File: //var/opt/nydus/ops/mysql/opentelemetry/trace/propagation/__pycache__/tracecontext.cpython-39.pyc
a
�,�hN � @ sf d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z m
Z
d dl m Z G dd� dej
�ZdS ) � N)�trace)�Context)�textmap)�format_span_id�format_trace_id)�
TraceStatec @ s� e Zd ZdZdZdZdZe�e�Z de
j fe
j e
je e
je
j ed�dd�Zde
jfe
j e
je e
je
j dd �d
d �Zee
je d �d
d��ZdS )� TraceContextTextMapPropagatorz6Extracts and injects using w3c TraceContext's headers.Z traceparent�
tracestatezK^[ ]*([0-9a-f]{2})-([0-9a-f]{32})-([0-9a-f]{16})-([0-9a-f]{2})(-.*)?[ ]*$N)�carrier�context�getter�returnc
C s |du rt � }|�|| j�}|s$|S t�| j|d �}|s>|S |�d�}|�d�}|�d�}|�d�} |dksv|dkrz|S |d kr�|�d
�r�|S |d kr�|S |�|| j�}
|
du r�d} n
t� |
�} t
j t |d �t |d �d
t
�
t | d ��| d�} t
�t
�| �|�S )z�Extracts SpanContext from the carrier.
See `mysql.opentelemetry.propagators.textmap.TextMapPropagator.extract`
Nr � � � � Z 00000000000000000000000000000000Z0000000000000000Z00� �ff� T)�trace_id�span_idZ is_remote� trace_flags� trace_state)r �get�_TRACEPARENT_HEADER_NAME�re�search� _TRACEPARENT_HEADER_FORMAT_RE�group�_TRACESTATE_HEADER_NAMEr Z from_headerr Z SpanContext�intZ
TraceFlagsZset_span_in_contextZNonRecordingSpan)
�selfr
r r �header�match�versionr r r Ztracestate_headersr � span_context� r& �e/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/opentelemetry/trace/propagation/tracecontext.py�extract$ s>
�z%TraceContextTextMapPropagator.extract)r
r �setterr
c C sz t �|�}|�� }|t jkr dS dt|j�� dt|j�� d|jd��}|� || j
|� |j rv|j � � }|� || j
|� dS )z~Injects SpanContext into the carrier.
See `mysql.opentelemetry.propagators.textmap.TextMapPropagator.inject`
Nz00-�-Z02x)r Zget_current_spanZget_span_contextZINVALID_SPAN_CONTEXTr r r r r �setr r Z to_headerr )r! r
r r) �spanr% Ztraceparent_stringZtracestate_stringr& r& r' �injectW s
&
z$TraceContextTextMapPropagator.inject)r
c C s | j | jhS )z�Returns a set with the fields set in `inject`.
See
`mysql.opentelemetry.propagators.textmap.TextMapPropagator.fields`
)r r )r! r& r& r' �fieldsk s z$TraceContextTextMapPropagator.fields)�__name__�
__module__� __qualname__�__doc__r r Z_TRACEPARENT_HEADER_FORMATr �compiler r Zdefault_getterZCarrierT�typing�Optionalr ZGetterr( Zdefault_setterZSetterr- �property�Set�strr. r& r& r& r' r s0 �
�
� 6�
� r )r r4 Zmysql.opentelemetryr Z#mysql.opentelemetry.context.contextr Z mysql.opentelemetry.propagatorsr Zmysql.opentelemetry.tracer r Z mysql.opentelemetry.trace.spanr ZTextMapPropagatorr r& r& r& r' �<module> s