File: //usr/lib/python3/dist-packages/trac/db/__pycache__/pool.cpython-310.pyc
o
�k�`$ � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z d dl
mZ G dd � d e�ZG d
d � d e�ZG d d
� d
e�Zee j�dd��Zee�ZG dd� de�ZdS )� N)� TracError)�ConnectionWrapper)�
get_thread_id� threading)�time_now)�exception_to_unicode)�_c @ s e Zd ZdZdS )� TimeoutErrorzjException raised by the connection pool when no connection has become
available after a given timeout.N)�__name__�
__module__� __qualname__�__doc__� r r �./usr/lib/python3/dist-packages/trac/db/pool.pyr s r c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
�PooledConnectionz]A database connection that can be pooled. When closed, it gets returned
to the pool.
Nc C s$ t �| ||� || _|| _|| _d S �N)r �__init__�_pool�_key�_tid)�self�pool�cnx�key�tid�logr r r r & s
zPooledConnection.__init__c C s4 | j r| j }d | _ d | _| j�|| j| j� d S d S r )r r r � _return_cnxr r )r r r r r �close, s �zPooledConnection.closec C s | � � d S r )r )r r r r �__del__3 s zPooledConnection.__del__r )r
r r r
r r r r r r r r ! s
r c @ s< e Zd ZdZdd� Zd
dd�Zdd� Zd d
� Zd
d d �ZdS )�ConnectionPoolBackendz.A process-wide LRU-based connection pool.
c C s8 t �t �� �| _|| _i | _g | _g | _g | _d| _ d S )Nr )
r � Condition�RLock�
_available�_maxsize�_activer � _pool_key�
_pool_time�_waiters)r �maxsizer r r r ; s
z ConnectionPoolBackend.__init__Nc
C s� d }|� d�}t|�}t� }t� }| j�T ||f| jv r*| j||f \}} | d7 } n,| jdkr7| �||||�}|sT| jd7 _| j�� | jd8 _| �||||�}d} |ra|| f| j||f< W d � n1 skw Y | dkoxt |t
�}
d } |
r�|\} }z | dkr�|� � n| dkr�|� � | dv r�|j
d i |��}W n5 ty� }
z
|
} d }W Y d }
~
n$d }
~
w ty� }
z|
} |r�|jd| dd � d }W Y d }
~
nd }
~
ww |�rt |t
��s|
r�| j� || f| j||f< W d � n1 s�w Y t| ||||�S |
�r-| j� | j||f= W d � n 1 �s w Y | dk�r-| �||�S t | t��r5| �t� | }td
|d �}t|�| �)
Nr � r �pingr )r �createzException caught on %sT)�exc_infoz:Unable to get database connection within %(time)d seconds.)�timer )�get�strr r r" r$ r' � _take_cnx�wait�
isinstance�tupler* r �get_connectionr � Exception�errorr �get_cnxr r )r � connector�kwargs�timeoutr r r �startr �num�deferred� exception�op�e�errmsgr r r r7 D s~
� �
���� � �
�
z ConnectionPoolBackend.get_cnxc C s� || j v r(| j �|�}| j �|� | j�|� | j�|�}t|d�r&d|fS |S t| j�t| j� | jk r7dS t| j�| jk rU| j�d�}| j �d� | j�d� d|fS dS )z<Note: _available lock must be held when calling this method.r* )r+ Nr r N) r% �index�popr& r �hasattr�lenr$ r# )r r8 r9 r r �idxr r r r r0 � s
�z ConnectionPoolBackend._take_cnxc C s | j �0 ||f| jv s
J �| j||f \}}|dkr!| j||f= n ||d f| j||f< W d � n1 s6w Y |dkr�z|�� W n tyT |�� d }Y nw | j �& |rq|jrq| j�|� | j�|� | j �t
� � | j � � W d � d S 1 s�w Y d S d S )Nr) ) r" r$ �rollbackr5 r �poolabler �appendr% r& r �notify)r r r r r< r r r r � s. � � �
"��z!ConnectionPoolBackend._return_cnxc C s� d}|du rd}t � | }| j�e |du r+| j�� D ]
\}}t|t�s'|�� qi | _| jr]| jd |krm| j� d�}|�� | j
� d� | j� d� | jre| jd |ks5W d � dS W d � dS W d � dS W d � dS 1 sxw Y dS )z,Close pooled connections not used in a while�x Nr ) r r" r$ �valuesr2 r3 r r& r rC r% )r r �delay�when�dbr< r r r �shutdown� s0
� ����"�z ConnectionPoolBackend.shutdownr ) r
r r r
r r7 r0 r rP r r r r r 8 s
Cr �TRAC_DB_POOL_SIZE�
c @ s( e Zd Zdd� Zddd�Zddd�ZdS ) �ConnectionPoolc K s || _ || _d S r )�
_connector�_kwargs)r r( r8 r9 r r r r � s
zConnectionPool.__init__Nc C s t �| j| j|�S r )�_backendr7 rT rU )r r: r r r r7 � s zConnectionPool.get_cnxc C s t �|� d S r )rV rP )r r r r r rP � s zConnectionPool.shutdownr )r
r r r r7 rP r r r r rS � s
rS )�os�sys� trac.corer � trac.db.utilr �trac.util.concurrencyr r �trac.util.datefmtr �trac.util.textr �trac.util.translationr r r �objectr �int�environr. �
_pool_sizerV rS r r r r �<module> s