U
|
£ý°dT ã @ s¸ U d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl m
|
Z
|
m Z m Z m Z mZmZmZ ddlmZ zddlZddlZW n ek
|
r dZY nX dZzd d
|
lmZ W n ek
|
rÈ dZY nX d d lmZ d d l m Z m Z d d l m Z m!Z!m"Z"m#Z#m$Z$ d dl%m&Z& e '¡ Z(dZ)dZ*e +d¡Z,dZ-edkr:eneefZ.de/d< i Z0de/d< dddddZ1dd dd d Z2ddd d d d!Z3dd"dd#d$Z4G d%d dZ5G d&d' d'Z6dS )(z5Implementing pooling of connections to MySQL servers.é )Ú annotationsN)Ú TracebackType)ÚAnyÚDictÚNoReturnÚOptionalÚTupleÚTypeÚUnion)Úuuid4FTé )ÚCMySQLConnection)ÚMySQLConnection)Ú CNX_POOL_ARGSÚDEFAULT_CONFIGURATION)ÚErrorÚInterfaceErrorÚNotSupportedErrorÚ PoolErrorÚProgrammingError)Úread_option_filesé é@ z[^a-zA-Z0-9._:\-*$#]z0MySQL Connector/Python C Extension not availablez Union[type, Tuple[type, ...]]ÚMYSQL_CNX_CLASSz Dict[str, MySQLConnectionPool]Ú_CONNECTION_POOLSr ÚPooledMySQLConnection©ÚkwargsÚreturnc K sÎ d| kr| d nt f | }| d¡dkr8tdkr8tttR |tkrVtf | t|< n4tt| trt| j }d| kr| d |krt
|
dW 5 Q R X zt| ¡ W S t k
|
rÈ t d| ddY nX dS ) z!Return a pooled MySQL connection.Ú pool_nameÚuse_pureFNÚ pool_sizez)Size can not be changed for active pools.z%Failed getting connection from pool 'ú')Úgenerate_pool_nameÚgetr Ú ImportErrorÚ ERROR_NO_CEXTÚCONNECTION_POOL_LOCKr ÚMySQLConnectionPoolÚ
|
isinstancer! r Úget_connectionÚAttributeErrorr )r r Z
|
check_size© r, úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\mysql/connector/pooling.pyÚ_get_pooled_connectionM s& ÿ
|
|
ÿþr. z?Union[PooledMySQLConnection, MySQLConnection, CMySQLConnection]c
|
K s< | ¡ }z |d }W n tk
|
r2 tddY nX |d= tdddddd d
|
d d g }d }|D ]¶}t| ¡ | }|r¦t|dkrdnd}d |¡}td| d| t|d r¸|d7 }| d d¡|d < |d d k sà|d dkròt d|d t
|
|d t s\t d|d q\d | k r0t|k r<n nt di } g }
|
t |dd ddD ]D}|d | kr|g| |d < |
|
|d ¡ n| |d |¡ qV|
|
D ]} | | } tt| D ]r} t| d }t d |¡}| |¡}| ¡ }| |¡ | d d¡ ztf |W S tk
|
r& Y nX q¸q t d dS ) aª Return 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.
|
Úfailoverz failover argument not providedNÚuserÚpasswordÚhostÚportÚ unix_socketÚdatabaser r! Úpriorityr r ÚsÚ z, z Unsupported connection argument z in failover: éd z9Priority 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 S s | d S )Nr6 r, ©Úxr, r, r- Ú<lambda>® ó z*_get_failover_connection.<locals>.<lambda>T)ÚkeyÚreversez,Unable to connect to any of the target hosts)ÚcopyÚKeyErrorÚ
|
ValueErrorÚsetÚkeysÚlenÚjoinÚhasattrr$ r r) Úintr ÚsortedÚappendÚrangeÚrandomÚrandintÚpopÚupdateÚconnectr )r Úconfigr/ Zsupport_cnx_argsZpriority_countÚserverZdiffÚargÚlstZserver_directoryZserver_priority_listr6 Z failover_listÚ_ÚlastÚindexÚ
|
new_configr, r, r- Ú_get_failover_connectionj s~ ÷ÿ
|
ÿ
|
ÿ ÿ ÿ
|
|
rY )Úargsr r c sþ d kr d¡nd}t|ts(td|rts:tdd krJtdd krZtdd krjtd
|
d kr~td d < ztj d d ¡}W n. tj j
|
k
|
r td d ddY nX g }|D ](}| |j j dd|j|j|jd¡ qÌ|jdd d dd |D d < d kr8 d d< d¡ d krVtf }tf |S d krjtf S z(t fddtD rtf W S W n tk
|
r¨ Y nX d d¡}d krÞ d = |sÞtdkrÞtt trô|sôt| S t | S ) a Create 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 documentationr4 z<Using Unix domain sockets with DNS SRV lookup is not allowedr3 z;Specifying a port number with DNS SRV lookup is not allowedr/ zASpecifying multiple hostnames with DNS SRV look up is not allowedr2 ZSRVz Unable to locate any hosts for 'r" NT)Zomit_final_dot)r2 r3 r6 Úweightc S s | d | d fS )Nr6 r\ r, r: r, r, r- r< r= zconnect.<locals>.<lambda>)r> c S s g | ]}|d |d dqS )r2 r3 )r2 r3 r, )Ú.0Úsrvr, r, r- Ú
|
<listcomp> s zconnect.<locals>.<listcomp>Zread_default_fileZ option_filesc 3 s | ]}| kV qd S ©Nr, )r] r> ©r r, r- Ú <genexpr> s zconnect.<locals>.<genexpr>r ) rN r) Úboolr ÚHAVE_DNSPYTHONr ÚdnsÚresolverÚqueryÚ exceptionZ DNSExceptionrJ ÚtargetZto_textr3 r6 r\ Úsortr rP rY Úanyr r. Ú NameErrorr$ r r% r& r )rZ r r[ Z srv_recordsr/ r^ rX r r, ra r- rP Ç s
|
ÿÿÿÿ ÿþ üÿ ÿ
|
|
|
|
|
|
|
|
|
|
rP Ústrc K sP g }dD ]0}z| t| | ¡ W q tk
|
r6 Y qX q|sFtdd |¡S )zøGenerate 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.
|
)r2 r3 r0 r5 z.Failed generating pool name; specify pool_namerU )rJ rm rA r rF )r Úpartsr> r, r, r- r# ) s r# c @ s e Zd ZdZddddddZd dd d
|
Zd d d ddddZdddddZddddZe dddddZ
|
e ddd d Z d S ) r a Class 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.
|
r( ú(Union[MySQLConnection, CMySQLConnection]ÚNone)ÚpoolÚcnxr c C s4 t |tstdt |ts$td|| _|| _dS )zInitialize
|
|
The pool argument must be an instance of MySQLConnectionPoll. cnx
|
if an instance of MySQLConnection.
|
z$pool should be a MySQLConnectionPoolz cnx should be a MySQLConnectionN)r) r( r+ r Ú _cnx_poolÚ_cnx)Úselfrq rr r, r, r- Ú__init__P s
|
|
z PooledMySQLConnection.__init__©r c C s | S r` r, ©ru r, r, r- Ú __enter___ s z PooledMySQLConnection.__enter__zType[BaseException]Ú BaseExceptionr )Úexc_typeÚ exc_valueÚ tracebackr c C s | ¡ d S r` )Úclose)ru r{ r| r} r, r, r- Ú__exit__b s z PooledMySQLConnection.__exit__r )Úattrr c C s t | j|S )z0Calls attributes of the MySQLConnection instance)Úgetattrrt )ru r r, r, r- Ú __getattr__j s z!PooledMySQLConnection.__getattr__c C s4 z| j}| j jr| ¡ W 5 | j |¡ d| _X dS )aY Do 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)rs Úadd_connectionrt Ú reset_session©ru rr r, r, r- r~ n s
|
zPooledMySQLConnection.closer r c K s t ddS )z&Configuration is done through the poolzKConfiguration for pooled connections should be done through the pool itselfN)r ra r, r, r- rQ s ÿz PooledMySQLConnection.configrm c C s | j jS ©z&Return the name of the connection pool)rs r rx r, r, r- r s z PooledMySQLConnection.pool_nameN) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__rv ry r r r~ Ú staticmethodrQ Úpropertyr r, r, r, r- r A s c @ sÌ e Zd ZdZd*ddddd d
|
d d Zed dddZeddddZeddddZdd dddZ dd dddZ
|
d d dd d Z d d d d d!Z d+d"d d d#d$Z d%dd&d'Zddd(d)ZdS ),r( z*Class defining a pool of MySQL connectionsé NTrH z Optional[str]rc r rp )r! r Úpool_reset_sessionr r c K s d| _ d| _|| _| |¡ | |p,tf |¡ i | _t | j ¡| _ t
|
| _ |r|| j f | d}|| j k r|| ¡ |d7 }q`dS )zõInitialize
|
|
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.
|
Nr r )Ú
|
_pool_sizeÚ
|
_pool_nameÚ_reset_sessionÚ_set_pool_sizeÚ_set_pool_namer# Ú _cnx_configÚqueueÚQueueÚ
|
_cnx_queuer Ú_config_versionÚ
|
set_configr )ru r! r r r Úcntr, r, r- rv s
|
|
þ
|
z MySQLConnectionPool.__init__rm rw c C s | j S r )r rx r, r, r- r ± s z MySQLConnectionPool.pool_namec C s | j S )z0Return number of connections managed by the pool)r rx r, r, r- r! ¶ s z MySQLConnectionPool.pool_sizec C s | j S )z Return whether to reset session)r rx r, r, r- r » s z!MySQLConnectionPool.reset_sessionr c K st |sdS t ^ z$t }|jf | || _t | _W n2 tk
|
rd } ztd| |W 5 d}~X Y nX W 5 Q R X dS )aT Set 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' rP rQ r r r r+ r )ru r Ztest_cnxÚerrr, r, r- r À s z MySQLConnectionPool.set_config)r! r c C s( |dks|t kr tdt || _dS )a Set 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.
|
r z8Pool size should be higher than 0 and lower or equal to N)ÚCNX_POOL_MAXSIZEr+ r )ru r! r, r, r- r Õ s
|
ÿz"MySQLConnectionPool._set_pool_size)r r c C s@ t |¡rtd| dt|tkr6td| d|| _dS )a Set 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_NAMEREGEXÚsearchr+ rE ÚCNX_POOL_MAXNAMESIZEr )ru r r, r, r- r ä s
|
|
z"MySQLConnectionPool._set_pool_namero )rr r c
|
C sZ t |tstdz| jj|dd W n. tjk
|
rT } ztd|W 5 d}~X Y nX dS )zôPut 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 MySQLConnectionF©Úblockú'Failed adding connection; queue is fullN)r) r r r Úputr ÚFull©ru rr r r, r, r- Ú_queue_connectionò s
|
z%MySQLConnectionPool._queue_connectionz2Optional[Union[MySQLConnection, CMySQLConnection]]c C s¨ t | jstd| j ¡ r&td|s~tf | j}z(| jr\| jd r\| ¡ dk r\tdW n t k
|
rr Y nX | j
|
|_ nt |t std| |¡ W 5 Q R X dS )aá Add 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 availabler£ Úcompress)r é é z^Pool reset session is not supported with compression for MySQL server version 5.7.2 or earlierr N)r' r r r ÚfullrP r Zget_server_versionr rA r Úpool_config_versionr) r r§ r
r, r, r- r s2
|
ÿþ
|
ýÿ
|
|
ÿz"MySQLConnectionPool.add_connectionr c C sÀ t ² z| jjdd}W n. tjk
|
rF } ztd|W 5 d}~X Y nX | ¡ r\| j|jkr |j f | j
|
z | ¡ W n t k
|
r | |¡ Y nX | j|_t| |W 5 Q R £ S Q R X dS )aw Get 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.
|
Fr¡ z)Failed getting connection; pool exhaustedN)r' r r$ r ÚEmptyr Z is_connectedr r¬ rQ r Z reconnectr r§ r r¦ r, r, r- r* 2 s" ÿ
|
þ
|
z"MySQLConnectionPool.get_connectionc
|
C s¢ t d}| j}| ¡ rz |jdd}| ¡ |d7 }W q tjk
|
r^ | Y W 5 Q R £ S tk
|
rr Y q tk
|
r Y qX q|W 5 Q R £ S Q R X dS )z¹Close all connections
|
|
This method closes all connections. It returns the number
|
of connections it closed.
|
|
Used mostly for tests.
|
|
Returns int.
|
r Fr¡ r N) r' r Úqsizer$ Z
|
disconnectr r r r )ru r Zcnxqrr r, r, r- Ú_remove_connectionsT s
|
z'MySQLConnectionPool._remove_connections)r NT)N)r r r r rv r r r! r r r r r§ r r* r¯ r, r, r, r- r( s& ü ÿ/"r( )7r Ú
|
__future__r r rL ÚreÚ threadingÚtypesr Útypingr r r r r r r
|
Úuuidr Z dns.exceptionre Z dns.resolverr% rd Zconnection_cextr Ú
|
connectionr Ú constantsr r Úerrorsr r r r r Z optionfilesr ÚRLockr' r r Úcompiler r& r Ú__annotations__r r. rY rP r# r r( r, r, r, r- Ú<module> sH $
|
|
|
ÿ
|
]bM
|
