File: //var/opt/nydus/ops/mysql/opentelemetry/sdk/trace/export/__pycache__/__init__.cpython-39.pyc
a
�,�hOD � @ s< d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlmZm Z d dl
m Z d dlm Z d dl
mZmZmZmZmZ d dlmZmZmZmZ d dlmZmZmZ d d l m Z d
Z d Z d Z d
Z!dZ"e�#e$�Z%G dd� de�Z&G dd� d�Z'G dd� de�Z(G dd� d�Z)e � Z*G dd� de�Z+G dd� de'�Z,dS )� N)�Enum)�environ�linesep)�time_ns)�Optional)� _SUPPRESS_INSTRUMENTATION_KEY�Context�attach�detach� set_value)�OTEL_BSP_EXPORT_TIMEOUT� OTEL_BSP_MAX_EXPORT_BATCH_SIZE�OTEL_BSP_MAX_QUEUE_SIZE�OTEL_BSP_SCHEDULE_DELAY)� ReadableSpan�Span�
SpanProcessor)�Oncei� i �0u i z:Unable to parse value for %s as integer. Defaulting to %s.c @ s e Zd ZdZdZdS )�SpanExportResultr � N)�__name__�
__module__� __qualname__�SUCCESS�FAILURE� r r �`/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/opentelemetry/sdk/trace/export/__init__.pyr 6 s r c @ sF e Zd ZdZeje dd�dd�Zdd�dd �Zde e
d �d d
�Z dS )� SpanExportera) Interface for exporting spans.
Interface to be implemented by services that want to export spans recorded
in their own format.
To export data this MUST be registered to the :class`mysql.opentelemetry.sdk.trace.Tracer` using a
`SimpleSpanProcessor` or a `BatchSpanProcessor`.
r ��spans�returnc C s dS )z�Exports a batch of telemetry data.
Args:
spans: The list of `mysql.opentelemetry.trace.Span` objects to be exported
Returns:
The result of the export
Nr )�selfr r r r �exportE s zSpanExporter.exportN�r! c C s dS )zLShuts down the exporter.
Called when the SDK is shut down.
Nr �r" r r r �shutdownO s zSpanExporter.shutdownr ��timeout_millisr! c C s dS )z�Hint to ensure that the export of any spans the exporter has received
prior to the call to ForceFlush SHOULD be completed as soon as possible, preferably
before returning from this method.
Nr �r" r( r r r � force_flushU s zSpanExporter.force_flush)r ) r r r �__doc__�typing�Sequencer r# r& �int�boolr* r r r r r ; s
r c @ sh e Zd ZdZed�dd�Zdeeje dd�dd�Z
e dd �d
d �Z dd �d
d�Z
deed�dd�ZdS )�SimpleSpanProcessorz�Simple SpanProcessor implementation.
SimpleSpanProcessor is an implementation of `SpanProcessor` that
passes ended spans directly to the configured `SpanExporter`.
��
span_exporterc C s
|| _ d S �Nr1 )r" r2 r r r �__init__c s z SimpleSpanProcessor.__init__N��span�parent_contextr! c C s d S r3 r �r" r6 r7 r r r �on_startf s z SimpleSpanProcessor.on_start�r6 r! c C sX |j jjsd S tttd��}z| j�|f� W n tyJ t �
d� Y n0 t |� d S )NTz Exception while exporting Span.) �context� trace_flags�sampledr r r r2 r# � Exception�logger� exceptionr
)r" r6 �tokenr r r �on_endk s
zSimpleSpanProcessor.on_endr$ c C s | j �� d S r3 )r2 r& r% r r r r& v s z SimpleSpanProcessor.shutdownr r' c C s dS �NTr r) r r r r* y s z SimpleSpanProcessor.force_flush)N)r )r r r r+ r r4 r r, r r r9 r rB r& r. r/ r* r r r r r0 \ s �
� r0 c @ s e Zd ZdZddgZdd� ZdS )�
_FlushRequestz?Represents a request for the BatchSpanProcessor to flush spans.�event� num_spansc C s t �� | _d| _d S )Nr )� threading�EventrE rF r% r r r r4 � s
z_FlushRequest.__init__N)r r r r+ � __slots__r4 r r r r rD ~ s rD c @ s" e Zd ZdZd-eeeeed�dd�Zd.ee j
e dd�dd�Z e
dd �d
d �Zd d
� Zdd� Ze j
e d�dd�Zee j
e d�dd��Zed�dd�Ze j
e d�dd�Zed�dd�Zd d � Zd/eed �d d �Zdd�d!d"�Zed#d$� �Z ed%d&� �Z ed'd(� �Z ed)d*� �Z ed+d,� �Z dS )0�BatchSpanProcessora� Batch span processor implementation.
`BatchSpanProcessor` is an implementation of `SpanProcessor` that
batches ended spans and pushes them to the configured `SpanExporter`.
`BatchSpanProcessor` is configurable with the following environment
variables which correspond to constructor parameters:
- :envvar:`OTEL_BSP_SCHEDULE_DELAY`
- :envvar:`OTEL_BSP_MAX_QUEUE_SIZE`
- :envvar:`OTEL_BSP_MAX_EXPORT_BATCH_SIZE`
- :envvar:`OTEL_BSP_EXPORT_TIMEOUT`
N)r2 �max_queue_size�schedule_delay_millis�max_export_batch_size�export_timeout_millisc C s� |d u rt �� }|d u r t �� }|d u r0t �� }|d u r@t �� }t �|||� || _t�g |�| _ t
j d| j dd�| _
t
�t
�� �| _d | _|| _|| _|| _|| _d| _d| _d g| j | _| j
�� ttd�r�tj | j d� t� � | _ d S )N�OtelBatchSpanProcessorT��name�target�daemonF�register_at_fork)�after_in_child) rJ �_default_max_queue_size� _default_schedule_delay_millis� _default_max_export_batch_size� _default_export_timeout_millis�_validate_argumentsr2 � collections�deque�queuerG �Thread�worker�
worker_thread� Condition�Lock� condition�_flush_requestrL rM rK rN �done�_spans_dropped�
spans_list�start�hasattr�osrT �_at_fork_reinit�getpid�_pid)r" r2 rK rL rM rN r r r r4 � sD ����
�zBatchSpanProcessor.__init__r5 c C s d S r3 r r8 r r r r9 � s zBatchSpanProcessor.on_startr: c C s� | j rt�d� d S |jjjs"d S | jt�� kr<t �
| j � t | j
�| jkrb| jsbt�d� d| _| j
�|� t | j
�| jkr�| j� | j�� W d � n1 s�0 Y d S )Nz Already shutdown, dropping span.z,Queue is full, likely spans will be dropped.T)re r? �warningr; r< r= rm rj rl �_BSP_RESET_ONCEZdo_oncerk �lenr] rK rf �
appendleftrM rc �notify)r" r6 r r r rB � s
zBatchSpanProcessor.on_endc C sF t �t �� �| _| j�� t jd| jdd�| _| j� � t
� � | _ d S )NrO TrP )
rG ra rb rc r] �clearr^ r_ r` rh rj rl rm r% r r r rk � s
�
z"BatchSpanProcessor._at_fork_reinitc C sX | j d }d }| j�s| j�� | jr4W d � �q| �� }t| j�| jk r�|d u r�| j�|� | �� }| js�| j d }| �|� d }W d � q| jr�W d � �qW d � n1 s�0 Y t � }| �
|� t � }|| d }| j d | }| �|� d }q| j� | �� }W d � n1 �s.0 Y | � � | �|� | �|� d S )N� @�@g e��A) rL re rc � _get_and_unset_flush_requestrp r] rM �wait� _notify_flush_request_finishedr �_export� _drain_queue)r" �timeout�
flush_requestrh �end�durationZshutdown_flush_requestr r r r_ � s@
��
.
(
zBatchSpanProcessor.workerr$ c C s$ | j }d| _ |dur t| j�|_|S )ztReturns the current flush request and makes it invisible to the
worker thread for subsequent calls.
N)rd rp r] rF )r" r{ r r r ru s
z/BatchSpanProcessor._get_and_unset_flush_request�r{ c C s | dur| j �� dS )zzNotifies the flush initiator(s) waiting on the given request/event
that the flush operation was finished.
N)rE �setr~ r r r rw * s z1BatchSpanProcessor._notify_flush_request_finishedc C s | j du rt� | _ | j S )a� Either returns the current active flush event or creates a new one.
The flush event will be visible and read by the worker thread before an
export operation starts. Callers of a flush operation may wait on the
returned event to be notified when the flush/export operation was
finished.
This method is not thread-safe, i.e. callers need to take care about
synchronization/locking.
N)rd rD r% r r r � _get_or_create_flush_request4 s
z/BatchSpanProcessor._get_or_create_flush_requestc C s< |s| � � dS |j}| jr8| � � }||8 }|dkrq8qdS )aU Exports spans considering the given flush_request.
In case of a given flush_requests spans are exported in batches until
the number of exported spans reached or exceeded the number of spans in
the flush request.
In no flush_request was given at most max_export_batch_size spans are
exported.
Nr )�
_export_batchrF r] )r" r{ rF Z num_exportedr r r rx C s zBatchSpanProcessor._exportc C s� d}|| j k r.| jr.| j�� | j|< |d7 }qtttd��}z| j�| jd|� � W n t yr t
� d� Y n0 t |� t
|�D ]}d| j|< q�|S )zfExports at most max_export_batch_size spans and returns the number of
exported spans.
r r TNz%Exception while exporting Span batch.)rM r] �poprg r r r r2 r# r> r? r@ r
�range)r" �idxrA �indexr r r r� X s
z BatchSpanProcessor._export_batchc C s | j r| �� q dS )z�Export all elements until queue is empty.
Can only be called from the worker thread context because it invokes
`export` that is not thread safe.
N)r] r� r% r r r ry p s z BatchSpanProcessor._drain_queuer' c C s| |d u r| j }| jr"t�d� dS | j�" | �� }| j�� W d � n1 sP0 Y |j�|d �}|sxt�d� |S )Nz1Already shutdown, ignoring call to force_flush().Trt z&Timeout was exceeded in force_flush().) rN re r? rn rc r� �
notify_allrE rv )r" r( r{ �retr r r r* y s
(
z BatchSpanProcessor.force_flushc C sN d| _ | j� | j�� W d � n1 s,0 Y | j�� | j�� d S rC )re rc r� r` �joinr2 r& r% r r r r& � s
(
zBatchSpanProcessor.shutdownc C s< zt t�tt��W S ty6 t�ttt� t Y S 0 d S r3 ) r. r �getr �_DEFAULT_MAX_QUEUE_SIZE�
ValueErrorr? r@ � _ENV_VAR_INT_VALUE_ERROR_MESSAGEr r r r rV � s �z*BatchSpanProcessor._default_max_queue_sizec C s< zt t�tt��W S ty6 t�ttt� t Y S 0 d S r3 ) r. r r� r � _DEFAULT_SCHEDULE_DELAY_MILLISr� r? r@ r� r r r r rW � s
� �z1BatchSpanProcessor._default_schedule_delay_millisc C s< zt t�tt��W S ty6 t�ttt� t Y S 0 d S r3 ) r. r r� r
� _DEFAULT_MAX_EXPORT_BATCH_SIZEr� r? r@ r� r r r r rX � s �� �z1BatchSpanProcessor._default_max_export_batch_sizec C s< zt t�tt��W S ty6 t�ttt� t Y S 0 d S r3 ) r. r r� r � _DEFAULT_EXPORT_TIMEOUT_MILLISr� r? r@ r� r r r r rY � s
� �z1BatchSpanProcessor._default_export_timeout_millisc C sD | dkrt d��|dkr t d��|dkr0t d��|| kr@t d��d S )Nr z*max_queue_size must be a positive integer.z'schedule_delay_millis must be positive.z1max_export_batch_size must be a positive integer.zCmax_export_batch_size must be less than or equal to max_queue_size.)r� )rK rL rM r r r rZ � s �z&BatchSpanProcessor._validate_arguments)NNNN)N)N)!r r r r+ r r. �floatr4 r r, r r r9 r rB rk r_ rD ru � staticmethodrw r� rx r� ry r/ r* r& rV rW rX rY rZ r r r r rJ � sR �� 3 �
� /� �
rJ c @ sj e Zd ZdZdejdd� fee ej ej
e gef d�dd�Z ej
e ed�d d
�Zdeed �d
d�ZdS )�ConsoleSpanExporterz�Implementation of :class:`SpanExporter` that prints spans to the
console.
This class can be used for diagnostic purposes. It prints the exported
spans to the console STDOUT.
Nc C s | � � t S r3 )�to_jsonr )r6 r r r �<lambda>� s �z ConsoleSpanExporter.<lambda>)� service_name�out� formatterc C s || _ || _|| _d S r3 )r� r� r� )r" r� r� r� r r r r4 � s z ConsoleSpanExporter.__init__r c C s, |D ]}| j �| �|�� q| j �� tjS r3 )r� �writer� �flushr r )r" r r6 r r r r# � s
zConsoleSpanExporter.exportr r' c C s dS rC r r) r r r r* � s z ConsoleSpanExporter.force_flush)r )r r r r+ �sys�stdoutr �strr, �IO�Callabler r4 r- r r# r. r/ r* r r r r r� � s �� r� )-r[ �loggingrj r� rG r, �enumr r r �timer r Zmysql.opentelemetry.contextr r r r
r Z-mysql.opentelemetry.sdk.environment_variablesr r
r r Z mysql.opentelemetry.sdk.tracer r r Z mysql.opentelemetry.util._oncer r� r� r� r� r� � getLoggerr r? r r r0 rD ro rJ r� r r r r �<module> s: �
!"
W