관리-도구
편집 파일: pooling.cpython-38.pyc
U )�e]0 � @ s� d Z ddlZddlmZ z ddlZW n ek r@ ddlZY nX ddlZddlm Z ddl m Z e� � Z dZdZe�d �Zd d � ZG d d � d e�ZG dd� de�ZdS )z6Implementing pooling of connections to MySQL servers. � N)�uuid4� )�errors)�MySQLConnection� �@ z[^a-zA-Z0-9._:\-*$#]c K sR g }dD ]0}z|� t| | �� W q tk r6 Y qX q|sHt�d��d�|�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. )�host�port�userZdatabasez.Failed generating pool name; specify pool_name�_)�append�str�KeyErrorr � PoolError�join)�kwargs�parts�key� r �l/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysql/connector/pooling.py�generate_pool_name/ s �r c @ s< e Zd ZdZdd� Zdd� Zdd� Zdd � Zed d � �Z d S ) �PooledMySQLConnectiona 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. c C s4 t |t�std��t |t�s$td��|| _|| _dS )z�Initialize 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)� isinstance�MySQLConnectionPool�AttributeErrorr � _cnx_pool�_cnx)�self�pool�cnxr r r �__init__V s � �z PooledMySQLConnection.__init__c C s t | j|�S )z0Calls attributes of the MySQLConnection instance)�getattrr )r �attrr r r � __getattr__e s z!PooledMySQLConnection.__getattr__c C s, | j }| jjr|�� | j�|� d| _ 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)r r � reset_session�add_connection�r r r r r �closei s zPooledMySQLConnection.closec K s t �d��dS )z&Configuration is done through the poolzLConfiguration for pooled connections should be done through the pool itself.N)r r )r r r r r �configz s �z PooledMySQLConnection.configc C s | j jS �z&Return the name of the connection pool)r � pool_name�r r r r r* � s z PooledMySQLConnection.pool_nameN) �__name__� __module__� __qualname__�__doc__r r# r' r( �propertyr* r r r r r H s r c @ sx e Zd ZdZddd�Zedd� �Zed d � �Zed d � �Zd d� Z dd� Z dd� Z dd� Z d dd�Z dd� Zdd� ZdS ) r z*Class defining a pool of MySQL connections� NTc 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% )r � pool_sizer* Zpool_reset_sessionr �cntr r r r � s z MySQLConnectionPool.__init__c C s | j S r) )r3 r+ r r r r* � s z MySQLConnectionPool.pool_namec C s | j S )z0Return number of connections managed by the pool)r2 r+ r r r r= � s z MySQLConnectionPool.pool_sizec C s | j S )z Return whether to reset session)r4 r+ r r r r$ � s z!MySQLConnectionPool.reset_sessionc K st |sdS t �^ z$t� }|jf |� || _t� | _W n2 tk rd } zt�d� |���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: {0}) �CONNECTION_POOL_LOCKr r( r7 r r; r r r �format)r r Ztest_cnx�errr r r r< � s �z MySQLConnectionPool.set_configc C s( |dks|t kr td�t ���|| _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 z;Pool size should be higher than 0 and lower or equal to {0}N)�CNX_POOL_MAXSIZEr r@ r2 )r r= r r r r5 � s ��z"MySQLConnectionPool._set_pool_sizec C s< t �|�rtd�|���t|�tkr2td�|���|| _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 '{0}' contains illegal characterszPool name '{0}' is too longN)�CNX_POOL_NAMEREGEX�searchr r@ �len�CNX_POOL_MAXNAMESIZEr3 )r r* r r r r6 � s � �z"MySQLConnectionPool._set_pool_namec C sN t |t�st�d��z| jj|dd� W n tjk rH t�d� 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. �4Connection instance not subclass of MySQLConnection.F��block�'Failed adding connection; queue is fullN)r r r r r: �putr8 �Fullr& r r r �_queue_connection� s �z%MySQLConnectionPool._queue_connectionc C s� t �� | jst�d��| j�� r*t�d��|s�tf | j�}z*| jrb| jd rb|�� dk rbt� d��W n t k rx Y nX | j |_ nt |t�s�t�d��| �|� 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 availablerJ �compress)r1 � � z_Pool reset session is not supported with compression for MySQL server version 5.7.2 or earlier.rG N)r? r7 r r r: �fullr r4 Zget_server_versionZNotSupportedErrorr r; �_pool_config_versionr rM r& r r r r% � s0 � � � �z"MySQLConnectionPool.add_connectionc C s� t �� z| jjdd�}W n tjk r8 t�d��Y nX |�� rN| j|j kr�|j f | j � z |� � W n" tj 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. FrH z)Failed getting connection; pool exhaustedN)r? r: �getr8 �Emptyr r Z is_connectedr; rR r( r7 Z reconnectZInterfaceErrorrM r r& r r r �get_connection$ s$ � � z"MySQLConnectionPool.get_connectionc C s� t �� d}| j}|�� r�z |jdd�}|�� |d7 }W q tjk r^ | Y W 5 Q R � S tjk rt � Y q tj k 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 FrH r N) r? r: �qsizerS Z disconnectr8 rT r r �Error)r r> Zcnxqr r r r �_remove_connectionsG s z'MySQLConnectionPool._remove_connections)r1 NT)N)r, r- r. r/ r r0 r* r= r$ r< r5 r6 rM r% rU rX r r r r r � s -#r )r/ �re�uuidr r8 � ImportErrorr9 � threading� r � connectionr �RLockr? rB rF �compilerC r �objectr r r r r r �<module> s ?