D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
opt
/
cloudlinux
/
venv
/
lib
/
python3.11
/
site-packages
/
psycopg2
/
__pycache__
/
Filename :
pool.cpython-311.pyc
back
Copy
� � 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