File: //var/opt/nydus/ops/mysql/opentelemetry/baggage/propagation/__pycache__/__init__.cpython-39.pyc
a
�,�hY � @ s� d dl mZ d dlmZ d dlmZmZmZmZm Z d dl
m Z m Z d dl
mZmZmZ d dlmZ d dlmZ d dlmZ d d lmZ ee�ZG d
d � d ej�Z ee e f e d �d
d�Z eeej eej d�dd�Z!dS )� )� getLogger)�split)�Iterable�List�Mapping�Optional�Set)�
quote_plus� unquote_plus)�_is_valid_pair�get_all� set_baggage)� get_current)�Context)�textmap)�_DELIMITER_PATTERNc @ s� e Zd ZdZdZdZdZdZdej fej
e e ej
ej
e d�dd �Zdejfej
e e ejej
dd
�d d �Zeee d
�dd��ZdS )�W3CBaggagePropagatorzAExtracts and injects Baggage which is used to annotate telemetry.i i � ZbaggageN)�carrier�context�getter�returnc
C s, |du rt � }t|�|| j��}|s(|S t|�| jkrFt�d|� |S tt |�}| j
}t|�| j
krpt�d|� |D ]�}t|�| j kr�t�d|� qt|s�qtz|�dd�\}} W n" t y� t�d|� Y qtY n0 t
|| �s�t�d|� qtt|��� }t| ��� } t|| |d �}|d8 }|d
krt �q(qt|S ) z�Extract Baggage from the carrier.
See
`mysql.opentelemetry.propagators.textmap.TextMapPropagator.extract`
NzKBaggage header `%s` exceeded the maximum number of bytes per baggage-stringz?Baggage header `%s` exceeded the maximum number of list-memberszGBaggage entry `%s` exceeded the maximum number of bytes per list-member�=� z1Baggage list-member `%s` doesn't match the formatzInvalid baggage entry: `%s`�r r )r �_extract_first_element�get�_BAGGAGE_HEADER_NAME�len�_MAX_HEADER_LENGTH�_logger�warningr r �
_MAX_PAIRS�_MAX_PAIR_LENGTH� Exceptionr r
�stripr
)
�selfr r r �header�baggage_entriesZtotal_baggage_entries�entry�name�value� r, �c/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/opentelemetry/baggage/propagation/__init__.py�extract% s` �
�� �
�z W3CBaggagePropagator.extract)r r �setterr c C s. t |d�}|sdS t|�}|�|| j|� dS )z�Injects Baggage into the carrier.
See
`mysql.opentelemetry.propagators.textmap.TextMapPropagator.inject`
r N)r �_format_baggage�setr )r& r r r/ r( Zbaggage_stringr, r, r- �injectl s
zW3CBaggagePropagator.inject)r c C s | j hS )z.Returns a set with the fields set in `inject`.)r )r& r, r, r- �fields~ s zW3CBaggagePropagator.fields)�__name__�
__module__� __qualname__�__doc__r r# r" r r Zdefault_getter�CarrierTr r ZGetterr. Zdefault_setterZSetterr2 �propertyr �strr3 r, r, r, r- r s. �
� J�
� r )r( r c C s d� dd� | �� D ��S )N�,c s s. | ]&\}}t t|��d t t|�� V qdS )r N)r r: )�.0�keyr+ r, r, r- � <genexpr>� s �z"_format_baggage.<locals>.<genexpr>)�join�items)r( r, r, r- r0 � s
�r0 )r@ r c C s | d u r d S t t| �d �S )N)�next�iter)r@ r, r, r- r � s r N)"�loggingr �rer �typingr r r r r � urllib.parser r
Zmysql.opentelemetry.baggager r r
Zmysql.opentelemetry.contextr Z#mysql.opentelemetry.context.contextr Z mysql.opentelemetry.propagatorsr Zmysql.opentelemetry.util.rer r4 r ZTextMapPropagatorr r: �objectr0 r8 r r, r, r, r- �<module> s g �