File: //opt/alt/python311/lib/python3.11/site-packages/elasticsearch/__pycache__/_otel.cpython-311.pyc
�
��;fN
� �� � U d dl mZ d dlZd dlZd dlmZmZmZ erd dlmZ d dl m
Z
e
j d� � Z de
d< n
# e$ r dZ Y nw xY wd d lmZ d
Zd Zd Z G d
� d� � ZdS )� )� annotationsN)�
TYPE_CHECKING� Generator�Mapping)�Literal)�tracezelasticsearch-api�trace.Tracer | None�_tracer)�OpenTelemetrySpan�1OTEL_PYTHON_INSTRUMENTATION_ELASTICSEARCH_ENABLED�>OTEL_PYTHON_INSTRUMENTATION_ELASTICSEARCH_CAPTURE_SEARCH_QUERY�omitc �@ � e Zd Z ddd�Zej dd�� � ZdS )�
OpenTelemetryN�enabled� bool | None�tracerr �
body_strategy� 'Literal["omit", "raw"]' | Nonec �" � |�)t j � t d� � dk }|pt | _ |o| j d u| _ |� || _ d S t j � t t � � | _ | j dv sJ �d S )N�true)r �raw)
�os�environ�get�ENABLED_ENV_VARr
r r r �BODY_STRATEGY_ENV_VAR�DEFAULT_BODY_STRATEGY)�selfr r r s �F/opt/alt/python311/lib/python3.11/site-packages/elasticsearch/_otel.py�__init__zOpenTelemetry.__init__/ s� � � �?��j�n�n�_�f�=�=��G�G� �'��� � �:�4�;�d�#:�� � � $�!.�D� � � �!#�� � �%�'<�"� "�D� � �%� �8�8�8�8�8�8� �method�str� endpoint_id�
str | None�
path_parts�Mapping[str, str]�return�(Generator[OpenTelemetrySpan, None, None]c # �� K � | j r| j �t d � � V � d S |p|}| j � |� � 5 }|� d|� � |� dd� � |�|� d|� � |� � � D ] \ }}|� d|� �|� � � t ||| j �� � V � d d d � � d S # 1 swxY w Y d S )Nzhttp.request.methodz db.system�
elasticsearchz db.operationz db.elasticsearch.path_parts.)r% r )r r r �start_as_current_span�
set_attribute�itemsr )r r# r% r' � span_name� otel_span�key�values r �spanzOpenTelemetry.spanC sl � � � � �|� �t�{� 2�#�D�)�)� )� )� )� �F� �)�6� �
�[�
.�
.�y�
9�
9� �Y� � #� #�$9�6� B� B� B� � #� #�K� � A� A� A��&��'�'�� �D�D�D� (� .� .� 0� 0�
U�
U�
��U��'�'�(L�s�(L�(L�e�T�T�T�T�#�� '� "� 0�� � �
�
�
� � � � � � � � � � � � ���� � � � � � s �BC � C$�'C$)NNN)r r r r r r )r# r$ r% r& r' r( r) r* )�__name__�
__module__� __qualname__r! �
contextlib�contextmanagerr4 � r" r r r . s\ � � � � � � $�&*�9=� 9� 9� 9� 9� 9�( � �� � � � �� � r" r )�
__future__r r8 r �typingr r r r �
opentelemetryr �
get_tracerr
�__annotations__�ModuleNotFoundError�elastic_transportr r r r r r: r" r �<module>rB s# ��$ #� "� "� "� "� "� "� � � � � � � � � 4� 4� 4� 4� 4� 4� 4� 4� 4� 4�� � � � � � � ��#�#�#�#�#�#�#3�5�#3�4G�#H�#H�G�H�H�H�H�H��� � � ��G�G�G����� 0� /� /� /� /� /� F��
Y� � � �.� .� .� .� .� .� .� .� .� .s �? �A �A