Dashboard
PHP:
8.2.30
OS:
Linux
User:
dh_nv8b2m
/
/
usr
/
lib
/
python3
/
dist-packages
/
psycopg2
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: pool.cpython-310.pyc
o ��a� � @ 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 �//usr/lib/python3/dist-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
Save
Cancel