/var/opt/nydus/ops/mysql/connector/opentelemetry/__pycache__
Edit: /var/opt/nydus/ops/mysql/connector/opentelemetry/__pycache__/instrumentation.cpython-312.pyc (29575B)
N ja N U d Z ddlmZ ddlZddlZddlmZmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZmZ e rddlmZmZ dd lmZ d
d lmZ dd lmZmZ dd
l m Z ddl m Z ddl m Z ddl!m"Z" ddl#m$Z$ ddlm)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4 ejj d Z6de7d< d(dZ8d)dZ9d*dZ: d+ d,dZ;d-d Z
G d d e Z? G d! d"e? Z@ G d# d$e? ZA d+ d.d%ZB G d& d' ZCy# e%$ r!Z&ejN jQ d e&dZ&[&ww xY w)/z1MySQL instrumentation supporting mysql-connector. ) annotationsN)ABCabstractmethod) nullcontext)
TYPE_CHECKINGAnyCallable
CollectionDictOptionalUnion )MySQLConnectionAbstractMySQLCursorAbstract)PooledMySQLConnection ) connector)
CNX_POOL_ARGSDEFAULT_CONFIGURATION)logger) VERSION_TEXT)trace)TracerProvider)SpanAttributeszGOpenTelemetry installation not found. You must install the API and SDK. ) CONNECTION_SPAN_NAME DB_SYSTEMDEFAULT_THREAD_IDDEFAULT_THREAD_NAMEFIRST_SUPPORTED_VERSIONNET_SOCK_FAMILYNET_SOCK_HOST_ADDRNET_SOCK_HOST_PORTNET_SOCK_PEER_ADDRNET_SOCK_PEER_PORTOPTION_CNX_SPANOPTION_CNX_TRACERz
^/\*.*?\*/z
re.Patternleading_comment_removerc | r| j r|sy| j t j t j j
| j
| y)z Records an exeception event.N) is_recording
set_statusr Status
StatusCodeERRORrecord_exception)spanexcs ^/opt/nydus/tmp/pip-target-a90h98xg/lib/python/mysql/connector/opentelemetry/instrumentation.pyrecord_exception_eventr3 X sD t((*#OOELL!1!1!7!789# c J | r| j sy| j y)z
Ends span.N)r* end)r0 s r2 end_spanr7 a s t((*HHJr4 c v | r7t | t r't j d| j d S y)z&Parse query to extract operation name. r )
isinstancestrr( subsplit) operations r2 get_operation_namer? i s2 Z 3/&**2y9??A!DD
r4 c |r|j sy|i }| r
| j nd|v}t j t t j |rdndt
|rdndi}|r | r | j n|j dt d |t j < | r | j n|j dt d |t j < t | d r| j r | j j j \ |t < }| j j j# \ |t$ < |t& < |t j |k7 rt||t( < nj| r | j n|j d |t < t | d r8| j r, | j j j# |t$ < |j1 | y# t* $ r }t- j. d
| Y d}~6d}~ww xY w# t* $ r }t- j. d
| Y d}~bd}~ww xY w) 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
N unix_socketip_tcpinprocinetunixhostport_socketz Connection socket is down %s)r* _unix_socketr r
NET_TRANSPORTr! _hostgetr
NET_PEER_NAME_port
NET_PEER_PORThasattrrH sock getpeernamer$ getsocknamer" r# r% Exceptionr warningset_attributes)cnxcnx_span
cnx_kwargsis_tcpattrssock_peer_portsock_errs r2 set_connection_span_attrsr^ q s 8 8002
%(!!
!m:.MF )$$&hh6v E CII*..9Nv9V"W n**+ CII*..9Nv9V"W n**+ 3 "s{{
? KK$$002,-" KK$$002,-,-
556.H 1?E,-
!$C
)F ! 3 "s{{
I,/KK,<,<,H,H,J()
E")
I =xHH
I"
I =xHH
Is1 +A#G* -+H * H3HH H? H::H?c d fd }|S )zAAttach the connection span while executing the connection method.c | j r;| j j r!t j | j d n t 5 | g|i |cddd S # 1 sw Y yxY w)z#Connection span attacher decorator.F end_on_exitN)_spanr* r use_spanr )rW argskwargsmethods r2 wrapperz'with_cnx_span_attached..wrapper sd YY399113 ^^ II5
9D G #///G G Gs A))A2)rW 'MySQLConnectionAbstract're r rf r returnr rg rh s ` r2 with_cnx_span_attachedrm s 0 Nr4 c d fd }|S )z:Create a query span while executing the connection method.c t j dj t j t t j
| j t j t t j t d| j i}| j r| j j r{| j j j j! t" j$ j& t# j( | j j+ g| n t- 5 | g|i |cddd S # 1 sw Y yxY w) Query span creator decorator.z%Creating query span for connection.%sconnection_typenamekindlinks
attributesN)r info__name__r r DB_USER_user THREAD_IDr THREAD_NAMEr get_wrapped_classrc r* _tracerstart_as_current_spanupperr SpanKindCLIENTLinkget_span_contextr )rW re rf query_span_attributesrg s r2 rh z$with_cnx_query_span..wrapper s ;V__M
$ $i " "CII $ $&7 & &(; s 4 4 6 '
YY399113 [[
.
. &&( &&::cii 8 8 :;<, /
:E G #///
G G Gs 5 E E)rW TracedMySQLConnectionre r rf r rj r rk rl s ` r2 with_cnx_query_spanr s 0( Nr4 c d fd }|S )z6Create a query span while executing the cursor method.c t j dj t | j d rt | j d nt | j d }t j t t j |j t j t t j t d| j i}| j j! t# |d xs dt$ j& j( | j* g| 5 | g|i |cddd S # 1 sw Y yxY w) rp z!Creating query span for cursor.%s _connection_cnx cursor_typer z
SQL statementrr N)r rw rx rP _wrappedgetattrr r ry rz r{ r r| r r} r~ r r? r r r _connection_span_link)curre rf
connectionr rg s r2 rh z'with_cursor_query_span..wrapper s 7 I s||]3
CLL- 0 v.
$ $i " "J$4$4 $ $&7 & &(; 3002 '
[[
.
.#DG,? &&,,-, /
#///
s D44D=)r TracedMySQLCursorre r rf r rj r rk rl s ` r2 with_cursor_query_spanr s 04 Nr4 c V e Zd ZdZedd Zd dZd
fd Zd dZd dZ d
dZ
xZ S )BaseMySQLTracerz s/ 07
%+ 16 , , .2
"r4 c : | j j |i |S Instrument method.)r executer s r2 r zTracedMySQLCursor.executeK s %t}}$$d5f55r4 c : | j j |i |S r )r executemanyr s r2 r z TracedMySQLCursor.executemanyP )t}}(($9&99r4 c : | j j |i |S r )r callprocr s r2 r zTracedMySQLCursor.callprocU &t}}%%t6v66r4 N)r z'MySQLCursorAbstract'r z trace.Tracerr
trace.Spanre r rf r rj r ) rx r r r r r r r r rk r4 r2 r r ; sb E
&
$
6 6 : : 7 7r4 r c p e Zd ZdZd"dZd#dZed$d Zed$d Zed$d Z ed$d Z
ed$d Z ed$d Z ed$d
Z
ed$d Zed$d Zed$d
Zed$d Zed$d Zeed%d Zed$d Zed$d Zed$d Zed$d Zed$d Zed$d Zed$d Zed$d Z ed$d Z ed$d Z eed%d Z eed%d Z eed&d Z!e!jD ed'd Z!ed$d Z#ed$d Z$y!)(r zCWrapper class for a `MySQLConnection` or `CMySQLConnection` object.c > || _ | j j }yr r sql_mode)r r _s r2 r z TracedMySQLConnection.__init__^ s 3:
MM " "r4 c z t | j j |i || j | j S )zWraps the object method.)r r r )r r cursorr~ rc r s r2 r z TracedMySQLConnection.cursorf s6 (DMM(($9&9<< JJ
r4 c : | j j |i |S r )r cmd_change_userr s r2 r z%TracedMySQLConnection.cmd_change_usern s -t}},,d=f==r4 c : | j j |i |S r )r commitr s r2 r z TracedMySQLConnection.commits s $t}}##T4V44r4 c : | j j |i |S r )r rollbackr s r2 r z TracedMySQLConnection.rollbackx r r4 c : | j j |i |S r )r cmd_queryr s r2 r z TracedMySQLConnection.cmd_query} 't}}&&777r4 c : | j j |i |S r )r cmd_init_dbr s r2 r z!TracedMySQLConnection.cmd_init_db r r4 c : | j j |i |S r )r cmd_refreshr s r2 r z!TracedMySQLConnection.cmd_refresh r r4 c : | j j |i |S r )r cmd_quitr s r2 r z TracedMySQLConnection.cmd_quit r r4 c : | j j |i |S r )r cmd_shutdownr s r2 r z"TracedMySQLConnection.cmd_shutdown *t}}))4:6::r4 c : | j j |i |S r )r cmd_statisticsr s r2 r z$TracedMySQLConnection.cmd_statistics ,t}}++Tv>>r4 c : | j j |i |S r )r cmd_debugr s r2 r z TracedMySQLConnection.cmd_debug r r4 c : | j j |i |S r )r cmd_pingr s r2 r z TracedMySQLConnection.cmd_ping r r4 c . | j j S r )r databaser s r2 r z TracedMySQLConnection.database }}%%%r4 c : | j j |i |S r )r is_connectedr s r2 r z"TracedMySQLConnection.is_connected r r4 c : | j j |i |S r )r
reset_sessionr s r2 r z#TracedMySQLConnection.reset_session s +t}}**D;F;;r4 c : | j j |i |S r )r pingr s r2 r zTracedMySQLConnection.ping s "t}}!!42622r4 c : | j j |i |S r )r
info_queryr s r2 r z TracedMySQLConnection.info_query s (t}}''888r4 c : | j j |i |S r )r cmd_stmt_preparer s r2 r z&TracedMySQLConnection.cmd_stmt_prepare r r4 c : | j j |i |S r )r cmd_stmt_executer s r2 r z&TracedMySQLConnection.cmd_stmt_execute r r4 c : | j j |i |S r )r cmd_stmt_closer s r2 r z$TracedMySQLConnection.cmd_stmt_close r r4 c : | j j |i |S r )r cmd_stmt_send_long_datar s r2 r z-TracedMySQLConnection.cmd_stmt_send_long_data s 5t}}44dEfEEr4 c : | j j |i |S r )r cmd_stmt_resetr s r2 r z$TracedMySQLConnection.cmd_stmt_reset r r4 c : | j j |i |S r )r cmd_reset_connectionr s r2 r z*TracedMySQLConnection.cmd_reset_connection s 2t}}114B6BBr4 c . | j j S r )r time_zoner s r2 r z TracedMySQLConnection.time_zone s }}&&&r4 c . | j j S r r r s r2 r z TracedMySQLConnection.sql_mode r r4 c . | j j S r r r r s r2 r z TracedMySQLConnection.autocommit s }}'''r4 c & || j _ y)r Nr )r r s r2 r z TracedMySQLConnection.autocommit s $)
r4 c : | j j |i |S r )r set_charset_collationr s r2 r z+TracedMySQLConnection.set_charset_collation s 3t}}22DCFCCr4 c : | j j |i |S r )r start_transactionr s r2 r z'TracedMySQLConnection.start_transaction s /t}}..???r4 N)r ri rj r )re r rf r rj r r r )rj bool)r r rj r )%rx r r r r r r 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 rk r4 r2 r r [ s M#
> > 5 5 7 7 8 8 : : : : 7 7 ; ; = = ? ? 8 8 7 7 & & ; ; < < 3 3 9 9 ? ? ? ? = = F F = = C C ' ' & & ( ( ) ) D D @ @r4 r c T t j d fd }|S )z.Retrurn the instrumented version of `connect`.c t fdt D r t j d | i S t j
dt }|j t t j j t < |t < t j t d 5 }t d| | i }t || t | cddd S # 1 sw Y yxY w)
zWraps the connection object returned by the method `connect`.
Instrumentation for PooledConnections is not supported.
c 3 & K | ] }|v
y wNrk ).0keyrf s r2 z7_instrument_connect..wrapper.. s 6
sf}
s z4Instrumentation for pooled connections not supportedMySQL Connector/Pythoninstrumenting_module_name instrumenting_library_versiontracer_providerrs rt Fra Nr )anyr r rU r
get_tracerr
start_spanr r r r& r' rd r^ r )re rf r rX rW connectr s ` r2 rh z$_instrument_connect..wrapper s 6
6 6 NNQ RD+F+ +!!&>*6 +
#)"3"3%ENN,A,A #4 #
%+ ! ^^F? 3
G8 %dHf = 4*6*C
&c8V <( H
G
Gs 1.C))C2)re r rf r rj z9Union['MySQLConnectionAbstract', 'PooledMySQLConnection']) functoolswraps)r r rh s `` r2 _instrument_connectr s@ __W '' !' B' 'R Nr4 c f e Zd ZU dZdZded< d dZd dZd
dZ d ddZ d
d Z
dd
Z y)MySQLInstrumentorz8MySQL instrumentation supporting mysql-connector-python.NzOptional[MySQLInstrumentor] _instancec | j Ct j | g|i || _ t | j dt j
| j S )zQSinglenton.
Restricts the instantiation to a singular instance.
_original_connect)r! object__new__setattrr r )clsre rf s r2 r% zMySQLInstrumentor.__new__> sI
== "NN3 @ @ @CM CMM#6 8I8I J}} r4 c 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 r s r2 instrumentation_dependenciesz.MySQLInstrumentor.instrumentation_dependenciesJ s --D,EFGGr4 c t j t | d k7 rt j d yt t | d |j
d t _ y)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 r r r rU r rL r rf s r2
instrumentz MySQLInstrumentor.instrumentO sO .A B B NNP Q /D"56 "JJ'8 9
r4 c t |t rt j d |S t |d r t |d s,t j d|j
j |S t j dt | }|j t t j j |_ ||_ t! ||j t | 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.rc r~ z+Instrumentation for class %s not supported.r r r r )r: r r rU rP r rx r r r r r r r rc r~ r^ )r r r r s r2 instrument_connectionz'MySQLInstrumentor.instrument_connection` s " j"7 8 NN= > z7+7:y3Q NN=$$--
!!&>*6 +
",,%ENN,A,A -
$
!*j.>.>?$Z88r4 c t j t | d k( rt j d yt | d t _ y)zUninstrument the library.r# z5MySQL Connector/Python module already uninstrumented.N)r r r r rU r+ s r2 uninstrumentz MySQLInstrumentor.uninstrument s: .A B B NNR S #D*= > r4 c f t |d s,t j d|j j |S t |t st j d |S |j r;|j j r!|j 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.
rc z-Uninstrumentation for class %s not supported.z"Connection already uninstrumented.N) rP r rU r rx r: r rc r* r6 r )r r s r2 uninstrument_connectionz)MySQLInstrumentor.uninstrument_connection s z7+ NN?$$--
*&;< NN? @
0 0 = = ? " #J """r4 )re r rf r rj r )rj zCollection[str])rf r rj r r
)r ri r Optional[trace.TracerProvider]rj ri )r ri rj ri ) rx r r r r! __annotations__r% r) r, r. r0 r2 rk r4 r2 r r 9 sa B-1I*1
H
( ;?(9-(9 8(9
# (9T? #3 # " #r4 r )r0 r r1 zOptional[Exception]rj r )r0 r rj r )r> r; rj r; r
)rW z#Optional['MySQLConnectionAbstract']rX r rY zOptional[Dict[str, Any]]rj r )rg r rj r )r HCallable[..., Union['MySQLConnectionAbstract', 'PooledMySQLConnection']]r r3 rj r5 )Dr
__future__r r reabcr r
contextlibr typingr r r r
r r r
abstractsr r poolingr r9 r constantsr r r versionr
opentelemetryr opentelemetry.sdk.tracer opentelemetry.semconv.tracer ImportErrormissing_dependencies_errerrorsProgrammingErrorr r r r r r! r" r# r$ r% r& r' compiler( r4 r3 r7 r? r^ rm r r r r r r r rk r4 r2 rG s\ : 8 # # " R R R I/ < " $ $6:
'1bjj &? ? ,0R# ,R#R# )R#
R#j
4 @+0c +0\ 7 7@g@O g@X 7;1
U131 N1hv# v#s $
+
+Q #$$s C> >D$ D D$