?????????? ????????? - ??????????????? - /home/agenciai/.local/lib/python3.9/site-packages/mysql/connector/opentelemetry/__pycache__/instrumentation.cpython-39.pyc
???????
a ��Jh�a � @ s, 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$ W n4 e%�y Z& zej'�(d�e&�W Y dZ&[&n dZ&[&0 0 ddlm)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4 e�5d�Z6de7d< dddd�d d �Z8ddd �d d �Z9d!d!d"�d#d$�Z:d?d%dd&dd'�d(d)�Z;d*d*d+�d,d-�Z<d*d*d+�d.d/�Z=d*d*d+�d0d1�Z>G d2d3� d3e�Z?G d4d5� d5e?�Z@G d6d7� d7e?�ZAd@d8d9d8d:�d;d<�ZBG d=d>� d>�ZCdS )Az1MySQL 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)�SpanAttributeszGOpenTelemetry installation not found. You must install the API and SDK.� ) �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/ �b/home/agenciai/.local/lib/python3.9/site-packages/mysql/connector/opentelemetry/instrumentation.py�record_exception_eventX s r1 )r* r, c C s | r | � � sdS | �� dS )z Ends span.N)r- �end)r* r/ r/ r0 �end_spana 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_namei 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_attrsq 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� - s zBaseMySQLTracer.__enter__�rN rO r, c O s | j j|i |�� dS )r� N)rq �__exit__�ry rN rO r/ r/ r0 r� 2 s zBaseMySQLTracer.__exit__c C s | j jjS )z Gets the wrapped class name.)rq r� r_ rx r/ r/ r0 rd 6 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 > s �zTracedMySQLCursor.__init__r r� c O s | j j|i |��S ��Instrument method.)rq �executer� r/ r/ r0 r� K s zTracedMySQLCursor.executec O s | j j|i |��S r� )rq � executemanyr� r/ r/ r0 r� P s z TracedMySQLCursor.executemanyc O s | j j|i |��S r� )rq �callprocr� r/ r/ r0 r� U s zTracedMySQLCursor.callprocN) r_ r� r� r� rz ru r� r� r� r/ r/ r/ r0 rn ; 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#��Zeed$d%�d&d'���Zedddd �d(d)��Zedddd �d*d+��Zedddd �d,d-��Zedddd �d.d/��Zedddd �d0d1��Zedddd �d2d3��Zedddd �d4d5��Zedddd �d6d7��Z edddd �d8d9��Z edddd �d:d;��Z eed$d%�d<d=���Z eed$d%�d>d?���Z eed@d%�dAdB���Z!e!j"ed@ddC�dDdB���Z!edddd �dEdF��Z#edddd �dGdH��Z$dIS )JrW 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 ^ 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� f s �z TracedMySQLConnection.cursorc O s | j j|i |��S r� )rq �cmd_change_userr� r/ r/ r0 r� n s z%TracedMySQLConnection.cmd_change_userc O s | j j|i |��S r� )rq �commitr� r/ r/ r0 r� s s z TracedMySQLConnection.commitc O s | j j|i |��S r� )rq �rollbackr� r/ r/ r0 r� x 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 rw c C s | j jS r� )rq �databaserx 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 C s | j jS r� )rq � time_zonerx r/ r/ r0 r� � s z TracedMySQLConnection.time_zonec C s | j jS r� r� rx r/ r/ r0 r� � s z TracedMySQLConnection.sql_mode�boolc C s | j jS r� �rq r rx 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 [ 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� = 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_dependenciesI 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 � instrumentN 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_connection_ 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� 8 s �*r� )N)N)Dr� � __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 � ImportErrorZmissing_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> sP $ ��8 �U . - �3
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????