XylotrechusZ
�
� h� � � � d Z ddlZddlmZ G d� dej � � Z G d� d� � Z G d� d e� � Z G d
� d e� � ZdS ) z`Connection pooling for psycopg2
This module implements thread-safe (and not) connection pools.
� N)�
extensionsc � � e Zd ZdS )� PoolErrorN)�__name__�
__module__� __qualname__� � �`/builddir/build/BUILD/cloudlinux-venv-1.0.8/venv/lib64/python3.11/site-packages/psycopg2/pool.pyr r s � � � � � ��Dr
r c �< � e Zd ZdZd� Zd
d�Zd� Zd
d�Zd d�Zd � Z dS ) �AbstractConnectionPoolz Generic key-based pooling code.c � � t |� � | _ t |� � | _ d| _ || _ || _ g | _ i | _ i | _ d| _ t | j � � D ]}| � � � �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�is r �__init__z AbstractConnectionPool.__init__' s� � � �7�|�|�� ��7�|�|�� ��� ���
� �� ���
���
��� ���
��t�|�$�$� � �A� �M�M�O�O�O�O� � r
Nc � � t j | j i | j ��}|�"|| j |< || j t
|� � < n| j � |� � |S )z;Create a new connection and assign it to 'key' if not None.) �psycopg2�connectr r r r �idr �append�r �key�conns r r z AbstractConnectionPool._connect= s[ � �� ���<�t�|�<�<�� �?� "�D�J�s�O�$'�D�K��4��� !� !� �J� � �d� #� #� #�� r
c �0 � | xj dz
c_ | j S )zReturn a new unique key.� )r �r s r �_getkeyz AbstractConnectionPool._getkeyG s � � �
�
�a��
�
��z�r
c � � | j rt d� � �|�| � � � }|| j v r
| j | S | j r<| j � � � x| j |< }|| j t |� � < |S t | j � � | j k rt d� � �| �
|� � 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% s r �_getconnz AbstractConnectionPool._getconnL s� � � �;� 9�� 7�8�8� 8� �;��,�,�.�.�C� �$�*� � ��:�c�?� "� �:� &�%)�Z�^�^�%5�%5� 5�D�J�s�O�d�$'�D�K��4��� !��K��4�:���$�,�.�.� � ;�<�<�<��=�=��%�%� %r
Fc � � | j rt d� � �|�8| j � t |� � � � }|�t d� � �t | j � � | j k r�|s�|j s�|j j }|t j k r|� � � nn|t j
k r/|� � � | j � |� � n/| j � |� � n|� � � | j r || j v r | j |= | j t |� � = dS dS )zPut away a connection.r. Nz trying to put unkeyed connection)r r r �getr# r0 r r �info�transaction_status�_ext�TRANSACTION_STATUS_UNKNOWN�close�TRANSACTION_STATUS_IDLE�rollbackr$ r )r r' r& r8 �statuss r �_putconnz AbstractConnectionPool._putconn_ s5 � � �;� 9�� 7�8�8� 8� �;��+�/�/�"�T�(�(�+�+�C��{� � B�C�C�C� �t�z�?�?�T�\� )� )�%� )� �;�
,� ��5���T� <�<�<��J�J�L�L�L�L��t� ;�;�;��M�M�O�O�O��J�%�%�d�+�+�+�+� �J�%�%�d�+�+�+��
�J�J�L�L�L� �{� &�c�T�Z� /� /��
�3��� �B�t�H�H�%�%�%� 0� /r
c �� � | j rt d� � �| j t | j � � � � � z D ]'} |� � � �# t $ r Y �$w xY wd| _ 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 �valuesr8 � Exception)r r' s r � _closeallz AbstractConnectionPool._closeall� s� � � �;� 9�� 7�8�8� 8��J��d�j�&7�&7�&9�&9�!:�!:�:� � �D�
��
�
� � � � �� �
�
�
���
������ � � s �A �
A*�)A*�N)NF)
r r r �__doc__r r r+ r1 r< rA r r
r r
r
$ s� � � � � � �)�)� � � �,� � � �� � �
&� &� &� &�&!&� !&� !&� !&�F� � � � r
r
c �< � e Zd ZdZej Zej Zej Z
dS )�SimpleConnectionPoolz@A connection pool that can't be shared across different threads.N) r r r rC r
r1 �getconnr< �putconnrA �closeallr r
r rE rE � s, � � � � � �J�J�$�-�G�$�-�G�%�/�H�H�Hr
rE c �. � e Zd ZdZd� Zdd�Zd d�Zd� ZdS )
�ThreadedConnectionPoolz7A connection pool that works with the threading module.c �n � ddl }t j | ||g|�R i |�� |� � � | _ dS )z Initialize the threading lock.r N)� threadingr
r �Lock�_lock)r r r r r rL s r r z ThreadedConnectionPool.__init__� sY � ����� �'� �'�7� 5�%)� 5� 5� 5�-3� 5� 5� 5� �^�^�%�%��
�
�
r
Nc �� � | j � � � | � |� � | j � � � S # | j � � � w xY w)r- )rN �acquirer1 �release)r r& s r rF z ThreadedConnectionPool.getconn� s[ � � �
��� � � � !��=�=��%�%� �J� � � � � � ��D�J� � � � � � ���s �A � A$Fc �� � | j � � � | � |||� � | j � � � dS # | j � � � w xY w)z Put away an unused connection.N)rN rP r< rQ )r r' r& r8 s r rG z ThreadedConnectionPool.putconn� se � � �
��� � � � !� �M�M�$��U� +� +� +� �J� � � � � � � ��D�J� � � � � � ���s �A
�
A(c �� � | j � � � | � � � | j � � � dS # | j � � � w xY w)z6Close all connections (even the one currently in use.)N)rN rP rA rQ r* s r rH z ThreadedConnectionPool.closeall� s_ � � �
��� � � � !� �N�N� � � � �J� � � � � � � ��D�J� � � � � � ���s �A
�
A%rB )NNF)r r r rC r rF rG rH r r
r rJ rJ � s` � � � � � �A�A�&� &� &�!� !� !� !�!� !� !� !�!� !� !� !� !r
rJ ) rC r! r r6 �Errorr r
rE rJ r r
r �<module>rU s� ��� �6 ���� '� '� '� '� '� '� � � � � ��� � � �l� l� l� l� l� l� l� l�^0� 0� 0� 0� 0�1� 0� 0� 0� !� !� !� !� !� 3� !� !� !� !� !r