Current File : //opt/alt/python37/lib64/python3.7/site-packages/psycopg2/__pycache__/pool.cpython-37.pyc
B
*��]� � @ s^ d Z ddlZddlmZ G dd� dej�ZG dd� de�ZG dd � d e�ZG d
d � d e�Z dS ) z`Connection pooling for psycopg2
This module implements thread-safe (and not) connection pools.
� N)�
extensionsc @ s e Zd ZdS )� PoolErrorN)�__name__�
__module__� __qualname__� r r �/pool.pyr s r c @ sF e Zd ZdZdd� Zddd�Zdd� Zdd d
�Zdd d
�Zdd� Z dS )�AbstractConnectionPoolz Generic key-based pooling code.c O s^ t |�| _t |�| _d| _|| _|| _g | _i | _i | _d| _ xt
| j�D ] }| � � qJW dS )z�Initialize the connection pool.
New 'minconn' connections are created immediately calling 'connfunc'
with given parameters. The connection pool will support a maximum of
about 'maxconn' connections.
Fr N) �int�minconn�maxconn�closed�_args�_kwargs�_pool�_used�_rused�_keys�range�_connect)�selfr r �args�kwargs�ir r r �__init__& s
z AbstractConnectionPool.__init__Nc C sB t j| j| j�}|dk r2|| j|< || jt|�< n | j�|� |S )z;Create a new connection and assign it to 'key' if not None.N) �psycopg2Zconnectr r r r �idr �append)r �key�connr r r r < s
z AbstractConnectionPool._connectc C s | j d7 _ | j S )zReturn a new unique key.� )r )r r r r �_getkeyF s z AbstractConnectionPool._getkeyc C s� | j rtd��|dkr | �� }|| jkr2| j| S | jr^| j�� | j|< }|| jt|�< |S t| j�| j krvtd��| �
|�S dS )z9Get a free connection and assign it to 'key' if not None.zconnection pool is closedNzconnection pool exhausted) r
r r! r r �popr r �lenr r )r r r r r r �_getconnK s
z AbstractConnectionPool._getconnFc C s� | j rtd��|dkr6| j�t|��}|dkr6td��t| j�| jk r�|s�|j s�|jj }|t
j krl|� � q�|t
j
kr�|�� | j�|� q�| j�|� n|� � | j r�|| jkr�| j|= | jt|�= dS )zPut away a connection.zconnection pool is closedNz trying to put unkeyed connection)r
r r �getr r# r r �infoZtransaction_status�_extZTRANSACTION_STATUS_UNKNOWN�closeZTRANSACTION_STATUS_IDLEZrollbackr r )r r r r( Zstatusr r r �_putconn^ s&
z AbstractConnectionPool._putconnc C sX | j rtd��x>| jt| j�� � D ]&}y |�� W q$ tk
rH Y q$X q$W d| _ dS )z�Close all connections.
Note that this can lead to some code fail badly when trying to use
an already closed connection. If you call .closeall() make sure
your code can deal with it.
zconnection pool is closedTN)r
r r �listr �valuesr( � Exception)r r r r r � _closeall� s
z AbstractConnectionPool._closeall)N)N)NF)
r r r �__doc__r r r! r$ r) r- r r r r r # s
#r c @ s" e Zd ZdZejZejZej Z
dS )�SimpleConnectionPoolz@A connection pool that can't be shared across different threads.N) r r r r. r r$ �getconnr) �putconnr- �closeallr r r r r/ � s r/ c @ s4 e Zd ZdZdd� Zd dd�Zd
dd �Zd
d � ZdS )�ThreadedConnectionPoolz7A connection pool that works with the threading module.c O s, ddl }tj| ||f|�|� |�� | _dS )z Initialize the threading lock.r N)� threadingr r ZLock�_lock)r r r r r r4 r r r r � s z ThreadedConnectionPool.__init__Nc C s& | j �� z
| �|�S | j �� X dS )z9Get a free connection and assign it to 'key' if not None.N)r5 �acquirer$ �release)r r r r r r0 � s
z ThreadedConnectionPool.getconnFc C s. | j �� z| �|||� W d| j �� X dS )z Put away an unused connection.N)r5 r6 r) r7 )r r r r( r r r r1 � s
z ThreadedConnectionPool.putconnc C s( | j �� z | �� W d| j �� X dS )z6Close all connections (even the one currently in use.)N)r5 r6 r- r7 )r r r r r2 � s
z ThreadedConnectionPool.closeall)N)NNF)r r r r. r r0 r1 r2 r r r r r3 � s
r3 )
r. r r r' �Errorr �objectr r/ r3 r r r r �<module> s o
Mini Shell