File: //var/opt/nydus/ops/mysql/opentelemetry/sdk/trace/__pycache__/__init__.cpython-39.pyc
a
�,�h�� �
@ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z d dl
mZ d dlmZ d dlmZmZ d dlmZmZmZmZmZmZmZmZm Z d dl m Z d d l m Z d d
l m!Z"m#Z$ d d l%m&Z& d d l'm(Z( d d
l)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1 d dl2m3Z3 d dl4m5Z5 d dl6m7Z7m8Z8 d dl9m:Z: d dl;m<Z<m=Z= d dl>m?Z? d dl@mAZAmBZB d dlCmZ e�DeE�ZFdZGdZHdZIdZJdZKdZLdZMG dd� d�ZNG dd� deN�ZOG d d � d eN�ZPG d d � d e jQ�ZRG d d!� d!eR�ZSd"d#� ZTG d$d%� d%�ZUG d&d'� d'�ZVeVeVjWeVjWeVjWeVjWeVjWeVjWeVjWeVjWd(�ZXeV�Yde.eH�ZZG d)d*� d*e$j[eU�Z[G d+d,� d,e[�Z\G d-d.� d.e$j]�Z]G d/d0� d0e$j^�Z^dS )1� N)� OrderedDict)�contextmanager)�environ)�time_ns)�MappingProxyType�
TracebackType) �Any�Callable�Dict�Iterator�Optional�Sequence�Tuple�Type�Union)�filterwarnings)�
deprecated)�context�trace)�BoundedAttributes)�util)�OTEL_ATTRIBUTE_COUNT_LIMIT�!OTEL_ATTRIBUTE_VALUE_LENGTH_LIMIT� OTEL_EVENT_ATTRIBUTE_COUNT_LIMIT� OTEL_LINK_ATTRIBUTE_COUNT_LIMIT� OTEL_SPAN_ATTRIBUTE_COUNT_LIMIT�&OTEL_SPAN_ATTRIBUTE_VALUE_LENGTH_LIMIT�OTEL_SPAN_EVENT_COUNT_LIMIT�OTEL_SPAN_LINK_COUNT_LIMIT)�Resource)�sampling)� IdGenerator�RandomIdGenerator)� BoundedList)�InstrumentationInfo�InstrumentationScope)� SpanContext)�Status�
StatusCode)�types� � c @ sZ e Zd ZdZddeej dd�dd�Zddd�d d
�Zdd �d d
�Z de
e d�dd�Z dS )�
SpanProcessora Interface which allows hooks for SDK's `Span` start and end method
invocations.
Span processors can be registered directly using
:func:`TracerProvider.add_span_processor` and they are invoked
in the same order as they were registered.
N�Span��span�parent_context�returnc C s dS )a� Called when a :class:`mysql.opentelemetry.trace.Span` is started.
This method is called synchronously on the thread that starts the
span, therefore it should not block or throw an exception.
Args:
span: The :class:`mysql.opentelemetry.trace.Span` that just started.
parent_context: The parent context of the span that just started.
N� ��selfr/ r0 r2 r2 �Y/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/opentelemetry/sdk/trace/__init__.py�on_startQ s zSpanProcessor.on_start� ReadableSpan�r/ r1 c C s dS )a2 Called when a :class:`mysql.opentelemetry.trace.Span` is ended.
This method is called synchronously on the thread that ends the
span, therefore it should not block or throw an exception.
Args:
span: The :class:`mysql.opentelemetry.trace.Span` that just ended.
Nr2 �r4 r/ r2 r2 r5 �on_end` s zSpanProcessor.on_end�r1 c C s dS )zPCalled when a :class:`mysql.opentelemetry.sdk.trace.TracerProvider` is shutdown.Nr2 �r4 r2 r2 r5 �shutdownj s zSpanProcessor.shutdown�0u ��timeout_millisr1 c C s dS )a+ Export all ended spans to the configured Exporter that have not yet
been exported.
Args:
timeout_millis: The maximum amount of time to wait for spans to be
exported.
Returns:
False if the timeout is exceeded, True otherwise.
Nr2 �r4 r@ r2 r2 r5 � force_flushm s zSpanProcessor.force_flush)N)r> )
�__name__�
__module__� __qualname__�__doc__r � context_api�Contextr6 r: r= �int�boolrB r2 r2 r2 r5 r, H s ��
r, c @ sr e Zd ZdZdd� Zedd�dd�Zddeej dd �d
d �Z
d dd
�dd�Z dd�dd�Z de
ed�dd�ZdS )� SynchronousMultiSpanProcessorz�Implementation of class:`SpanProcessor` that forwards all received
events to a list of span processors sequentially.
The underlying span processors are called in sequential order as they were
added.
c C s d| _ t�� | _d S )Nr2 )�_span_processors� threading�Lock�_lockr< r2 r2 r5 �__init__� s z&SynchronousMultiSpanProcessor.__init__N��span_processorr1 c C s: | j � | j|f7 _W d � n1 s,0 Y dS �z:Adds a SpanProcessor to the list handled by this instance.N�rO rL �r4 rR r2 r2 r5 �add_span_processor� s z0SynchronousMultiSpanProcessor.add_span_processorr- r. c C s | j D ]}|j||d� qd S )N�r0 )rL r6 )r4 r/ r0 �spr2 r2 r5 r6 � s
z&SynchronousMultiSpanProcessor.on_startr7 r8 c C s | j D ]}|�|� qd S �N)rL r: )r4 r/ rX r2 r2 r5 r: � s
z$SynchronousMultiSpanProcessor.on_endr; c C s | j D ] }|�� qdS )z7Sequentially shuts down all underlying span processors.N)rL r= )r4 rX r2 r2 r5 r= � s
z&SynchronousMultiSpanProcessor.shutdownr> r? c C sJ t � |d }| jD ]0}t � }||kr, dS |�|| d �s dS qdS )a� Sequentially calls force_flush on all underlying
:class:`SpanProcessor`
Args:
timeout_millis: The maximum amount of time over all span processors
to wait for spans to be exported. In case the first n span
processors exceeded the timeout followup span processors will be
skipped.
Returns:
True if all span processors flushed their spans within the
given timeout, False otherwise.
i@B FT)r rL rB )r4 r@ Z deadline_nsrX Zcurrent_time_nsr2 r2 r5 rB � s
z)SynchronousMultiSpanProcessor.force_flush)N)r> )rC rD rE rF rP r, rV r rG rH r6 r: r= rI rJ rB r2 r2 r2 r5 rK | s �� rK c @ s� e Zd ZdZd ed�dd�Zedd�dd �Zeeged
f e e d �d d
�Z
d de e j
dd�dd�Zddd�dd�Zdd�dd�Zd eed�dd �ZdS ) � ConcurrentMultiSpanProcessora� Implementation of :class:`SpanProcessor` that forwards all received
events to a list of span processors in parallel.
Calls to the underlying span processors are forwarded in parallel by
submitting them to a thread pool executor and waiting until each span
processor finished its work.
Args:
num_threads: The number of threads managed by the thread pool executor
and thus defining how many span processors can work in parallel.
� )� num_threadsc C s$ d| _ t�� | _tjj|d�| _d S )Nr2 )� max_workers)rL rM rN rO �
concurrent�futures�ThreadPoolExecutor� _executor)r4 r\ r2 r2 r5 rP � s
z%ConcurrentMultiSpanProcessor.__init__NrQ c C s: | j � | j|f7 _W d � n1 s,0 Y dS rS rT rU r2 r2 r5 rV � s z/ConcurrentMultiSpanProcessor.add_span_processor).N)�func�args�kwargsc O sN g }| j D ],}| jj||�g|�R i |��}|�|� q
|D ] }|�� q<d S rY )rL ra �submit�append�result)r4 rb rc rd r_ rX �futurer2 r2 r5 �_submit_and_await� s
z.ConcurrentMultiSpanProcessor._submit_and_awaitr- r. c C s | j dd� ||d� d S )Nc S s | j S rY )r6 �rX r2 r2 r5 �<lambda>� � z7ConcurrentMultiSpanProcessor.on_start.<locals>.<lambda>rW �ri r3 r2 r2 r5 r6 � s
�z%ConcurrentMultiSpanProcessor.on_startr7 r8 c C s | � dd� |� d S )Nc S s | j S rY )r: rj r2 r2 r5 rk � rl z5ConcurrentMultiSpanProcessor.on_end.<locals>.<lambda>rm r9 r2 r2 r5 r: � s z#ConcurrentMultiSpanProcessor.on_endr; c C s | � dd� � dS )z6Shuts down all underlying span processors in parallel.c S s | j S rY )r= rj r2 r2 r5 rk � rl z7ConcurrentMultiSpanProcessor.shutdown.<locals>.<lambda>Nrm r< r2 r2 r5 r= � s z%ConcurrentMultiSpanProcessor.shutdownr> r? c C sh g }| j D ] }| j�|j|�}|�|� q
|d }tj�||�\}}|rLdS |D ]}|�� sP dS qPdS )aF Calls force_flush on all underlying span processors in parallel.
Args:
timeout_millis: The maximum amount of time to wait for spans to be
exported.
Returns:
True if all span processors flushed their spans within the given
timeout, False otherwise.
g @�@FT) rL ra re rB rf r^ r_ �waitrg )r4 r@ r_ rX rh Z timeout_secZ done_futuresZnot_done_futuresr2 r2 r5 rB � s
z(ConcurrentMultiSpanProcessor.force_flush)r[ )N)r> )rC rD rE rF rI rP r, rV r r ri r rG rH r6 r: r= rJ rB r2 r2 r2 r5 rZ � s � �� rZ c @ sb e Zd Zd eee dd�dd�Zeed�dd��Zeed�dd ��Z ee
j e j
d�d
d ���ZdS )
� EventBaseN)�name� timestampr1 c C s" || _ |d u rt� | _n|| _d S rY )�_namer �
_timestamp)r4 rp rq r2 r2 r5 rP
s
zEventBase.__init__r; c C s | j S rY �rr r< r2 r2 r5 rp s zEventBase.namec C s | j S rY )rs r< r2 r2 r5 rq s zEventBase.timestampc C s d S rY r2 r< r2 r2 r5 �
attributes s zEventBase.attributes)N)rC rD rE �strr rI rP �propertyrp rq �abc�abstractmethodr) �
Attributesru r2 r2 r2 r5 ro s ro c sT e Zd ZdZddefeejee ee dd�� fdd�
Z
e ejd�dd��Z � Z
S ) �Eventa A text annotation with a set of attributes. The attributes of an event
are immutable.
Args:
name: Name of the event.
attributes: Attributes of the event.
timestamp: Timestamp of the event. If `None` it will filled
automatically.
N)rp ru rq �limitr1 c s t � �||� || _d S rY )�superrP � _attributes)r4 rp ru rq r| �� __class__r2 r5 rP - s zEvent.__init__r; c C s | j S rY )r~ r< r2 r2 r5 ru 7 s zEvent.attributes)rC rD rE rF �(_DEFAULT_OTEL_SPAN_ATTRIBUTE_COUNT_LIMITrv r) rz r rI rP rw ru �
__classcell__r2 r2 r r5 r{ " s
��
r{ c s � fdd�}|S )Nc sf d}| j �6 | jd u r.� | g|�R i |�� nd}W d � n1 sF0 Y |rbt�d� j� d S )NFTz"Tried calling %s on an ended span.)rO � _end_time�logger�warningrC )r4 rc rd Z
already_ended�rb r2 r5 �wrapper= s
"z"_check_span_ended.<locals>.wrapperr2 )rb r� r2 r� r5 �_check_span_ended< s r� c @ s e Zd ZdZddde�i �dddejjde e
j �dddf
e ej
eej
eejee eej ejee ee ee edd�dd�Zeed�dd ��Zeed�d
d ��Zeed�d d
��Zee d�dd��Z dd� Z edd� �Z eejd�dd��Z eeej
d�dd��Z eee d�dd��Z!eee d�dd��Z"eej d�d d ��Z#eejd�d d ��Z$eee d�d d!��Z%eeej d�d"d#��Z&eed�d$d%��Z'ee(d&d'd(�ed�d)d*���Z)eed�d+d,��Z*d8d.d/�Z+e,d0d1� �Z-e,d2d3� �Z.e,d4d5� �Z/e,d6d7� �Z0dS )9r7 z,Provides read-only access to span attributesNr2 )rp r �parent�resourceru �events�links�kind�instrumentation_info�status�
start_time�end_time�instrumentation_scoper1 c C sR || _ || _|| _| | _|
| _|| _| | _| | _|| _|| _ || _
|| _ |
| _ d S rY )
rr �_context�_kind�_instrumentation_info�_instrumentation_scope�_parent� _start_timer� r~ �_events�_links� _resource�_status)r4 rp r r� r� ru r� r� r� r� r� r� r� r� r2 r2 r5 rP N s zReadableSpan.__init__r; c C s | j r| j jS dS �Nr )r~ �droppedr< r2 r2 r5 �dropped_attributesl s z ReadableSpan.dropped_attributesc C s | j r| j jS dS r� )r� r� r< r2 r2 r5 �dropped_eventsr s zReadableSpan.dropped_eventsc C s | j r| j jS dS r� )r� r� r< r2 r2 r5 �
dropped_linksx s zReadableSpan.dropped_linksc C s | j S rY rt r< r2 r2 r5 rp ~ s zReadableSpan.namec C s | j S rY �r� r< r2 r2 r5 �get_span_context� s z ReadableSpan.get_span_contextc C s | j S rY r� r< r2 r2 r5 r � s zReadableSpan.contextc C s | j S rY )r� r< r2 r2 r5 r� � s zReadableSpan.kindc C s | j S rY )r� r< r2 r2 r5 r� � s zReadableSpan.parentc C s | j S rY )r� r< r2 r2 r5 r� � s zReadableSpan.start_timec C s | j S rY �r� r< r2 r2 r5 r� � s zReadableSpan.end_timec C s | j S rY )r� r< r2 r2 r5 r� � s zReadableSpan.statusc C s
t | j�S rY )r r~ r< r2 r2 r5 ru � s zReadableSpan.attributesc C s t dd� | jD ��S )Nc s s | ]
}|V qd S rY r2 )�.0�eventr2 r2 r5 � <genexpr>� rl z&ReadableSpan.events.<locals>.<genexpr>)�tupler� r< r2 r2 r5 r� � s zReadableSpan.eventsc C s t dd� | jD ��S )Nc s s | ]
}|V qd S rY r2 )r� �linkr2 r2 r5 r� � rl z%ReadableSpan.links.<locals>.<genexpr>)r� r� r< r2 r2 r5 r� � s zReadableSpan.linksc C s | j S rY �r� r< r2 r2 r5 r� � s zReadableSpan.resourcez1.11.1z$You should use instrumentation_scope)�version�reasonc C s | j S rY )r� r< r2 r2 r5 r� � s z!ReadableSpan.instrumentation_infoc C s | j S rY )r� r< r2 r2 r5 r� � s z"ReadableSpan.instrumentation_scope� c C sd d }| j d urVt| j t�r6| j j}dt�|j�� �}n t| j t�rVdt�| j j�� �}d }| jrlt �
| j�}d }| j r�t �
| j �}| j d ur�t
� }t| j jj�|d< | j jr�| j j|d< t
� }| j|d< | �| j�|d< t| j�|d< ||d< ||d< ||d < | j d u�r||d
< | �| j�|d < | �| j�|d < | �| j�|d
< t � | j � � �|d< t j ||d�S )N�0x� status_code� descriptionrp r r� � parent_idr� r� r� ru r� r� r� )�indent)!r� �
isinstancer- r � trace_api�format_span_id�span_idr& r� r �
ns_to_iso_strr� r� r rv r� rp r� rr �_format_contextr� r� �_format_attributesr~ �_format_eventsr� �
_format_linksr� �json�loadsr� �to_json�dumps)r4 r� r� �ctxr� r� r� Zf_spanr2 r2 r5 r� � s@
zReadableSpan.to_jsonc C sD t � }dt�| j�� �|d< dt�| j�� �|d< t| j�|d< |S )Nr� �trace_idr� � trace_state)r r� Zformat_trace_idr� r� r� �reprr� )r Zx_ctxr2 r2 r5 r� � s
z ReadableSpan._format_contextc C s& t | t�r| jS t | t�r"| �� S | S rY )r� r �_dictr �copy)ru r2 r2 r5 r� � s
z ReadableSpan._format_attributesc C sL g }| D ]>}t � }|j|d< t�|j�|d< t�|j�|d< |�|� q|S )Nrp rq ru ) r rp r r� rq r- r� ru rf )r� Zf_eventsr� Zf_eventr2 r2 r5 r� � s
��� zReadableSpan._format_eventsc C sB g }| D ]4}t � }t�|j�|d< t�|j�|d< |�|� q|S )Nr ru )r r- r� r r� ru rf )r� Zf_linksr� Zf_linkr2 r2 r5 r� � s" ������ zReadableSpan._format_links)r� )1rC rD rE rF r �creater� �SpanKind�INTERNALr' r( �UNSETrv r& r r) rz r
r{ �Linkr$ rI r% rP rw r� r� r� rp r� r r� r� r� r� r� ru r� r� r� r r� r� r� � staticmethodr� r� r� r� r2 r2 r2 r5 r7 K s� ��
(
r7 c
@ s� e Zd ZdZdZd ee ee ee ee ee ee ee ee d�dd�Zdd� Ze d
ee e
ee ee d �d
d ��Z dS )�
SpanLimitsa/ The limits that should be enforce on recorded data such as events, links, attributes etc.
This class does not enforce any limits itself. It only provides an a way read limits from env,
default values and from user provided arguments.
All limit arguments must be either a non-negative integer, ``None`` or ``SpanLimits.UNSET``.
- All limit arguments are optional.
- If a limit argument is not set, the class will try to read its value from the corresponding
environment variable.
- If the environment variable is not set, the default value, if any, will be used.
Limit precedence:
- If a model specific limit is set, it will be used.
- Else if the corresponding global limit is set, it will be used.
- Else if the model specific limit has a default value, the default value will be used.
- Else if the global limit has a default value, the default value will be used.
Args:
max_attributes: Maximum number of attributes that can be added to a span, event, and link.
Environment variable: OTEL_ATTRIBUTE_COUNT_LIMIT
Default: {_DEFAULT_ATTRIBUTE_COUNT_LIMIT}
max_events: Maximum number of events that can be added to a Span.
Environment variable: OTEL_SPAN_EVENT_COUNT_LIMIT
Default: {_DEFAULT_SPAN_EVENT_COUNT_LIMIT}
max_links: Maximum number of links that can be added to a Span.
Environment variable: OTEL_SPAN_LINK_COUNT_LIMIT
Default: {_DEFAULT_SPAN_LINK_COUNT_LIMIT}
max_span_attributes: Maximum number of attributes that can be added to a Span.
Environment variable: OTEL_SPAN_ATTRIBUTE_COUNT_LIMIT
Default: {_DEFAULT_OTEL_SPAN_ATTRIBUTE_COUNT_LIMIT}
max_event_attributes: Maximum number of attributes that can be added to an Event.
Default: {_DEFAULT_OTEL_EVENT_ATTRIBUTE_COUNT_LIMIT}
max_link_attributes: Maximum number of attributes that can be added to a Link.
Default: {_DEFAULT_OTEL_LINK_ATTRIBUTE_COUNT_LIMIT}
max_attribute_length: Maximum length an attribute value can have. Values longer than
the specified length will be truncated.
max_span_attribute_length: Maximum length a span attribute value can have. Values longer than
the specified length will be truncated.
���N��max_attributes�
max_events� max_links�max_span_attributes�max_event_attributes�max_link_attributes�max_attribute_length�max_span_attribute_lengthc
C s� | � |tt�| _| � |tt�| _| � |t�} | d ur8| nt| _ | � |t
| d urR| nt �| _ | � |t
| d urn| nt�| _| � |t| d ur�| nt�| _| � |t�| _| � |t| j�| _d S rY )�_from_env_if_absentr �$_DEFAULT_OTEL_SPAN_EVENT_COUNT_LIMITr� r �#_DEFAULT_OTEL_SPAN_LINK_COUNT_LIMITr� r �#_DEFAULT_OTEL_ATTRIBUTE_COUNT_LIMITr� r r� r� r �)_DEFAULT_OTEL_EVENT_ATTRIBUTE_COUNT_LIMITr� r �(_DEFAULT_OTEL_LINK_ATTRIBUTE_COUNT_LIMITr� r r� r r� )
r4 r� r� r� r� r� r� r� r� Zglobal_max_attributesr2 r2 r5 rP = s^ ����������� ��zSpanLimits.__init__c C sH t | �j� d| j� d| j� d| j� d| j� d| j� d| j� d| j� d�S ) Nz(max_span_attributes=z, max_events_attributes=z, max_link_attributes=z, max_attributes=z
, max_events=z , max_links=z, max_attribute_length=�)) �typerC r� r� r� r� r� r� r� r< r2 r2 r5 �__repr__� s zSpanLimits.__repr__)�value�env_var�defaultr1 c C s� || j krd S d}|d u rv|tvr&|S t�|d��� �� }|tkrFd S z t|�}W n" tyt t|�||���Y n0 |dk r�t|�||���|S )Nz-{0} must be a non-negative integer but got {}r+ r ) r� r �get�strip�lower�_ENV_VALUE_UNSETrI �
ValueError�format)�clsr� r� r� �err_msgZ str_valuer2 r2 r5 r� � s
z SpanLimits._from_env_if_absent)NNNNNNNN)N) rC rD rE rF r� r rI rP r� � classmethodrv r� r2 r2 r2 r5 r� s6 * �� C ��r� r� c s� e Zd ZdZ� fdd�Zddde�i �dddejj e
� ddde dfe ej
eej
eej deejee eej eje
eeeedd�� fdd �
Zd
d � Zd d
� Zdd� Z dd� Z e e ej f dd�dd�Z e ej dd�dd�Z!e"e#dd�dd��Z$d6e ejee% dd�d d �Z&e'd �d d �Z(d7ee% ee)j* dd!�d"d#�Z+d8ee% dd$�d%d&�Z,e"e dd'�d(d)��Z-ed �d*d+�Z.e"d9e/j0e1e2f e/je dd,�d-d.��Z3ee4e5 ee5 ee6 dd/�� fd0d1� Z7d:e8ejee% edd3�d4d5�Z9� Z:S );r- a? See `mysql.opentelemetry.trace.Span`.
Users should create `Span` objects via the `Tracer` instead of this
constructor.
Args:
name: The name of the operation this span represents
context: The immutable span context
parent: This span's parent's `mysql.opentelemetry.trace.SpanContext`, or
None if this is a root span
sampler: The sampler used to create this span
trace_config: TODO
resource: Entity producing telemetry
attributes: The span's attributes to be exported
events: Timestamped events to be exported
links: Links to other spans to be exported
span_processor: `SpanProcessor` to invoke when starting and ending
this `Span`.
limits: `SpanLimits` instance that was passed to the `TracerProvider`
c s | t u rtd��t� �| �S )Nz'Span must be instantiated via a tracer.)r- � TypeErrorr} �__new__)r� rc rd r r2 r5 r� � s z Span.__new__Nr2 T)rp r r� �sampler� trace_configr� ru r� r� r� rR r� �record_exception�set_status_on_exceptionr� r1 c s� t � j||||
|| |d� || _|| _|
| _|| _| | _|| _t� � | _
t | jj |d| jj
d�| _| �� | _|r�|D ]*}t | jj|j| jjd�|_| j�|� qt| d u r�| �� | _n6| D ] }t | jj|j| jjd�|_q�t�| jj| �| _d S )N)rp r r� r� r� r� r� F)Z immutable�
max_value_len�r� )r} rP Z_samplerZ
_trace_config�_record_exception�_set_status_on_exception�_span_processor�_limitsrM rN rO r r� r� r~ � _new_eventsr� r� ru r� rf �
_new_linksr� r� r# Zfrom_seqr� )r4 rp r r� r� r� r� ru r� r� r� rR r� r� r� �limitsr� r� r� r r2 r5 rP � sP �
�
� �
z
Span.__init__c C s t | �j� d| j� d| j� d�S )Nz(name="z ", context=r� )r� rC rr r� r< r2 r2 r5 r� s z
Span.__repr__c C s t | jj�S rY )r# r� r� r< r2 r2 r5 r� s zSpan._new_eventsc C s t | jj�S rY )r# r� r� r< r2 r2 r5 r� s zSpan._new_linksc C s | j S rY r� r< r2 r2 r5 r� s zSpan.get_span_context)ru r1 c C sj | j �P | jd ur,t�d� W d � d S |�� D ]\}}|| j|< q4W d � n1 s\0 Y d S )Nz Setting attribute on ended span.)rO r� r� r� �itemsr~ )r4 ru �keyr� r2 r2 r5 �set_attributes s
zSpan.set_attributes)r� r� r1 c C s | � ||i�S rY )r� )r4 r� r� r2 r2 r5 �
set_attribute$ s zSpan.set_attribute)r� r1 c C s | j �|� d S rY )r� rf )r4 r� r2 r2 r5 �
_add_event' s zSpan._add_event)rp ru rq r1 c C s. t | jj|| jjd�}| �t|||d�� d S )Nr� �rp ru rq )r r� r� r� r� r{ )r4 rp ru rq r2 r2 r5 � add_event+ s ���zSpan.add_eventr; c C s< t | j| j| j| j| j| j| j| j| j | j
| j | j | j
d�
S )N)
rp r r� r� ru r� r� r� r� r� r� r� r� )r7 rr r� r� r� r~ r� r� r� r� r� r� r� r� r< r2 r2 r5 �_readable_span> s �zSpan._readable_span)r� r0 r1 c C sr | j �H | jd ur,t�d� W d � d S |d ur8|nt� | _W d � n1 sT0 Y | jj| |d� d S )Nz"Calling start() on a started span.rW )rO r� r� r� r r� r6 )r4 r� r0 r2 r2 r5 �startO s
2z
Span.start)r� r1 c C s� | j �Z | jd u rtd��| jd ur>t�d� W d � d S |d urJ|nt� | _W d � n1 sf0 Y | j�| � � � d S )Nz$Calling end() on a not started span.z Calling end() on an ended span.)
rO r� � RuntimeErrorr� r� r� r r� r: r )r4 r� r2 r2 r5 �end\ s
2zSpan.end)rp r1 c C s
|| _ d S rY rt )r4 rp r2 r2 r5 � update_nameh s zSpan.update_namec C s
| j d u S rY r� r< r2 r2 r5 � is_recordingl s zSpan.is_recording)r� r� r1 c C s� t |t�rJ| jr | jjtju s*|jtju r.d S |d urBt�d|� || _n8t |t�r�| jrh| jjtju sr|tju rvd S t||�| _d S )NzJDescription %s ignored. Use either `Status` or `(StatusCode, Description)`) r� r' r� r� r( �OKr� r� r� )r4 r� r� r2 r2 r5 �
set_statuso s.
� �
��
� ��zSpan.set_status)�exc_type�exc_val�exc_tbr1 c s^ |durJ| � � rJ| jr$| j|dd� | jrJ| �ttj|j� d|� �d�� t � �
|||� dS )z3Ends context manager and calls `end` on the `Span`.NT)� exception�escapedz: )r� r� ) r r� r� r� r r' r( �ERRORrC r} �__exit__)r4 r
r r r r2 r5 r � s ��z
Span.__exit__F)r
ru rq r r1 c C s` z t �� }W n ty" d}Y n0 |jjt|�|t|�d�}|rL|�|� | jd||d� dS )z%Records an exception as a span event.z+Exception occurred on stacktrace formatting)zexception.typezexception.messagezexception.stacktracezexception.escapedr
r N)� traceback�
format_exc� Exceptionr� rC rv �updater )r4 r
ru rq r Z
stacktracer~ r2 r2 r5 r� � s
�
zSpan.record_exception)NN)NN)N)N)NNF);rC rD rE rF r� r r� r� r� r� r, � _UnsetLimitsrv r& r r �Samplerr) rz r
r{ r� r$ rJ r% rP r� r� r� r� r
ZAttributeValuer� r� r� ro r� rI r r7 r rG rH r r r r �typingr r' r( r r �
BaseExceptionr r r r� r� r2 r2 r r5 r- � s�
��> �� ��
� �
� ��r- c @ s e Zd ZdZdS )�_Spanz�Protected implementation of `mysql.opentelemetry.trace.Span`.
This constructor exists to prevent the instantiation of the `Span` class
by other mechanisms than through the `Tracer`.
N)rC rD rE rF r2 r2 r2 r5 r � s r c
@ s� e Zd ZdZejeeee f e
e e e
dd�dd�Zedejjddddddfeeej ejejeej ee e e e e ej d�
d d
��Z dejjdddddfeeej ejejeej ee e e ej d � d d
�Z dS )�Tracerz'See `mysql.opentelemetry.trace.Tracer`.N)r� r� rR � id_generatorr� � span_limitsr� r1 c C s. || _ || _|| _|| _|| _|| _|| _d S rY )r� r� rR r r� � _span_limitsr� )r4 r� r� rR r r� r r� r2 r2 r5 rP � s zTracer.__init__r2 T)
rp r r� ru r� r� r� r� � end_on_exitr1 c
c sV | j ||||||||d�}
tj|
| ||d��} | V W d � n1 sH0 Y d S )N)rp r r� ru r� r� r� r� )r r� r� )�
start_spanr� Zuse_span) r4 rp r r� ru r� r� r� r� r r/ � span_contextr2 r2 r5 �start_as_current_span� s$
�
�z Tracer.start_as_current_span) rp r r� ru r� r� r� r� r1 c C s t �|��� } | d ur*t| t j�s*td��| d u s8| jsHd } | j�� }
n| j }
| j
� ||
||||�} | j �
� r|t �t jj�n t �t jj�} t j|
| j�� d| | jd�}
| j �� r�t||
| | j
| j| j�� | j||| j||| j| jd�}|j ||d� n t j |
d�}|S )Nz2parent_span_context must be a SpanContext or None.F)Z is_remote� trace_flagsr� )rp r r� r� r� ru rR r� r� r� r� r� r� r� )r� r0 )r ) r� Zget_current_spanr� r� r& r� Zis_validr Zgenerate_trace_idr� r� Z
should_sampleZdecisionZ
is_sampledZ
TraceFlagsZSAMPLED�DEFAULTZgenerate_span_idr� r r r� ru r� rR r� r r� r ZNonRecordingSpan)r4 rp r r� ru r� r� r� r� Zparent_span_contextr� Zsampling_resultr" r r/ r2 r2 r5 r � sV
� �� ��
� zTracer.start_span)!rC rD rE rF r r r r rK rZ r! r$ r� r% rP r r� r� r� rv r rG rH r) rz r
r� rI rJ r r- r! r r2 r2 r2 r5 r � sf �� ��!��r c @ s� e Zd ZdZde�i �ddddfejeee e
e f e e
d�dd�Zeed�dd ��Zdeeje eje d
d �d d
�Zedd�dd�Zdd� Zdeed�dd�ZdS )�TracerProviderz/See `mysql.opentelemetry.trace.TracerProvider`.NT)r� r� �shutdown_on_exit�active_span_processorr r c C sd |pt � | _|d u r t� | _n|| _|| _|s6t�� }|| _|pDt� | _ d | _
|r`t � | j
�| _
d S rY )rK �_active_span_processorr" r r� r Z_get_from_env_or_defaultr� r� r �_atexit_handler�atexit�registerr= )r4 r� r� r% r& r r r2 r2 r5 rP I s �
zTracerProvider.__init__r; c C s | j S rY r� r< r2 r2 r5 r� e s zTracerProvider.resourceztrace_api.Tracer)�instrumenting_module_name� instrumenting_library_version�
schema_urlr1 c C s` |sd}t �d� |d u r d}tddtdd� t|||�}t| j| j| j| j || j
t |||��S )Nr+ z+get_tracer called with missing module name.�ignorezoCall to deprecated method __init__. \(You should use InstrumentationScope\) -- Deprecated since version 1.11.1.z mysql.opentelemetry.sdk.trace)�message�category�module) r� �errorr �DeprecationWarningr$ r r� r� r' r r r% )r4 r+ r, r- r� r2 r2 r5 �
get_traceri s:
�
���zTracerProvider.get_tracerrQ c C s | j �|� dS )z�Registers a new :class:`SpanProcessor` for this `TracerProvider`.
The span processors are invoked in the same order they are registered.
N)r' rV rU r2 r2 r5 rV � s z!TracerProvider.add_span_processorc C s* | j �� | jdur&t�| j� d| _dS )z;Shut down the span processors added to the tracer provider.N)r' r= r( r) �
unregisterr< r2 r2 r5 r= � s
zTracerProvider.shutdownr> r? c C s | j �|�S )a� Requests the active span processor to process all spans that have not
yet been processed.
By default force flush is called sequentially on all added span
processors. This means that span processors further back in the list
have less time to flush their spans.
To have span processors flush their spans in parallel it is possible to
initialize the tracer provider with an instance of
`ConcurrentMultiSpanProcessor` at the cost of using multiple threads.
Args:
timeout_millis: The maximum amount of time to wait for spans to be
processed.
Returns:
False if the timeout is exceeded, True otherwise.
)r' rB rA r2 r2 r5 rB � s zTracerProvider.force_flush)NN)r> )rC rD rE rF r r� r r rJ r rK rZ r! r� rP rw r� rv r r r4 r, rV r= rI rB r2 r2 r2 r5 r$ F s< ��� �� *
r$ )_rx r) �concurrent.futuresr^ r� �loggingrM r r � collectionsr �
contextlibr �osr �timer r) r r r r r
r r r
r r r �warningsr r Zmysql.opentelemetryr rG r r� Z mysql.opentelemetry.attributesr Zmysql.opentelemetry.sdkr Z-mysql.opentelemetry.sdk.environment_variablesr r r r r r r r Z!mysql.opentelemetry.sdk.resourcesr Z mysql.opentelemetry.sdk.tracer Z*mysql.opentelemetry.sdk.trace.id_generatorr! r" Z mysql.opentelemetry.sdk.utilr# Z,mysql.opentelemetry.sdk.util.instrumentationr$ r% Zmysql.opentelemetry.tracer& Z mysql.opentelemetry.trace.statusr' r( Zmysql.opentelemetry.util� getLoggerrC r� r� r� r� r� r� r� r� r, rK rZ �ABCro r{ r� r7 r� r� r r� ZSPAN_ATTRIBUTE_COUNT_LIMITr- r r r$ r2 r2 r2 r5 �<module> s� , (
4>R F �
��