File: //var/opt/nydus/ops/mysql/connector/opentelemetry/__pycache__/instrumentation.cpython-39.pyc
a
�,�hd �
@ sf U d Z ddlmZ ddlZddlZddlmZmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZmZ e r�ddlmZmZ dd lmZ d
d lmZ dd lmZmZ dd
l m Z ddl m Z z,ddl m Z ddl!m"Z" ddl#m$Z$ dZ%W nj e&�yL z ddl'm Z ddl(m$Z$ dZ%W n4 e&�yF Z) zej*�+d�e)�W Y dZ)[)n
dZ)[)0 0 Y n0 ddlm,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 e�8d�Z9de:d< ddd d �d d �Z;dd d �d!d"�Z<d#d#d$�d%d&�Z=dAd'dd(d d)�d*d+�Z>d,d,d-�d.d/�Z?d,d,d-�d0d1�Z@d,d,d-�d2d3�ZAG d4d5� d5e�ZBG d6d7� d7eB�ZCG d8d9� d9eB�ZDdBd:d;d:d<�d=d>�ZEG d?d@� d@�ZFdS )Cz1MySQL instrumentation supporting mysql-connector.� )� annotationsN)�ABC�abstractmethod)� nullcontext)�
TYPE_CHECKING�Any�Callable�
Collection�Dict�Optional�Union� )�MySQLConnectionAbstract�MySQLCursorAbstract)�PooledMySQLConnection� )� connector)�
CNX_POOL_ARGS�DEFAULT_CONFIGURATION)�logger)� VERSION_TEXT)�trace)�TracerProvider)�SpanAttributesTFz�Bundled installation has missing dependencies. Please use `pip install mysql-connector-python[opentelemetry]`, or for an editable install use `pip install -e '.[opentelemetry]'`, to install the dependencies required by the bundled opentelemetry package.� ) �CONNECTION_SPAN_NAME� DB_SYSTEM�DEFAULT_THREAD_ID�DEFAULT_THREAD_NAME�FIRST_SUPPORTED_VERSION�NET_SOCK_FAMILY�NET_SOCK_HOST_ADDR�NET_SOCK_HOST_PORT�NET_SOCK_PEER_ADDR�NET_SOCK_PEER_PORT�OPTION_CNX_SPAN�OPTION_CNX_TRACERz
^/\*.*?\*/z
re.Pattern�leading_comment_remover�
trace.SpanzOptional[Exception]�None)�span�exc�returnc C s6 | r| � � r|sdS | �t�tjj�� | �|� dS )z Records an exeception event.N)� is_recordingZ
set_statusr ZStatusZ
StatusCode�ERRORZrecord_exception)r* r+ � r/ �`/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/connector/opentelemetry/instrumentation.py�record_exception_eventd s r1 )r* r, c C s | r | � � sdS | �� dS )z
Ends span.N)r- �end)r* r/ r/ r0 �end_spanm s r3 �str)� operationr, c C s& | r"t | t�r"t�d| ��� d S dS )z&Parse query to extract operation name.� r )�
isinstancer4 r' �sub�split)r5 r/ r/ r0 �get_operation_nameu s r: z#Optional['MySQLConnectionAbstract']zOptional[Dict[str, Any]])�cnx�cnx_span�
cnx_kwargsr, c
C s� |r |� � sdS |du r i }| r(| j nd|v}tjttj|rBdndt|rNdndi}|�r,| rd| jn|�dtd �|tj < | r�| j
n|�dtd �|tj < t | d ��r�| j
�r�z0| j
j�� \|t< }| j
j�� \|t< |t< W n2 t�y } zt�d
|� W Y d}~n"d}~0 0 |tj |k�r�||t< nv| �r8| jn|�d�|t< t | d ��r�| j
�r�z| j
j�� |t< W n2 t�y� } zt�d
|� W Y d}~n
d}~0 0 |�|� dS ) a� Defines connection span attributes. If `cnx` is None then we use `cnx_kwargs`
to get basic net information. Basic net attributes are defined such as:
* DB_SYSTEM
* NET_TRANSPORT
* NET_SOCK_FAMILY
Socket-level attributes [*] are also defined [**].
[*]: Socket-level attributes identify peer and host that are directly connected to
each other. Since instrumentations may have limited knowledge on network
information, instrumentations SHOULD populate such attributes to the best of
their knowledge when populate them at all.
[**]: `CMySQLConnection` connections have no access to socket-level
details so socket-level attributes aren't included. `MySQLConnection`
connections, on the other hand, do include socket-level attributes.
References:
[1]: https://github.com/open-telemetry/opentelemetry-specification/blob/main/ specification/trace/semantic_conventions/span-general.md
NZ unix_socketZip_tcpZinprocZinet�unix�host�port�_socketz Connection socket is down %s)r- Z _unix_socketr r Z
NET_TRANSPORTr Z_host�getr Z
NET_PEER_NAMEZ_portZ
NET_PEER_PORT�hasattrrA �sock� getpeernamer# � getsocknamer! r" � Exceptionr �warningr$ Zset_attributes)r; r< r= Zis_tcp�attrsZsock_peer_portZsock_errr/ r/ r0 �set_connection_span_attrs} sJ ���
�
��"
�"rJ r )�methodr, c s ddddd�� fdd� }|S )zAAttach the connection span while executing the connection method.�'MySQLConnectionAbstract'r �r; �args�kwargsr, c s` | j r | j �� r tj| j dd�nt� �& � | g|�R i |��W d � S 1 sR0 Y dS )z#Connection span attacher decorator.F�Z end_on_exitN)�_spanr- r �use_spanr )r; rN rO �rK r/ r0 �wrapper� s ����z'with_cnx_span_attached.<locals>.wrapperr/ �rK rT r/ rS r0 �with_cnx_span_attached� s rV c s ddddd�� fdd� }|S )z:Create a query span while executing the connection method.�TracedMySQLConnectionr rM c s� t �d� j� tjttj| jtjttj t
d| � � i}| j rp| j �
� rp| jj� j�� tjjt�| j �� �g|d�nt� �& � | g|�R i |��W d � S 1 s�0 Y dS )� Query span creator decorator.z%Creating query span for connection.%sZconnection_type��name�kind�links�
attributesN)r �info�__name__r r �DB_USER�_user� THREAD_IDr � THREAD_NAMEr �get_wrapped_classrQ r- �_tracer�start_as_current_span�upperr �SpanKind�CLIENT�Link�get_span_contextr )r; rN rO �query_span_attributesrS r/ r0 rT � s& �
����z$with_cnx_query_span.<locals>.wrapperr/ rU r/ rS r0 �with_cnx_query_span� s rm c s ddddd�� fdd� }|S )z6Create a query span while executing the cursor method.�TracedMySQLCursorr )�currN rO r, c s� t �d� j� t| jd�r&t| jd�n
t| jd�}tjttj|j tj
t tj t
d| �� i}| jjt|d �pldtjj| jg|d��& � | g|�R i |��W d � S 1 s�0 Y dS ) rX z!Creating query span for cursor.%s� _connectionZ_cnxZ cursor_typer z
SQL statementrY N)r r^ r_ rC �_wrapped�getattrr r r` ra rb r rc r rd re rf r: r rh ri �_connection_span_link)ro rN rO �
connectionrl rS r/ r0 rT � s$
�
���z'with_cursor_query_span.<locals>.wrapperr/ rU r/ rS r0 �with_cursor_query_span� s ru c sz e Zd ZdZedd�dd��Zddd�d d
�Zdddd �� fd d
� Zdd�dd�Zdddd�dd�Z dd�dd�Z
� Z S )�BaseMySQLTracerz<Base class that provides basic object wrapper functionality.r) �r, c C s dS )z"Must be implemented by subclasses.Nr/ ��selfr/ r/ r0 �__init__ s zBaseMySQLTracer.__init__r4 r )�attrr, c C s || j v rt| |�S t| j|�S )z�Gets an attribute.
Attributes defined in the wrapper object have higher precedence
than those wrapped object equivalent. Attributes not found in
the wrapper are then searched in the wrapped object.
)�__dict__rr rq )ry r{ r/ r/ r0 � __getattr__ s
zBaseMySQLTracer.__getattr__)rZ �valuer, c sN d| j vr|| j d< d S || j v s*|dkr<t� �||� d S | j�||� d S )Nrq �
autocommit)r| �super� __setattr__rq )ry rZ r~ �� __class__r/ r0 r� - s
zBaseMySQLTracer.__setattr__c C s | j �� | S )�
Magic method.)rq � __enter__rx r/ r/ r0 r� 9 s
zBaseMySQLTracer.__enter__�rN rO r, c O s | j j|i |�� dS )r� N)rq �__exit__�ry rN rO r/ r/ r0 r� > s zBaseMySQLTracer.__exit__c C s
| j jjS )z Gets the wrapped class name.)rq r� r_ rx r/ r/ r0 rd B s z!BaseMySQLTracer.get_wrapped_class) r_ �
__module__� __qualname__�__doc__r rz r} r� r� r� rd �
__classcell__r/ r/ r� r0 rv s
rv c @ sd e Zd ZdZdddd�dd�Zedddd �d
d ��Zedddd �d d
��Zedddd �dd��ZdS )rn z;Wrapper class for a `MySQLCursor` or `CMySQLCursor` object.z'MySQLCursorAbstract'z trace.Tracerr( ��wrapped�tracer�connection_spanc C s || _ || _t�|�� �| _dS �z Constructor.N)rq re r rj rk rs )ry r� r� r� r/ r/ r0 rz J s
�zTracedMySQLCursor.__init__r r� c O s | j j|i |��S ��Instrument method.)rq �executer� r/ r/ r0 r� W s zTracedMySQLCursor.executec O s | j j|i |��S r� )rq � executemanyr� r/ r/ r0 r� \ s z TracedMySQLCursor.executemanyc O s | j j|i |��S r� )rq �callprocr� r/ r/ r0 r� a s zTracedMySQLCursor.callprocN) r_ r� r� r� rz ru r� r� r� r/ r/ r/ r0 rn G s
rn c @ s� e Zd ZdZddd�dd�Zdddd �d
d �Zedddd �d d
��Zedddd �dd��Zedddd �dd��Z edddd �dd��Z
edddd �dd��Z edddd �dd��Z edddd �dd��Z
edddd �dd��Zedddd �d d ��Zedddd �d d ��Zedddd �d d!��Zedddd �d"d#��Zeeddd$d �d%d&���Zedddd �d'd(��Zedddd �d)d*��Zedddd �d+d,��Zedddd �d-d.��Zedddd �d/d0��Zedddd �d1d2��Zedddd �d3d4��Zedddd �d5d6��Z edddd �d7d8��Z edddd �d9d:��Z eeddd$d �d;d<���Z eeddd$d �d=d>���Z eeddd?d �d@dA���Z!e!j"ed?ddB�dCdA���Z!edddd �dDdE��Z#edddd �dFdG��Z$dHS )IrW zCWrapper class for a `MySQLConnection` or `CMySQLConnection` object.rL r) )r� r, c C s || _ | j j}dS r� �rq �sql_mode)ry r� �_r/ r/ r0 rz j s z TracedMySQLConnection.__init__r rn r� c O s t | jj|i |��| j| jd�S )zWraps the object method.r� )rn rq �cursorre rQ r� r/ r/ r0 r� r s
�z TracedMySQLConnection.cursorc O s | j j|i |��S r� )rq �cmd_change_userr� r/ r/ r0 r� z s z%TracedMySQLConnection.cmd_change_userc O s | j j|i |��S r� )rq �commitr� r/ r/ r0 r� s z TracedMySQLConnection.commitc O s | j j|i |��S r� )rq �rollbackr� r/ r/ r0 r� � s z TracedMySQLConnection.rollbackc O s | j j|i |��S r� )rq � cmd_queryr� r/ r/ r0 r� � s z TracedMySQLConnection.cmd_queryc O s | j j|i |��S r� )rq � cmd_init_dbr� r/ r/ r0 r� � s z!TracedMySQLConnection.cmd_init_dbc O s | j j|i |��S r� )rq � cmd_refreshr� r/ r/ r0 r� � s z!TracedMySQLConnection.cmd_refreshc O s | j j|i |��S r� )rq �cmd_quitr� r/ r/ r0 r� � s z TracedMySQLConnection.cmd_quitc O s | j j|i |��S r� )rq � cmd_shutdownr� r/ r/ r0 r� � s z"TracedMySQLConnection.cmd_shutdownc O s | j j|i |��S r� )rq �cmd_statisticsr� r/ r/ r0 r� � s z$TracedMySQLConnection.cmd_statisticsc O s | j j|i |��S r� )rq �cmd_process_killr� r/ r/ r0 r� � s z&TracedMySQLConnection.cmd_process_killc O s | j j|i |��S r� )rq � cmd_debugr� r/ r/ r0 r� � s z TracedMySQLConnection.cmd_debugc O s | j j|i |��S r� )rq �cmd_pingr� r/ r/ r0 r� � s z TracedMySQLConnection.cmd_pingr4 c O s | j jS r� )rq �databaser� r/ r/ r0 r� � s z TracedMySQLConnection.databasec O s | j j|i |��S r� )rq � is_connectedr� r/ r/ r0 r� � s z"TracedMySQLConnection.is_connectedc O s | j j|i |��S r� )rq �
reset_sessionr� r/ r/ r0 r� � s z#TracedMySQLConnection.reset_sessionc O s | j j|i |��S r� )rq �pingr� r/ r/ r0 r� � s zTracedMySQLConnection.pingc O s | j j|i |��S r� )rq �
info_queryr� r/ r/ r0 r� � s z TracedMySQLConnection.info_queryc O s | j j|i |��S r� )rq �cmd_stmt_preparer� r/ r/ r0 r� � s z&TracedMySQLConnection.cmd_stmt_preparec O s | j j|i |��S r� )rq �cmd_stmt_executer� r/ r/ r0 r� � s z&TracedMySQLConnection.cmd_stmt_executec O s | j j|i |��S r� )rq �cmd_stmt_closer� r/ r/ r0 r� � s z$TracedMySQLConnection.cmd_stmt_closec O s | j j|i |��S r� )rq �cmd_stmt_send_long_datar� r/ r/ r0 r� � s z-TracedMySQLConnection.cmd_stmt_send_long_datac O s | j j|i |��S r� )rq �cmd_stmt_resetr� r/ r/ r0 r� � s z$TracedMySQLConnection.cmd_stmt_resetc O s | j j|i |��S r� )rq �cmd_reset_connectionr� r/ r/ r0 r� � s z*TracedMySQLConnection.cmd_reset_connectionc O s | j jS r� )rq � time_zoner� r/ r/ r0 r� � s z TracedMySQLConnection.time_zonec O s | j jS r� r� r� r/ r/ r0 r� � s z TracedMySQLConnection.sql_mode�boolc O s | j jS r� �rq r r� r/ r/ r0 r � s z TracedMySQLConnection.autocommit)r~ r, c C s || j _dS )r� Nr� )ry r~ r/ r/ r0 r s c O s | j j|i |��S r� )rq �set_charset_collationr� r/ r/ r0 r� s z+TracedMySQLConnection.set_charset_collationc O s | j j|i |��S r� )rq �start_transactionr� r/ r/ r0 r� s z'TracedMySQLConnection.start_transactionN)%r_ r� r� r� rz r� rm r� r� r� r� r� r� r� r� r� r� r� r� �propertyr� r� r� r� r� r� r� r� r� r� r� r� r� r �setterr� r� r/ r/ r/ r0 rW g s� rW zHCallable[..., Union['MySQLConnectionAbstract', 'PooledMySQLConnection']]� Optional[trace.TracerProvider])�connect�tracer_providerr, c s& t �� �dddd�� �fdd� �}|S )z.Retrurn the instrumented version of `connect`.r z9Union['MySQLConnectionAbstract', 'PooledMySQLConnection']r� c s� t � fdd�tD ��r.t�d� �| i � ��S tjdt�d�}|jttj j
d�� t < |� t < tj
� t dd��@}td |� � �| i � ��}t||� � t|d
�W d � S 1 s�0 Y d S ) z�Wraps the connection object returned by the method `connect`.
Instrumentation for PooledConnections is not supported.
c 3 s | ]}|� v V qd S )Nr/ )�.0�key�rO r/ r0 � <genexpr> � z7_instrument_connect.<locals>.wrapper.<locals>.<genexpr>z4Instrumentation for pooled connections not supported�MySQL Connector/Python�Zinstrumenting_module_nameZ instrumenting_library_versionr� �rZ r[ FrP N�r� )�anyr r rH r �
get_tracerr �
start_spanr rh ri r% r& rR rJ rW )rN rO r� r< r; �r� r� r� r0 rT s&
��
�z$_instrument_connect.<locals>.wrapper)� functools�wraps)r� r� rT r/ r� r0 �_instrument_connect s (r� c @ s� e Zd ZU dZdZded< ddd d�dd�Zd d
�d d �Zdd
d�dd�Zddddd�dd�Z dd
d�dd�Z
ddd�dd�Z dS ) �MySQLInstrumentorz8MySQL instrumentation supporting mysql-connector-python.NzOptional[MySQLInstrumentor]� _instancer r� c O s: | j du r4tj| g|�R i |��| _ t| j dtj� | j S )zQSinglenton.
Restricts the instantiation to a singular instance.
N�_original_connect)r� �object�__new__�setattrr r� )�clsrN rO r/ r/ r0 r� I s
zMySQLInstrumentor.__new__zCollection[str]rw c C s dt � �gS )zoReturn a list of python packages with versions
that the will be instrumented (e.g., versions >= 8.1.0).zmysql-connector-python >= )r rx r/ r/ r0 � instrumentation_dependenciesU s z.MySQLInstrumentor.instrumentation_dependenciesr) )rO r, c K s< t jt| d�kr t�d� dS tt| d�|�d�d�t _dS )a Instrument the library.
Args:
trace_module: reference to the 'trace' module from opentelemetry.
tracer_provider (optional): TracerProvider instance.
NOTE: Instrumentation for pooled connections not supported.
r� z3MySQL Connector/Python module already instrumented.Nr� r� )r r� rr r rH r� rB �ry rO r/ r/ r0 �
instrumentZ s
�z MySQLInstrumentor.instrumentrL r� )rt r� r, c C s� t |t�rt�d� |S t|d�r,t|d�s@t�d|jj� |S tjdt |d�}|j
t tj j
d�|_||_t||j� t|d�S ) a� Enable instrumentation in a MySQL connection.
Args:
connection: uninstrumented connection instance.
trace_module: reference to the 'trace' module from opentelemetry.
tracer_provider (optional): TracerProvider instance.
Returns:
connection: instrumented connection instace.
NOTE: Instrumentation for pooled connections not supported.
z Connection already instrumented.rQ re z+Instrumentation for class %s not supported.r� r� r� r� )r7 rW r rH rC r� r_ r r� r r� r rh ri rQ re rJ )ry rt r� r� r/ r/ r0 �instrument_connectionk s(
��� z'MySQLInstrumentor.instrument_connectionc K s. t jt| d�kr t�d� dS t| d�t _dS )zUninstrument the library.r� z5MySQL Connector/Python module already uninstrumented.N)r r� rr r rH r� r/ r/ r0 � uninstrument� s
z MySQLInstrumentor.uninstrument)rt r, c C s\ t |d�s t�d|jj� |S t|t�s6t�d� |S |jrV|j�� rV|j� � d|_|j
S )a Disable instrumentation in a MySQL connection.
Args:
connection: instrumented connection instance.
Returns:
connection: uninstrumented connection instace.
NOTE: Instrumentation for pooled connections not supported.
rQ z-Uninstrumentation for class %s not supported.z"Connection already uninstrumented.N) rC r rH r� r_ r7 rW rQ r- r2 rq )ry rt r/ r/ r0 �uninstrument_connection� s
�
z)MySQLInstrumentor.uninstrument_connection)N) r_ r� r� r� r� �__annotations__r� r� r� r� r� r� r/ r/ r/ r0 r� D s
�*r� )N)N)Gr� �
__future__r r� �re�abcr r �
contextlibr �typingr r r r r
r r Z abstractsr r Zpoolingr r6 r � constantsr r r �versionr Z
opentelemetryr Zopentelemetry.sdk.tracer Zopentelemetry.semconv.tracer ZOTEL_SYSTEM_AVAILABLE� ImportErrorZmysql.opentelemetryZ!mysql.opentelemetry.semconv.traceZmissing_dependencies_err�errorsZProgrammingErrorr r r r r r r! r" r# r$ r% r& �compiler' r� r1 r3 r: rJ rV rm ru rv rn rW r� r� r/ r/ r/ r0 �<module> s\ $ �� 8 �U
. - �3