hTUdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZmZddlmZ ddlZddlZdZn #e$rdZYnwxYw d d lmZn #e$rdZYnwxYwd d lmZd d lmZmZd d lm Z m!Z!m"Z"m#Z#m$Z$d dl%m&Z&ej'Z(dZ)dZ*ej+dZ,dZ-eeneefZ.de/d<iZ0de/d<d%dZ1d&dZ2d'dZ3d(d!Z4Gd"dZ5Gd#d$Z6dS))z5Implementing pooling of connections to MySQL servers.) annotationsN) TracebackType)AnyDictNoReturnOptionalTupleTypeUnion)uuid4TF)CMySQLConnection)MySQLConnection) CNX_POOL_ARGSDEFAULT_CONFIGURATION)ErrorInterfaceErrorNotSupportedError PoolErrorProgrammingError)read_option_files @z[^a-zA-Z0-9._:\-*$#]z0MySQL Connector/Python C Extension not availablezUnion[type, Tuple[type, ...]]MYSQL_CNX_CLASSzDict[str, MySQLConnectionPool]_CONNECTION_POOLSkwargsrreturnPooledMySQLConnectionc 6d|vr|dn td i|}|ddurtttt 5|t vrtd i|t |<nQtt |tr1t |j }d|vr|d|krtddddn #1swxYwY t | S#t$rtd|ddwxYw) z!Return a pooled MySQL connection. pool_nameuse_pureFN pool_sizez)Size can not be changed for active pools.z%Failed getting connection from pool '')generate_pool_namegetr ImportError ERROR_NO_CEXTCONNECTION_POOL_LOCKrMySQLConnectionPool isinstancer"rget_connectionAttributeErrorr)rr check_sizes _/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/mysql/connector/pooling.py_get_pooled_connectionr0Ms +f44{:L:V:Vv:V:Vzz*&&+;+C-((( MM - - -+>+H+H+H+H i ( ( ))46I J J M*95?Jf$$ )< )J)J KLLLMMMMMMMMMMMMMMM +::<<<  @I @ @ @   sA0CCCC99D?Union[PooledMySQLConnection, MySQLConnection, CMySQLConnection]c ~|} |d}n#t$rtddwxYw|d=tgd}d}|D]}t||z }|rAt |dkrdnd}d |}td |d |t|d r|dz }|d d |d <|d dks |d d krtd|d t|d tstd|d d|cxkrt |krnntdi} g} t|ddD]U}|d | vr(|g| |d <| |d 4| |d |V| D]} | | } tt | D]} t | dz }t!jd|}| |}|}|||d d t)di|ccS#t*$rYwxYwtd)aReturn a MySQL connection and try to failover if needed. An InterfaceError is raise when no MySQL is available. ValueError is raised when the failover server configuration contains an illegal connection argument. Supported arguments are user, password, host, port, unix_socket and database. ValueError is also raised when the failover argument was not provided. Returns MySQLConnection instance. failoverzfailover argument not providedN) userpasswordhostport unix_socketdatabaser r"priorityrr sz, z Unsupported connection argument z in failover: r:dz9Priority value should be in the range of 0 to 100, got : zDPriority value should be an integer in the range of 0 to 100, got : z\You must either assign no priority to any of the routers or give a priority for every routerc|dS)Nr:r$xs r/z*_get_failover_connection..s :T)keyreversez,Unable to connect to any of the target hostsr$)copyKeyError ValueErrorsetkeyslenjoinhasattrr&rr+intrsortedappendrangerandomrandintpopupdateconnectr)rconfigr3support_cnx_argspriority_countserverdiffarglstserver_directoryserver_priority_listr: failover_list_lastindex new_configs r/_get_failover_connectionrdjs[[]]FE*% EEE9::DEz   N6;;==!!$44  TQ##BC))D//CK3KKcKK  6: & & a N#ZZ C88z *  ! !VJ%7#%=%= . +.. &,c22  3$Z033    >))))CMM)))))    '>'>MMM@@ * %5 5 54:8 VJ/ 0 ' 'z(: ; ; ; ; VJ/ 0 7 7 ? ? ? ?(  (2 s=))**  A}%%)DN1d++E"&&u--FJ   f % % % NN:t , , , ,,,,,,,,,      G H HHs: J J,+J,argscdvrdnd}t|tstd|r,tstddvrtddvrtdd vrtd d vrt d d < t jd d }n2#t j j $rtd d ddwxYwg}|D]D}| |j d|j|j|jdE|dd|Dd <dvr dd<ddvrt%di}t'di|Sd vr t)diS t+fdt,Dr t/diSn#t0$rYnwxYwdd}dvr d=|st4t7t8t4r|s t5|iSt;|iS)aCreate or get a MySQL connection object. In its simpliest form, connect() will open a connection to a MySQL server and return a MySQLConnection object. When any connection pooling arguments are given, for example pool_name or pool_size, a pool is created or a previously one is used to return a PooledMySQLConnection. Returns MySQLConnection or PooledMySQLConnection. dns_srvFz(The value of 'dns-srv' must be a booleanztMySQL host configuration requested DNS SRV. This requires the Python dnspython module. Please refer to documentationr8z.sQz]Qx[L$ArB)rCc0g|]}|d|ddS)r6r7)r6r7r$).0srvs r/ zconnect..s5   ;>S[#f+ 6 6   rBread_default_file option_filesc3 K|]}|vV dSNr$)rmrCrs r/ zconnect..s'66sf}666666rBr!r$)rSr+boolrHAVE_DNSPYTHONrdnsresolverquery exception DNSExceptionrOtargetto_textr7r:rjsortrrUrdanyrr0 NameErrorr&rr'r(r)rerrg srv_recordsr3rnrcr!s ` r/rUrUsT(1F':':fjj###G gt $ $IGHHH+   8  F " " N  V   M     S    26:F6N ,,,VF^UCCKK})    D6&>DDD     C OOJ..d.CCH # !j       AA BBB  BJ   z f$$!'(;!<~ &'''&0000 $$$$$V'11&111 6666 666 6 6 4)33F33 3 4      zz*e,,HV :  -,4m,, ,110000 D +F + ++s(&C/C>+G.. G;:G;strc g}dD];} |t||,#t$rY8wxYw|stdd|S)zGenerate a pool name This function takes keyword arguments, usually the connection arguments for MySQLConnection, and tries to generate a name for a pool. Raises PoolError when no name can be generated. Returns a string. )r6r7r4r9z.Failed generating pool name; specify pool_namer`)rOrrFrrK)rpartsrCs r/r%r%)s E3  LLVC[)) * * * *    D  JHIII 88E??s (1 >>cjeZdZdZddZdd ZddZddZddZe d dZ e d!dZ dS)"raClass holding a MySQL Connection in a pool PooledMySQLConnection is used by MySQLConnectionPool to return an instance holding a MySQL connection. It works like a MySQLConnection except for methods like close() and config(). The close()-method will add the connection back to the pool rather than disconnecting from the MySQL server. Configuring the connection have to be done through the MySQLConnectionPool method set_config(). Using config() on pooled connection will raise a PoolError. poolr*cnx(Union[MySQLConnection, CMySQLConnection]rNonect|tstdt|tstd||_||_dS)zInitialize The pool argument must be an instance of MySQLConnectionPoll. cnx if an instance of MySQLConnection. z$pool should be a MySQLConnectionPoolzcnx should be a MySQLConnectionN)r+r*r-r _cnx_pool_cnx)selfrrs r/__init__zPooledMySQLConnection.__init__Ps[$ 344 I !GHH H#// D !BCC C.2>A rBc|Srsr$rs r/ __enter__zPooledMySQLConnection.__enter___s rBexc_typeType[BaseException] exc_value BaseException tracebackrc.|dSrs)close)rrrrs r/__exit__zPooledMySQLConnection.__exit__bs rBattrrc,t|j|S)z0Calls attributes of the MySQLConnection instance)getattrr)rrs r/ __getattr__z!PooledMySQLConnection.__getattr__jsty$'''rBc |j}|jjr||j|d|_dS#|j|d|_wxYw)aYDo not close, but add connection back to pool The close() method does not close the connection with the MySQL server. The connection is added back to the pool so it can be reused. When the pool is configured to reset the session, the session state will be cleared by re-authenticating the user. N)rr reset_sessionadd_connectionrrs r/rzPooledMySQLConnection.closensz )C~+ $!!### N ) )# . . .DIII N ) )# . . .DI    s 'A #A/rrc  td)z&Configuration is done through the poolzKConfiguration for pooled connections should be done through the pool itself)r)rs r/rVzPooledMySQLConnection.configs    rBrc|jjSz&Return the name of the connection pool)rr rs r/r zPooledMySQLConnection.pool_names~''rBN)rr*rrrrrr)rrrrrrrr)rrrr)rr)rrrrrr) __name__ __module__ __qualname____doc__rrrrr staticmethodrVpropertyr r$rBr/rrAs   B B B B(((($   \ (((X(((rBceZdZdZ dd dZed!dZed"dZed#dZd$dZ d%dZ d&dZ d'dZ d(d)dZ d*dZd"dZdS)+r*z*Class defining a pool of MySQL connectionsNTr"rMr Optional[str]pool_reset_sessionrurrrrc d|_d|_||_||||p t di|i|_tj|j|_ t|_ |r@|j di|d}||jkr(| |dz }||jk$dSdSdS)zInitialize Initialize a MySQL connection pool with a maximum number of connections set to pool_size. The rest of the keywords arguments, kwargs, are configuration arguments for MySQLConnection instances. Nrr r$) _pool_size _pool_name_reset_session_set_pool_size_set_pool_namer% _cnx_configqueueQueue _cnx_queuer _config_version set_configr)rr"r rrcnts r/rzMySQLConnectionPool.__init__s*.)-0 I&&& IE);)E)Ef)E)EFFF+- K ( (  %ww   DO % %f % % %C''##%%%q''''  ('rBrc|jSr)rrs r/r zMySQLConnectionPool.pool_name rBc|jS)z0Return number of connections managed by the pool)rrs r/r"zMySQLConnectionPool.pool_sizerrBc|jS)zReturn whether to reset session)rrs r/rz!MySQLConnectionPool.reset_sessions ""rBc |sdSt5 t}|jdi|||_t |_n%#t $r}td||d}~wwxYw ddddS#1swxYwYdS)aTSet the connection configuration for MySQLConnection instances This method sets the configuration used for creating MySQLConnection instances. See MySQLConnection for valid connection arguments. Raises PoolError when a connection argument is not valid, missing or not supported by MySQLConnection. Nz$Connection configuration not valid: r$)r)rUrVrr rr-r)rrtest_cnxerrs r/rzMySQLConnectionPool.set_configs  F ! W W W"99))&)))#) ',ww$$! W W W Ls L LMMSVV W%  W W W W W W W W W W W W W W W W W Ws4A75AA7 A&A!!A&&A77A;>A;cd|dks |tkrtdt||_dS)aSet the size of the pool This method sets the size of the pool but it will not resize the pool. Raises an AttributeError when the pool_size is not valid. Invalid size is 0, negative or higher than pooling.CNX_POOL_MAXSIZE. rz8Pool size should be higher than 0 and lower or equal to N)CNX_POOL_MAXSIZEr-r)rr"s r/rz"MySQLConnectionPool._set_pool_sizesI >>Y)999 &#&& $rBct|rtd|dt|tkrtd|d||_dS)aSet the name of the pool. This method checks the validity and sets the name of the pool. Raises an AttributeError when pool_name contains illegal characters ([^a-zA-Z0-9._\-*$#]) or is longer than pooling.CNX_POOL_MAXNAMESIZE. z Pool name 'z' contains illegal charactersz ' is too longN)CNX_POOL_NAMEREGEXsearchr-rJCNX_POOL_MAXNAMESIZEr)rr s r/rz"MySQLConnectionPool._set_pool_namesl  $ $Y / / Y !Wy!W!W!WXX X y>>0 0 0 !Gy!G!G!GHH H#rBrrct|tstd |j|ddS#t j$r}td|d}~wwxYw)zPut connection back in the queue This method is putting a connection back in the queue. It will not acquire a lock as the methods using _queue_connection() will have it set. Raises PoolError on errors. 3Connection instance not subclass of MySQLConnectionFblock'Failed adding connection; queue is fullN)r+rrrputrFullrrrs r/_queue_connectionz%MySQLConnectionPool._queue_connections#// SQRR R P O  5  1 1 1 1 1z P P PEFFC O PsAA(A##A(2Optional[Union[MySQLConnection, CMySQLConnection]]ct5|jstd|jrtd|skt di|j} |jr4|jdr'|dkrtdn#t$rYnwxYw|j |_ n$t|tstd||ddddS#1swxYwYdS) aAdd a connection to the pool This method instantiates a MySQLConnection using the configuration passed when initializing the MySQLConnectionPool instance or using the set_config() method. If cnx is a MySQLConnection instance, it will be added to the queue. Raises PoolError when no configuration is set, when no more connection can be added (maximum reached) or when the connection can not be instantiated. z&Connection configuration not availablercompress)rz^Pool reset session is not supported with compression for MySQL server version 5.7.2 or earlierrNr$)r)rrrfullrUrget_server_versionrrFrpool_config_versionr+rrrs r/rz"MySQLConnectionPool.add_connections" ( (# J HIII##%% K IJJJ 11 011 +  ,Z8  2244y@@/)  D+/*>''!#77#M  " "3 ' ' '= ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s7AC:;BC: B$!C:#B$$A C::C>C>rct5 |jd}n'#tj$r}t d|d}~wwxYw|r|j|jkrX|j di|j  | n$#t$r| |wxYw|j|_t||cdddS#1swxYwYdS)awGet a connection from the pool This method returns an PooledMySQLConnection instance which has a reference to the pool that created it, and the next available MySQL connection. When the MySQL connection is not connect, a reconnect is attempted. Raises PoolError on errors. Returns a PooledMySQLConnection instance. Frz)Failed getting connection; pool exhaustedNr$)r)rr&rEmptyr is_connectedrrrVr reconnectrrrrs r/r,z"MySQLConnectionPool.get_connection2sg" 4 4 Vo)))66; V V V KLLRUU V$$&& ?'3+BBB ..T-...MMOOOO%**3///+/*>'(s33' 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4sHC%&C%A AA  9C%BC%!B::C%%C),C)c|t5d}|j}|r} |d}||dz }n8#t j$r|cYcdddSt$rt$rYnwxYw|}|cdddS#1swxYwYdS)zClose all connections This method closes all connections. It returns the number of connections it closed. Used mostly for tests. Returns int. rFrr N) r)rqsizer& disconnectrrrr)rrcnxqrs r/_remove_connectionsz'MySQLConnectionPool._remove_connectionsTsN"  C?D**,,  (((//CNN$$$1HCC{JJ        !D**,, !                  s@B1/AB1B (B16B  B1 B  B11B58B5)rNT) r"rMr rrrurrrrr)rrM)rru)rrrr)r"rMrr)r rrr)rrrrrs)rrrrr)rrrrrrr r"rrrrrrr,rr$rBr/r*r*sA44#'#' @XX###X#WWWW* $ $ $ $ $ $ $ $PPPP$IM-(-(-(-(-(^ 4 4 4 4DrBr*)rrrr)rrrr1)rerrrrr1)rrrr)7r __future__rrrQre threadingtypesrtypingrrrrr r r uuidr dns.exceptionrw dns.resolverrvr'connection_cextr connectionr constantsrrerrorsrrrrr optionfilesrRLockr)rrcompilerr(r__annotations__rr0rdrUr%rr*r$rBr/rs:<;;"""""" DDDDDDDDDDDDDDDDDDNNNNN 1111111(''''';;;;;;;;+*****&y((RZ 788B '/OOoGW5X576666:ZIZIZIZIz_,_,_,_,D0J(J(J(J(J(J(J(J(Z``````````s#AA AAA#"A#