File: //home/arjun/projects/env/lib64/python3.10/site-packages/psycopg2/__pycache__/pool.cpython-310.pyc
o
we� � @ s\ d Z ddlZddlmZ G dd� dej�ZG dd� d�ZG dd � d e�ZG d
d � d e�ZdS ) 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 �F/home/arjun/projects/env/lib/python3.10/site-packages/psycopg2/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 sZ t |�| _t |�| _d| _|| _|| _g | _i | _i | _d| _ t
| j�D ]}| � � q$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 sH t j| ji | j��}|dur || j|< || jt|�< |S | j�|� |S )z;Create a new connection and assign it to 'key' if not None.N) �psycopg2�connectr 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 �_getkeyG s z AbstractConnectionPool._getkeyc C s� | j rtd��|du r| �� }|| jv r| j| S | jr/| j�� | j|< }|| jt|�< |S t| j�| j kr;td��| �
|�S )�9Get a free connection and assign it to 'key' if not None.�connection pool is closedNzconnection pool exhausted) r
r r$ r r �popr r �lenr r r r r r �_getconnL s
z AbstractConnectionPool._getconnFc C s� | j rtd��|du r| j�t|��}|du rtd��t| j�| jk rM|sM|j sL|jj }|t
j kr6|� � n|t
j
krF|�� | j�|� n | j�|� n|� � | j rY|| jv re| j|= | jt|�= dS dS )zPut away a connection.r& Nz trying to put unkeyed connection)r
r r �getr r( r r �info�transaction_status�_ext�TRANSACTION_STATUS_UNKNOWN�close�TRANSACTION_STATUS_IDLE�rollbackr r )r r! r r/ �statusr r r �_putconn_ s*
��z AbstractConnectionPool._putconnc C sP | j rtd��| jt| j�� � D ]}z|�� W q ty" Y qw 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.
r& TN)r
r r �listr �valuesr/ � Exception)r r! r r r � _closeall� s �
z AbstractConnectionPool._closeall�N)NF)
r r r �__doc__r r r$ r) r3 r7 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 r9 r r) �getconnr3 �putconnr7 �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 s2 ddl }tj| ||g|�R i |�� |�� | _dS )z Initialize the threading lock.r N)� threadingr r �Lock�_lock)r r r r r r? r r r r � s ���z ThreadedConnectionPool.__init__Nc C s. | j �� z | �|�W | j �� S | j �� w )r% )rA �acquirer) �release)r r r r r r; � s
z ThreadedConnectionPool.getconnFc C s6 | j �� z| �|||� W | j �� dS | j �� w )z Put away an unused connection.N)rA rB r3 rC )r r! r r/ r r r r< � s
z ThreadedConnectionPool.putconnc C s0 | j �� z | �� W | j �� dS | j �� w )z6Close all connections (even the one currently in use.)N)rA rB r7 rC r# r r r r= � s
z ThreadedConnectionPool.closeallr8 )NNF)r r r r9 r r; r< r= r r r r r> � s
r> ) r9 r r r- �Errorr r r: r> r r r r �<module> s o