관리-도구
편집 파일: base.cpython-38.pyc
U -?�f � � @ s� U d Z ddlmZ ddlmZ ddlZddlmZ ddlZddl Z ddl Z ddl m Z ddl m Z ddl m Z dd l mZ dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z e�rlddl m Z ddl m Z ddl m!Z! ddlm"Z" ddlm#Z# dd lm$Z$ dd l%m&Z& ej'd d �G d d!� d!��Z(G d"d#� d#e�Z)ee)e d$ f Z*e+e)�\Z,Z-Z.G d%d&� d&�Z/G d'd(� d(e/�Z0G d)d*� d*e �Z1G d+d,� d,e �Z2G d-d.� d.ej3ej4�Z5G d/d0� d0�Z6G d1d2� d2e6�Z7G d3d4� d4e7�Z8dHd6d7d.d8d9d:d;d<d=�d>d?�Z9i Z:d@e;dA< G dBdC� dCe6�Z<G dDdE� dEe<�Z=G dFdG� dGe<�Z>dS )Iz'Base constructs for connection pools. � )� annotations)�dequeN)�Enum)�Any)�Callable)�cast)�Deque)�Dict)�List)�Optional)�Tuple)� TYPE_CHECKING)�Union� )�event)�exc)�log)�util)�Literal)�Protocol)�DBAPIConnection)� DBAPICursor)�Dialect)�_DispatchCommon)�_ListenerFnType)� dispatcher)� _InfoTypeT)�frozenc @ s. e Zd ZU dZdZded< ded< ded< dS )�PoolResetStatez�describes the state of a DBAPI connection as it is being passed to the :meth:`.PoolEvents.reset` connection pool event. .. versionadded:: 2.0.0b3 ��transaction_was_reset�terminate_only� asyncio_safe�boolr r! r" N)�__name__� __module__� __qualname__�__doc__� __slots__�__annotations__� r* r* �D/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/pool/base.pyr 2 s r c @ s e Zd ZdZdZdZdZdS )� ResetStylez1Describe options for "reset on return" behaviors.r � r N)r$ r% r& r' �reset_rollback� reset_commit� reset_noner* r* r* r+ r, _ s r, )TNF�commit�rollbackc @ sx e Zd ZdZdZdZddd�dd�Zddd�dd �Zd dd�d d �Zd dd�d d�Z d dd�dd�Z d dd�dd�Z dS )� _ConnDialectz�partial implementation of :class:`.Dialect` which provides DBAPI connection methods. When a :class:`_pool.Pool` is combined with an :class:`_engine.Engine`, the :class:`_engine.Engine` replaces this with its own :class:`.Dialect`. F�PoolProxiedConnection�None)�dbapi_connection�returnc C s |� � d S �N)r2 ��selfr6 r* r* r+ � do_rollback{ s z_ConnDialect.do_rollbackc C s |� � d S r8 )r1 r9 r* r* r+ � do_commit~ s z_ConnDialect.do_commitr c C s |� � d S r8 ��closer9 r* r* r+ � do_terminate� s z_ConnDialect.do_terminatec C s |� � d S r8 r= r9 r* r* r+ �do_close� s z_ConnDialect.do_closer# c C s t d��d S )NzJThe ping feature requires that a dialect is passed to the connection pool.��NotImplementedErrorr9 r* r* r+ �_do_ping_w_event� s �z _ConnDialect._do_ping_w_eventr )� connectionr7 c C s |S r8 r* )r: rD r* r* r+ �get_driver_connection� s z"_ConnDialect.get_driver_connectionN) r$ r% r& r' �is_async� has_terminater; r< r? r@ rC rE r* r* r* r+ r3 n s r3 c @ s e Zd ZdZdS )�_AsyncConnDialectTN)r$ r% r& rF r* r* r* r+ rH � s rH c @ s e Zd Zdd�dd�ZdS )�_CreatorFnTyper �r7 c C s d S r8 r* �r: r* r* r+ �__call__� � z_CreatorFnType.__call__N�r$ r% r& rL r* r* r* r+ rI � s rI c @ s e Zd Zddd�dd�ZdS )�_CreatorWRecFnType�ConnectionPoolEntryr )�recr7 c C s d S r8 r* �r: rQ r* r* r+ rL � rM z_CreatorWRecFnType.__call__NrN r* r* r* r+ rO � s rO c @ sr e Zd ZU dZded< ded< ded< e� Zded < d ed < d ed < ded< dEd ddddddddd� dd �Zej dd �d d ��Z e d d �d d!��Z e j d d"d#�d$d!��Z e jd"d �d%d!��Z d d d#�d&d'�Zdd(�d)dd"d*�d+d,�Zd-d �d.d/�ZdFd0d1dd"d2�d3d4�Zd d �d5d6�Zd"d �d7d8�Zd0d �d9d:�Zd-d"d;�d<d=�Zd-d �d>d?�Zd-d"d;�d@dA�ZdBd �dCdD�ZdS )G�Poolz)Abstract base class for connection pools.zdispatcher[Pool]�dispatch�log._EchoFlagType�echoz Optional[str]�_orig_logging_namez Union[_ConnDialect, Dialect]�_dialectz)Union[_CreatorFnType, _CreatorWRecFnType]� _creator_argrO �_invoke_creator�float�_invalidate_time���NTF�int�_ResetStyleArgTypez+Optional[List[Tuple[_ListenerFnType, str]]]z&Optional[Union[_ConnDialect, Dialect]]r# z Optional[_DispatchCommon[Pool]]) �creator�recyclerV � logging_name�reset_on_return�events�dialect�pre_ping� _dispatchc C s� |r| | _ | _nd| _tj| |d� || _|| _d| _|| _t� |t j ddgt j dddgt j dgid �| _|| _| r�| jj| dd � |r�|| _|r�|D ]\} } t�| | | � q�dS ) a4 Construct a Pool. :param creator: a callable function that returns a DB-API connection object. The function will be called with parameters. :param recycle: If set to a value other than -1, number of seconds between connection recycling, which means upon checkout, if this timeout is surpassed the connection will be closed and replaced with a newly opened connection. Defaults to -1. :param logging_name: String identifier which will be used within the "name" field of logging records generated within the "sqlalchemy.pool" logger. Defaults to a hexstring of the object's id. :param echo: if True, the connection pool will log informational output such as when connections are invalidated as well as when connections are recycled to the default log handler, which defaults to ``sys.stdout`` for output.. If set to the string ``"debug"``, the logging will include pool checkouts and checkins. The :paramref:`_pool.Pool.echo` parameter can also be set from the :func:`_sa.create_engine` call by using the :paramref:`_sa.create_engine.echo_pool` parameter. .. seealso:: :ref:`dbengine_logging` - further detail on how to configure logging. :param reset_on_return: Determine steps to take on connections as they are returned to the pool, which were not otherwise handled by a :class:`_engine.Connection`. Available from :func:`_sa.create_engine` via the :paramref:`_sa.create_engine.pool_reset_on_return` parameter. :paramref:`_pool.Pool.reset_on_return` can have any of these values: * ``"rollback"`` - call rollback() on the connection, to release locks and transaction resources. This is the default value. The vast majority of use cases should leave this value set. * ``"commit"`` - call commit() on the connection, to release locks and transaction resources. A commit here may be desirable for databases that cache query plans if a commit is emitted, such as Microsoft SQL Server. However, this value is more dangerous than 'rollback' because any data changes present on the transaction are committed unconditionally. * ``None`` - don't do anything on the connection. This setting may be appropriate if the database / DBAPI works in pure "autocommit" mode at all times, or if a custom reset handler is established using the :meth:`.PoolEvents.reset` event handler. * ``True`` - same as 'rollback', this is here for backwards compatibility. * ``False`` - same as None, this is here for backwards compatibility. For further customization of reset on return, the :meth:`.PoolEvents.reset` event hook may be used which can perform any connection activity desired on reset. .. seealso:: :ref:`pool_reset_on_return` :meth:`.PoolEvents.reset` :param events: a list of 2-tuples, each of the form ``(callable, target)`` which will be passed to :func:`.event.listen` upon construction. Provided here so that event listeners can be assigned via :func:`_sa.create_engine` before dialect-level listeners are applied. :param dialect: a :class:`.Dialect` that will handle the job of calling rollback(), close(), or commit() on DBAPI connections. If omitted, a built-in "stub" dialect is used. Applications that make use of :func:`_sa.create_engine` should not use this parameter as it is handled by the engine creation strategy. :param pre_ping: if True, the pool will emit a "ping" (typically "SELECT 1", but is dialect-specific) on the connection upon checkout, to test if the connection is alive or not. If not, the connection is transparently re-connected and upon success, all other pooled connections established prior to that timestamp are invalidated. Requires that a dialect is passed as well to interpret the disconnection error. .. versionadded:: 1.2 N)Zechoflagr r2 T�noneFr1 rc )Zonly_propagate)rb rW r Zinstance_logger�_creator�_recycler\ � _pre_pingr Z parse_user_argument_for_enumr, r. r0 r/ �_reset_on_returnrV rT �_updaterX r �listen) r: r` ra rV rb rc rd re rf rg �fn�targetr* r* r+ �__init__� s6 l �� z Pool.__init__rJ c C s | j jS r8 )rX rF rK r* r* r+ � _is_asyncio3 s zPool._is_asyncioc C s | j S r8 )rY rK r* r* r+ ri 7 s z Pool._creatorr5 )r` r7 c C s || _ | �|�| _d S r8 )rY �_should_wrap_creatorrZ )r: r` r* r* r+ ri ; s c C s | ` | `d S r8 )rY rZ rK r* r* r+ ri E s c s� zt j| jdd�}W n* tk r> tt|�� � fdd� Y S X |jdk rVt|j�}nd}t|d �| }|d |d fdgd fkr�tt|�S |d kr�tt|�S tt|�� � fd d�S dS ) zlDetect if creator accepts a single argument, or is sent as a legacy style no-arg function. T)Zno_selfc s � � S r8 r* �rQ �Z creator_fnr* r+ �<lambda>W rM z+Pool._should_wrap_creator.<locals>.<lambda>Nr � �connection_recordr8 r- c s � � S r8 r* rt ru r* r+ rv j rM ) r Zget_callable_argspecri � TypeErrorr rI �defaults�lenrO )r: r` ZargspecZ defaulted� positionalsr* ru r+ rs K s zPool._should_wrap_creator�� terminater )rD r~ r7 c C s� | j �d|rdnd|� z"|r,| j�|� n | j�|� W nP tk r� } z2| j jd|r\dnd� d�|dd � t|t�sz� W 5 d }~X Y nX d S ) Nz%s connection %rz Hard-closingZClosingz Exception Z terminating�closingz connection %rT��exc_info) �logger�debugrX r? r@ � BaseException�error� isinstance� Exception)r: rD r~ �er* r* r+ �_close_connectionl s" �� zPool._close_connectionrP c C s t | �S )z6Called by subclasses to create a new ConnectionRecord.)�_ConnectionRecordrK r* r* r+ �_create_connection� s zPool._create_connectionr4 �Optional[BaseException])rD � exception�_checkinr7 c C sD t |dd�}|r | j|jk r&t�� | _|r@t |dd�r@|�|� dS )a� Mark all connections established within the generation of the given connection as invalidated. If this pool's last invalidate time is before when the given connection was created, update the timestamp til now. Otherwise, no action is performed. Connections with a start time prior to this pool's invalidation time will be recycled upon next checkout. �_connection_recordN�is_validF)�getattrr\ � starttime�time� invalidate)r: rD r� r� rQ r* r* r+ � _invalidate� s zPool._invalidatec C s t � �dS )a" Return a new :class:`_pool.Pool`, of the same class as this one and configured with identical creation arguments. This method is used in conjunction with :meth:`dispose` to close out an entire :class:`_pool.Pool` and create a new one in its place. NrA rK r* r* r+ �recreate� s z Pool.recreatec C s t � �dS )z�Dispose of this pool. This method leaves the possibility of checked-out connections remaining open, as it only affects connections that are idle in the pool. .. seealso:: :meth:`Pool.recreate` NrA rK r* r* r+ �dispose� s z Pool.disposec C s t �| �S )z�Return a DBAPI connection from the pool. The connection is instrumented such that when its ``close()`` method is called, the connection will be returned to the pool. )�_ConnectionFairy� _checkoutrK r* r* r+ �connect� s z Pool.connect)�recordr7 c C s | � |� dS )z�Given a _ConnectionRecord, return it to the :class:`_pool.Pool`. This method is called when an instrumented DBAPI connection has its ``close()`` method called. N)�_do_return_conn�r: r� r* r* r+ � _return_conn� s zPool._return_connc C s t � �dS )z7Implementation for :meth:`get`, supplied by subclasses.NrA rK r* r* r+ �_do_get� s z Pool._do_getc C s t � �dS )z?Implementation for :meth:`return_conn`, supplied by subclasses.NrA r� r* r* r+ r� � s zPool._do_return_conn�strc C s t � �d S r8 rA rK r* r* r+ �status� s z Pool.status)r] NNTNNFN)NT)r$ r% r& r' r) r3 rX rq r Zhybridpropertyrr �propertyri �setter�deleterrs r� r� r� r� r� r� r� r� r� r� r* r* r* r+ rS � sP � "� � rS c @ sb e Zd ZU dZdZded< ded< ejdd�d d ��Zejd d�d d ��Z ddddd�dd�Z dS )�ManagesConnectiona3 Common base for the two connection-management interfaces :class:`.PoolProxiedConnection` and :class:`.ConnectionPoolEntry`. These two objects are typically exposed in the public facing API via the connection pool event hooks, documented at :class:`.PoolEvents`. .. versionadded:: 2.0 r* �Optional[DBAPIConnection]r6 � Optional[Any]�driver_connectionr rJ c C s t � �dS )a� Info dictionary associated with the underlying DBAPI connection referred to by this :class:`.ManagesConnection` instance, allowing user-defined data to be associated with the connection. The data in this dictionary is persistent for the lifespan of the DBAPI connection itself, including across pool checkins and checkouts. When the connection is invalidated and replaced with a new one, this dictionary is cleared. For a :class:`.PoolProxiedConnection` instance that's not associated with a :class:`.ConnectionPoolEntry`, such as if it were detached, the attribute returns a dictionary that is local to that :class:`.ConnectionPoolEntry`. Therefore the :attr:`.ManagesConnection.info` attribute will always provide a Python dictionary. .. seealso:: :attr:`.ManagesConnection.record_info` NrA rK r* r* r+ �info s zManagesConnection.info�Optional[_InfoType]c C s t � �dS )a� Persistent info dictionary associated with this :class:`.ManagesConnection`. Unlike the :attr:`.ManagesConnection.info` dictionary, the lifespan of this dictionary is that of the :class:`.ConnectionPoolEntry` which owns it; therefore this dictionary will persist across reconnects and connection invalidation for a particular entry in the connection pool. For a :class:`.PoolProxiedConnection` instance that's not associated with a :class:`.ConnectionPoolEntry`, such as if it were detached, the attribute returns None. Contrast to the :attr:`.ManagesConnection.info` dictionary which is never None. .. seealso:: :attr:`.ManagesConnection.info` NrA rK r* r* r+ � record_info+ s z ManagesConnection.record_infoNFr� r# r5 �r� �softr7 c C s t � �dS )aO Mark the managed connection as invalidated. :param e: an exception object indicating a reason for the invalidation. :param soft: if True, the connection isn't closed; instead, this connection will be recycled on next checkout. .. seealso:: :ref:`pool_connection_invalidation` NrA �r: r� r� r* r* r+ r� C s z ManagesConnection.invalidate)NF) r$ r% r& r' r( r) r �ro_memoized_propertyr� r� r� r* r* r* r+ r� � s �r� c @ s4 e Zd ZdZdZedd�dd��Zdd�dd �Zd S ) rP a Interface for the object that maintains an individual database connection on behalf of a :class:`_pool.Pool` instance. The :class:`.ConnectionPoolEntry` object represents the long term maintainance of a particular connection for a pool, including expiring or invalidating that connection to have it replaced with a new one, which will continue to be maintained by that same :class:`.ConnectionPoolEntry` instance. Compared to :class:`.PoolProxiedConnection`, which is the short-term, per-checkout connection manager, this object lasts for the lifespan of a particular "slot" within a connection pool. The :class:`.ConnectionPoolEntry` object is mostly visible to public-facing API code when it is delivered to connection pool event hooks, such as :meth:`_events.PoolEvents.connect` and :meth:`_events.PoolEvents.checkout`. .. versionadded:: 2.0 :class:`.ConnectionPoolEntry` provides the public facing interface for the :class:`._ConnectionRecord` internal class. r* r# rJ c C s t � �dS )z3Return True the connection is currently checked outNrA rK r* r* r+ �in_usem s zConnectionPoolEntry.in_user5 c C s t � �dS )zAClose the DBAPI connection managed by this connection pool entry.NrA rK r* r* r+ r> s s zConnectionPoolEntry.closeN)r$ r% r& r' r( r� r� r> r* r* r* r+ rP V s rP c @ st e Zd ZU dZdZded< ded< ded< d ed < dEd dd �dd�Zded< edd�dd��Zee � dd�dd�dd���Z dZ d ed< e j d d�d d ��Ze j d d�d d!��Zed d"d#�d$d%��ZdFd&dd'd(�d)d*�ZdGdd'd+�d,d-�Zedd�d.d/��Zed d�d0d1��Zd'd�d2d3�ZdHd6dd'd7�d8d9�Zd:d�d;d<�Zdd�d=d>�Zd5d?�dd'd@�dAdB�Zd'd�dCdD�Zd4S )Ir� a Maintains a position in a connection pool which references a pooled connection. This is an internal object used by the :class:`_pool.Pool` implementation to provide context management to a DBAPI connection maintained by that :class:`_pool.Pool`. The public facing interface for this class is described by the :class:`.ConnectionPoolEntry` class. See that class for public API details. .. seealso:: :class:`.ConnectionPoolEntry` :class:`.PoolProxiedConnection` )Z__pool� fairy_ref�finalize_callback�freshr� r6 � __weakref__�__dict__z(Deque[Callable[[DBAPIConnection], None]]r� r# r� �'Optional[weakref.ref[_ConnectionFairy]]r� r[ r� TrS )�poolr� c C s6 d| _ d | _d| _d | _|| _|r*| �� t� | _d S )NFr )r� r� r� r6 �_ConnectionRecord__pool�_ConnectionRecord__connectr r� )r: r� r� r* r* r+ rq � s z_ConnectionRecord.__init__r� r6 r� rJ c C s" | j d krd S | jj�| j �S d S r8 )r6 r� rX rE rK r* r* r+ r� � s �z#_ConnectionRecord.driver_connection�2.0zXThe _ConnectionRecord.connection attribute is deprecated; please use 'driver_connection'c C s | j S r8 �r6 rK r* r* r+ rD � s z _ConnectionRecord.connectionr �_soft_invalidate_timer c C s i S r8 r* rK r* r* r+ r� � s z_ConnectionRecord.infor� c C s i S r8 r* rK r* r* r+ r� � s z _ConnectionRecord.record_infor� )r� r7 c s� t rtt��� ��n��� �z ��� }W nF tk rn } z(t�� � �j|dd� W 5 Q R X � W 5 d }~X Y nX �� � � t �|�� �}t � |� ��fdd�� �_ }�t|< � r��j�d|� |S )NF��_fairy_was_createdc s t d k r t d ��| � dd�S d S )NF�r )�_finalize_fairy)�ref�rV r� rQ r* r+ rv � s � �z,_ConnectionRecord.checkout.<locals>.<lambda>z#Connection %r checked out from pool)r r r� r� �get_connectionr� r � safe_reraise�_checkin_failedZ_should_log_debugr� �weakrefr� r� � _strong_ref_connection_recordsr� r� )�clsr� r6 �err�fairyr� r* r� r+ �checkout� s, � �z_ConnectionRecord.checkoutr� r5 )r� r� r7 c C s | j |d� | j|d� d S )N�r� r� )r� �checkin)r: r� r� r* r* r+ r� � s �z!_ConnectionRecord._checkin_failed)r� r7 c C sx | j d kr |r t�d| � d S d | _ | j}| j}| jrT| j�� }|d k r2||� q2|jjrj|j�|| � |� | � d S )Nz Double checkin attempted on %s) r� r �warnr6 r� r� �poprT r� r� )r: r� rD r� � finalizerr* r* r+ r� � s z_ConnectionRecord.checkinc C s | j d k S r8 )r� rK r* r* r+ r� s z_ConnectionRecord.in_usec C s | j S r8 )r� rK r* r* r+ �last_connect_time s z#_ConnectionRecord.last_connect_timec C s | j d k r| �� d S r8 )r6 �_ConnectionRecord__closerK r* r* r+ r> s z_ConnectionRecord.closeNFr� r� c C s� | j d krd S |r(| jj�| j | |� n| jj�| j | |� |d k rj| jj�d|rVdnd| j |jj|� n | jj�d|r|dnd| j � |r�t � � | _ n| j dd� d | _ d S )Nz*%sInvalidate connection %r (reason: %s:%s)zSoft � z%sInvalidate connection %rTr} ) r6 r� rT Zsoft_invalidater� r� r� � __class__r$ r� r� r� r� r* r* r+ r� s4 � � � z _ConnectionRecord.invalidater c C s� d}| j d kr"| j�� | �� n�| jjdkr\t�� | j | jjkr\| jj�d| j � d}nH| jj | jkr�| jj�d| j � d}n"| j | jkr�| jj�d| j � d}|r�| j dd� | j�� | �� | j d k s�t �| j S )NFr] z)Connection %r exceeded timeout; recyclingTz=Connection %r invalidated due to pool invalidation; recyclingzCConnection %r invalidated due to local soft invalidation; recyclingr} ) r6 r� �clearr� r� rj r� r� r� r\ r� r� �AssertionError)r: ra r* r* r+ r� 3 s> ���� � z _ConnectionRecord.get_connectionc C s$ | j d kp"| jj| jkp"| j| jkS r8 )r6 r� r\ r� r� rK r* r* r+ � _is_hard_or_soft_invalidatedg s � �z._ConnectionRecord._is_hard_or_soft_invalidatedr} )r~ r7 c C sP | j �� | jjjr&| jj�| j| � | jd k s4t�| jj| j|d� d | _d S )Nr} )r� r� r� rT r> r6 r� r� )r: r~ r* r* r+ �__closen s �z_ConnectionRecord.__closec C s� | j }d | _z2t�� | _|�| � | _}|j�d|� d| _W nD tk r� } z&t � � � |j�d|� W 5 Q R X W 5 d }~X Y n>X |j j r�|j j � |j ��| j| � |j j� |j ��| j| � d S )NzCreated new connection %rTzError on connect(): %s)r� r6 r� r� rZ r� r� r� r� r r� rT Z first_connectZ for_modifyZexec_once_unless_exceptionr� Z_exec_w_sync_on_first_run)r: r� rD r� r* r* r+ Z __connectx s. *� �� �z_ConnectionRecord.__connect)T)T)T)NF) r$ r% r& r' r( r) rq r� r� r � deprecatedrD r� r� r� r� � classmethodr� r� r� r� r� r> r� r� r� r� r� r* r* r* r+ r� x sL � $ � �!4 r� Fr� �Optional[_ConnectionRecord]r� zOptional[log._EchoFlagType]r# �Optional[_ConnectionFairy]r5 )r6 rx r� r� rV r r� r7 c C sD |dk }|rR|dk st �t�|d� |dk s0t �|j|k r>dS | dksJt �|j} n|rht�t�|�d� |jj}|r�|dkp~|} | } |jj p�|jj } |jjo�|jj } n|dk} d } } d} | dk �r |r�|r�|j � d| � z�z�|�s|dk s�t �t|| ||�}|j| k�st �|j||| | d � | �rh|�r>||_|�� | �rh|jj�rZ|j�| � |j| | d � W nT tk �r� } z4|j j d dd � |�r�|j|d� t|t��s�� W 5 d}~X Y nX W 5 | �r|�r|�rd| �d| �s�dnd� d �} |j � | � t �| � X |�r$|jdk �r$|�� |dk �r:d|_d|_~ ~~dS )ac Cleanup for a :class:`._ConnectionFairy` whether or not it's already been garbage collected. When using an async dialect no IO can happen here (without using a dedicated thread), since this is called outside the greenlet context and with an already running loop. In this case function will only log a message and raise a warning. NTFz$Connection %r being returned to poolzFThe garbage collector is trying to clean up non-checked-in connection z, which will be z*dropped, as it cannot be safely terminatedZ terminatedz�. Please ensure that SQLAlchemy pooled connections are returned to the pool explicitly, either by calling ``close()`` or by using appropriate context managers to manage their lifecycle.r r} z!Exception during reset or similarr� r� ) r� r� r� r� r6 r� r� rX rF rG r� r� r� r r� r� �_reset�_pool�detachrT Zclose_detachedr� r� r� r� r� r� r� )r6 rx r� r� rV r r� Z is_gc_cleanupZdont_restore_gcedr� Zcan_manipulate_connectionZ!can_close_or_terminate_connectionZ requires_terminate_for_close�messager� r* r* r+ r� � s� �� � �� � � ��� r� z6Dict[weakref.ref[_ConnectionFairy], _ConnectionRecord]r� c @ s� e Zd ZdZdZejr@dd�dd�Zdd�dd �Zdd�d d �Z e d d�d d��Z e d d�dd��Z dd�dd�Z dd�dd�ZdS )r4 a A connection-like adapter for a :pep:`249` DBAPI connection, which includes additional methods specific to the :class:`.Pool` implementation. :class:`.PoolProxiedConnection` is the public-facing interface for the internal :class:`._ConnectionFairy` implementation object; users familiar with :class:`._ConnectionFairy` can consider this object to be equivalent. .. versionadded:: 2.0 :class:`.PoolProxiedConnection` provides the public- facing interface for the :class:`._ConnectionFairy` internal class. r* r5 rJ c C s d S r8 r* rK r* r* r+ r1 3 rM z PoolProxiedConnection.commitr c C s d S r8 r* rK r* r* r+ �cursor5 rM z PoolProxiedConnection.cursorc C s d S r8 r* rK r* r* r+ r2 7 rM z PoolProxiedConnection.rollbackr# c C s t � �dS )zgReturn True if this :class:`.PoolProxiedConnection` still refers to an active DBAPI connection.NrA rK r* r* r+ r� 9 s z PoolProxiedConnection.is_validc C s t � �dS )zVReturn True if this :class:`.PoolProxiedConnection` is detached from its pool.NrA rK r* r* r+ � is_detached@ s z!PoolProxiedConnection.is_detachedc C s t � �dS )a Separate this connection from its Pool. This means that the connection will no longer be returned to the pool when closed, and will instead be literally closed. The associated :class:`.ConnectionPoolEntry` is de-associated from this DBAPI connection. Note that any overall connection limiting constraints imposed by a Pool implementation may be violated after a detach, as the detached connection is removed from the pool's knowledge and control. NrA rK r* r* r+ r� G s z PoolProxiedConnection.detachc C s t � �dS )a$ Release this connection back to the pool. The :meth:`.PoolProxiedConnection.close` method shadows the :pep:`249` ``.close()`` method, altering its behavior to instead :term:`release` the proxied connection back to the connection pool. Upon release to the pool, whether the connection stays "opened" and pooled in the Python process, versus actually closed out and removed from the Python process, is based on the pool implementation in use and its configuration and current state. NrA rK r* r* r+ r> W s zPoolProxiedConnection.closeN)r$ r% r& r' r( �typingr r1 r� r2 r� r� r� r� r> r* r* r* r+ r4 " s r4 c @ s� e Zd ZU dZdZded< ded< ddd�dd �Zed d �d d ��Zedd �dd��Z edd �dd��Z d$dddd�dd�Z e j dd �dd ��Zd d d d �d d �Zd d d!�d"d#�ZdS )%�_AdhocProxiedConnectionad provides the :class:`.PoolProxiedConnection` interface for cases where the DBAPI connection is not actually proxied. This is used by the engine internals to pass a consistent :class:`.PoolProxiedConnection` object to consuming dialects in response to pool events that may not always have the :class:`._ConnectionFairy` available. �r6 r� � _is_validr r6 rP r� )r6 rx c C s || _ || _d| _d S )NTr� )r: r6 rx r* r* r+ rq w s z _AdhocProxiedConnection.__init__r rJ c C s | j jS r8 �r� r� rK r* r* r+ r� � s z)_AdhocProxiedConnection.driver_connectionc C s | j S r8 r� rK r* r* r+ rD � s z"_AdhocProxiedConnection.connectionr# c C s | j S )z�Implement is_valid state attribute. for the adhoc proxied connection it's assumed the connection is valid as there is no "invalidate" routine. �r� rK r* r* r+ r� � s z _AdhocProxiedConnection.is_validNFr� r5 r� c C s d| _ d S )NFr� r� r* r* r+ r� � s z"_AdhocProxiedConnection.invalidater� c C s | j jS r8 �r� r� rK r* r* r+ r� � s z#_AdhocProxiedConnection.record_infor ��args�kwargsr7 c O s | j j||�S r8 )r6 r� �r: r� r� r* r* r+ r� � s z _AdhocProxiedConnection.cursor��keyr7 c C s t | j|�S r8 �r� r6 �r: r� r* r* r+ � __getattr__� s z#_AdhocProxiedConnection.__getattr__)NF)r$ r% r& r' r( r) rq r� r� rD r� r� r �ro_non_memoized_propertyr� r� r� r* r* r* r+ r� g s$ �r� c @ s� e Zd ZU dZdZded< ded< ded< ddd dd �d d �Zd ed< edd�dd��Zee � dd�dd�dd���Z e dKdddd d�dd ��Z d d�d d �ZdLd d!d"�d#d$�Zd!d�d%d&�Zdd d d d!d'�d(d)�Zed*d�d+d,��Zed d�d-d.��Zed d�d/d0��Ze jd1d�d2d3��Ze jd4d�d5d6��ZdMd7d d!d8�d9d:�Zd;d;d<d=�d>d?�Zd@d;dA�dBdC�Zd!d�dDdE�Z d!d�dFdG�Z dNd d!dH�dIdJ�Z dS )Or� a& Proxies a DBAPI connection and provides return-on-dereference support. This is an internal object used by the :class:`_pool.Pool` implementation to provide context management to a DBAPI connection delivered by that :class:`_pool.Pool`. The public facing interface for this class is described by the :class:`.PoolProxiedConnection` class. See that class for public API details. The name "fairy" is inspired by the fact that the :class:`._ConnectionFairy` object's lifespan is transitory, as it lasts only for the length of a specific DBAPI connection being checked out from the pool, and additionally that as a transparent proxy, it is mostly invisible. .. seealso:: :class:`.PoolProxiedConnection` :class:`.ConnectionPoolEntry` )r6 r� �_echor� �_counterr� r� rS r� r r6 rU r� r� )r� r6 rx rV c C s" || _ d| _|| _|| _|| _d S )Nr )r� r� r6 r� r� )r: r� r6 rx rV r* r* r+ rq � s z_ConnectionFairy.__init__r� r� r� rJ c C s | j d krd S | j jS r8 r� rK r* r* r+ r� � s z"_ConnectionFairy.driver_connectionr� zWThe _ConnectionFairy.connection attribute is deprecated; please use 'driver_connection'c C s | j S r8 r� rK r* r* r+ rD � s z_ConnectionFairy.connectionNzOptional[threading.local]r� )r� � threadconnsr� r7 c C sl |s"t �|�}|d k r"t�|�|_|jd k s4td��|jd k sFtd��| jd7 _|j jsb|j rl|jdkrp|S d}|dk�rJ|jj }d|j_ z�|j r�|s�|j r�|j �d|j� |j�|j�}|s�|j r�|j �d|j� t�� �n|j r�|j �d |j� |j �|j|j|� |W S tjk �r� } z�|j�rZ|j �d |� |j�|� |j||dd � n |j �d |j|� |j�|� z|j�� |_W nL tk �r� } z,t�� � |jj |d d� ~W 5 Q R X � W 5 d }~X Y nX |d8 }W 5 d }~X Y qt tk �rF } z:t�� �& |j} | d k �r(| j | d d� ~W 5 Q R X � W 5 d } ~ X Y qtX qt|j �d� |�� t� d��d S )Nz,can't 'checkout' a detached connection fairyz0can't 'checkout' an invalidated connection fairyr- r r Fz Pool pre-ping on connection %sz;Pool pre-ping on connection %s failed, will invalidate poolz)Connection %s is fresh, skipping pre-pingzoDisconnection detected on checkout, invalidating all pooled connections prior to current timestamp (reason: %r)�r� zVDisconnection detected on checkout, invalidating individual connection %s (reason: %r)Tr� z+Reconnection attempts exhausted on checkoutzThis connection is closed) r� r� r� r� �currentr� r� r6 r� rT rk r� r� r� r� rX rC r ZInvalidatePoolErrorZDisconnectionErrorZinvalidate_poolr� r� r� r� r� r r� r� ZInvalidRequestError) r� r� r� r� ZattemptsZconnection_is_fresh�resultr� r� Zbe_outerrQ r* r* r+ r� � s� ������� ��� � �� � � � � z_ConnectionFairy._checkoutc C s t j| j| d�S )N)r� )r� r� r� rK r* r* r+ �_checkout_existinge s z#_ConnectionFairy._checkout_existingFr# r5 )r r7 c C s" t | j| j| jd | j|| d� d S )N)r r� )r� r6 r� r� r� )r: r r* r* r+ r� h s �z_ConnectionFairy._checkinc C s | � � d S r8 r� rK r* r* r+ �_closes s z_ConnectionFairy._close)r� r r! r" r7 c C s� |j jr&|j �| j| jt|||d�� |s.d S |jtkrx|rT| jrv|j� d| j� q�| jrj|j� d| j� |j � | � n,|jt kr�| jr�|j� d| j� |j � | � d S )Nr z.Connection %s reset, transaction already resetz Connection %s rollback-on-returnz Connection %s commit-on-return)rT �resetr6 r� r rl r. r� r� r� rX r; r/ r< )r: r� r r! r" r* r* r+ r� v s@ �� �� �z_ConnectionFairy._resetzlog._IdentifiedLoggerTypec C s | j jS r8 )r� r� rK r* r* r+ �_logger� s z_ConnectionFairy._loggerc C s | j d k S r8 r� rK r* r* r+ r� � s z_ConnectionFairy.is_validc C s | j d kS r8 )r� rK r* r* r+ r� � s z _ConnectionFairy.is_detachedr c C s | j d kri S | j jS d S r8 )r� r� rK r* r* r+ r� � s z_ConnectionFairy.infor� c C s | j d krd S | j jS d S r8 r� rK r* r* r+ r� � s z _ConnectionFairy.record_infor� r� c C sD | j d krt�d� d S | jr.| jj||d� |s@d | _ | �� d S )Nz.Can't invalidate an already-closed connection.)r� r� )r6 r r� r� r� r� r� r* r* r+ r� � s z_ConnectionFairy.invalidater r r� c O s | j d k st�| j j||�S r8 )r6 r� r� r� r* r* r+ r� � s z_ConnectionFairy.cursorr� r� c C s t | j|�S r8 r� r� r* r* r+ r� � s z _ConnectionFairy.__getattr__c C s\ | j d k rX| j }d |_d |_| j�| j � | j�� | _d | _ | jjjrX| jj�| j|� d S r8 ) r� r� r6 r� r� r� �copyrT r� rR r* r* r+ r� � s z_ConnectionFairy.detachc C s$ | j d8 _ | j dkr | �� d S )Nr- r �r� r� rK r* r* r+ r> � s z_ConnectionFairy.close)�transaction_resetr7 c C s( | j d8 _ | j dkr$| j|d� d S )Nr- r r� r� )r: r r* r* r+ �_close_special� s z _ConnectionFairy._close_special)NN)F)NF)F) r$ r% r& r' r( r) rq r� r� r r� rD r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r> r r* r* r* r+ r� � sR � �} + �r� )FN)?r' � __future__r � collectionsr � dataclasses�enumr � threadingr� r� r r r r r r r r r r r� r� r r r r Z util.typingr r Zengine.interfacesr r r r r r Z sql._typingr � dataclassr r, r_ �listr. r/ r0 r3 rH rI rO Z IdentifiedZ EventTargetrS r� rP r� r� r� r) r4 r� r� r* r* r* r+ �<module> s| ,��# ?|" ' � � E;