File: //lib/python3/dist-packages/hgext/remotefilelog/__pycache__/connectionpool.cpython-310.pyc
o
�]Lb� � @ sN d dl mZ d dlmZmZmZmZ ejZG dd� de �Z
G dd� de �Z dS )� )�absolute_import)�hg�pycompat�sshpeer�utilc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�connectionpoolc C s || _ t� | _d S �N)�_repo�dict�_pool)�self�repo� r �D/usr/lib/python3/dist-packages/hgext/remotefilelog/connectionpool.py�__init__ s zconnectionpool.__init__c s� | j �|�}|d u rt� }|| j |< d }t|�dkr;z|�� }|j}t|t�r0|j� � d ur0d }W n t
y: Y nw |d u rat �| j j
i |�}t�|d�r\G � fdd�d|j�� � |_t||�}|S )Nr �_cleanupc s e Zd Zd� �fdd� Z� ZS )z"connectionpool.get.<locals>.mypeerNc s$ t �| d�r | j�� t�| ��� S )N�pipee)r � safehasattrr �close�superr )r �warn)� __class__�mypeerr r r 1 s
z+connectionpool.get.<locals>.mypeer._cleanupr )�__name__�
__module__� __qualname__r �
__classcell__r �r )r r r 0 s r )r �get�list�len�pop�peer�
isinstance�
_sshv1peer� _subprocess�poll�
IndexErrorr r �uir r r �
connection)r �path�pathpool�connr" r r r r s.
� �
zconnectionpool.getc C s2 t �| j�D ]}|D ]}|�� q
|d d �= qd S r )r �
itervaluesr r )r r+ r, r r r r ? s
�zconnectionpool.closeN)r r r r r r r r r r r s &r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r) c C s || _ || _d S r )r r" )r �poolr" r r r r G s
zconnection.__init__c C s | S r r �r r r r � __enter__K s zconnection.__enter__c C s$ |d u r | j �| � d S | �� d S r )r �appendr )r �type�value� tracebackr r r �__exit__N s zconnection.__exit__c C s t �| jd�r| j�� d S d S )N�cleanup)r r r" r6 r/ r r r r W s �zconnection.closeN)r r r r r0 r5 r r r r r r) F s
r) N) �
__future__r � mercurialr r r r � sshv1peerr$ �objectr r) r r r r �<module> s
2