U
|
¸ý°dRG ã @ s´ d Z ddlmZ ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm
|
Z
|
dd lm Z dd
|
lm Z dd lm Z ddlZd d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddl m Z ej r@ddl m!Z! G d d d eZ"G d d d e"Z#G d d! d!e#Z$G d"d# d#eZ%G d$d% d%eZ&G d&d' d'eZ'G d(d) d)eZ(dS )*z Pool implementation classes.
|
|
é )Ú annotationsN)ÚAny)Úcast)ÚList)ÚOptional)ÚSet)ÚType)Ú TYPE_CHECKING)ÚUnioné )Ú_AsyncConnDialect)Ú_ConnectionFairy©Ú_ConnectionRecord)Ú_CreatorFnType)Ú_CreatorWRecFnType)ÚConnectionPoolEntry)ÚPool)ÚPoolProxiedConnectioné )Úexc)Úutil)Úchop_traceback)Úqueue)ÚLiteral)ÚDBAPIConnectionc @ sð e Zd ZU dZdZejZded< ded< d1d
|
d d d d ddddZ dddddZ
|
ddddZ d dddZ d dd d Z d dd d Zddd!d"Zd#dd$d%Zd dd&d'Zd dd(d)Zd dd*d+Zd dd,d-Zd dd.d/Zd0S )2Ú QueuePoolzíA :class:`_pool.Pool`
|
that imposes a limit on the number of open connections.
|
|
:class:`.QueuePool` is the default pooling implementation used for
|
all :class:`_engine.Engine` objects, unless the SQLite dialect is in use.
|
|
Fú1Type[sqla_queue.QueueCommon[ConnectionPoolEntry]]Ú _queue_classz+sqla_queue.QueueCommon[ConnectionPoolEntry]Ú_poolé é
|
ç >@ú)Union[_CreatorFnType, _CreatorWRecFnType]ÚintÚfloatÚboolr )ÚcreatorÚ pool_sizeÚ max_overflowÚtimeoutÚuse_lifoÚkwc K sP t j| |f| | j||d| _d| | _|dkr6dn|| _|| _t ¡ | _ dS )aÙ
|
Construct a QueuePool.
|
|
:param creator: a callable function that returns a DB-API
|
connection object, same as that of :paramref:`_pool.Pool.creator`.
|
|
:param pool_size: The size of the pool to be maintained,
|
defaults to 5. This is the largest number of connections that
|
will be kept persistently in the pool. Note that the pool
|
begins with no connections; once this number of connections
|
is requested, that number of connections will remain.
|
``pool_size`` can be set to 0 to indicate no size limit; to
|
disable pooling, use a :class:`~sqlalchemy.pool.NullPool`
|
instead.
|
|
:param max_overflow: The maximum overflow size of the
|
pool. When the number of checked-out connections reaches the
|
size set in pool_size, additional connections will be
|
returned up to this limit. When those additional connections
|
are returned to the pool, they are disconnected and
|
discarded. It follows then that the total number of
|
simultaneous connections the pool will allow is pool_size +
|
`max_overflow`, and the total number of "sleeping"
|
connections the pool will allow is pool_size. `max_overflow`
|
can be set to -1 to indicate no overflow limit; no limit
|
will be placed on the total number of concurrent
|
connections. Defaults to 10.
|
|
:param timeout: The number of seconds to wait before giving up
|
on returning a connection. Defaults to 30.0. This can be a float
|
but is subject to the limitations of Python time functions which
|
may not be reliable in the tens of milliseconds.
|
|
:param use_lifo: use LIFO (last-in-first-out) when retrieving
|
connections instead of FIFO (first-in-first-out). Using LIFO, a
|
server-side timeout scheme can reduce the number of connections used
|
during non-peak periods of use. When planning for server-side
|
timeouts, ensure that a recycle or pre-ping strategy is in use to
|
gracefully handle stale connections.
|
|
.. versionadded:: 1.3
|
|
.. seealso::
|
|
:ref:`pool_use_lifo`
|
|
:ref:`pool_disconnects`
|
|
:param \**kw: Other keyword arguments including
|
:paramref:`_pool.Pool.recycle`, :paramref:`_pool.Pool.echo`,
|
:paramref:`_pool.Pool.reset_on_return` and others are passed to the
|
:class:`_pool.Pool` constructor.
|
|
)r+ r éÿÿÿÿN)
|
r Ú__init__r r Ú _overflowÚ _max_overflowÚ_timeoutÚ threadingÚLockÚ_overflow_lock)Úselfr' r( r) r* r+ r, © r6 úKd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/pool/impl.pyr. ? s ?
|
zQueuePool.__init__r ÚNone©ÚrecordÚreturnc C sF z| j |d¡ W n. tjk
|
r@ z | ¡ W 5 | ¡ X Y nX d S ©NF)r ÚputÚ
|
sqla_queueÚFullÚ _dec_overflowÚclose©r5 r: r6 r6 r7 Ú_do_return_conn
s zQueuePool._do_return_conn©r; c C sÎ | j dk}|o| j| j k}z| j || j¡W S tjk
|
rB Y nX |r| j| j kr|s`| ¡ S tj d|
|
¡ | ¡ | jf dd| ¡ rÂz
|
| ¡ W S t ¡ | ¡ W 5 Q R X Y qÊX n| ¡ S d S )Nr- zSQueuePool limit of size %d overflow %d reached, connection timed out, timeout %0.2fZ3o7r)Úcode)r0 r/ r Úgetr1 r> ÚEmptyÚ_do_getr Ú TimeoutErrorÚsizeÚoverflowÚ _inc_overflowÚ_create_connectionr Z safe_reraiser@ )r5 Z use_overflowÚwaitr6 r6 r7 rH s.
|
þü
|
|
|
zQueuePool._do_getc C sh | j dkr | jd7 _dS | j< | j| j k rL| jd7 _W 5 Q R £ dS W 5 Q R £ dS W 5 Q R X d S )Nr- r TF©r0 r/ r4 ©r5 r6 r6 r7 rL ® s
|
zQueuePool._inc_overflowz Literal[True]c C sJ | j dkr | jd8 _dS | j | jd8 _W 5 Q R £ dS Q R X d S )Nr- r TrO rP r6 r6 r7 r@ ¹ s
|
zQueuePool._dec_overflowc C sJ | j d¡ | j| j| jj| j| j| jj| j | j
|
| j | j | j | j| jd S )NúPool recreating) r( r) Úpre_pingr+ r* ÚrecycleÚechoÚ logging_nameÚreset_on_returnÚ _dispatchÚdialect)ÚloggerÚinfoÚ __class__Ú_creatorr Úmaxsizer0 Ú _pre_pingr+ r1 Ú_recyclerT Ú_orig_logging_nameÚ_reset_on_returnÚdispatchÚ_dialectrP r6 r6 r7 ÚrecreateÁ s ôzQueuePool.recreatec C sZ z| j d¡}| ¡ W q tjk
|
r2 Y q6Y q X q d| ¡ | _| j d| ¡ ¡ d S )NFr zPool disposed. %s)
|
r rF rA r> rG rJ r/ rY rZ Ústatus©r5 Úconnr6 r6 r7 ÚdisposeÒ s zQueuePool.disposeÚstrc C s d| ¡ | ¡ | ¡ | ¡ f S )Nz_Pool size: %d Connections in pool: %d Current Overflow: %d Current Checked out connections: %d)rJ Ú checkedinrK Ú
|
checkedoutrP r6 r6 r7 re Ý s üýÿzQueuePool.statusc C s | j jS ©N)r r] rP r6 r6 r7 rJ ê s zQueuePool.sizec C s | j S rl )r1 rP r6 r6 r7 r* í s zQueuePool.timeoutc C s
|
| j ¡ S rl )r ÚqsizerP r6 r6 r7 rj ð s zQueuePool.checkedinc C s | j jr| jS dS )Nr )r r] r/ rP r6 r6 r7 rK ó s zQueuePool.overflowc C s | j j| j ¡ | j S rl )r r] rm r/ rP r6 r6 r7 rk ö s zQueuePool.checkedoutN)r r! r" F)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__Ú _is_asyncior> ÚQueuer Ú__annotations__r. rC rH rL r@ rd rh re rJ r* rj rK rk r6 r6 r6 r7 r - s,
|
þ
|
úF r c @ s& e Zd ZU dZejZded< e Z dS )ÚAsyncAdaptedQueuePoolTr r N)
|
rn ro rp rr r> ZAsyncAdaptedQueuer rt r rc r6 r6 r6 r7 ru ú s
|
þ
|
ru c @ s e Zd ZejZdS )Ú FallbackAsyncAdaptedQueuePoolN)rn ro rp r> ZFallbackAsyncAdaptedQueuer r6 r6 r6 r7 rv s rv c @ sX e Zd ZdZddddZdddd d
|
Zddd d Zd dd dZddddZdS )ÚNullPoolaQ A Pool which does not pool connections.
|
|
Instead it literally opens and closes the underlying DB-API connection
|
per each connection open/close.
|
|
Reconnect-related functions such as ``recycle`` and connection
|
invalidation are not supported by this Pool implementation, since
|
no connections are held persistently.
|
|
ri rD c C s dS )Nrw r6 rP r6 r6 r7 re s zNullPool.statusr r8 r9 c C s | ¡ d S rl )rA rB r6 r6 r7 rC s zNullPool._do_return_connc C s | ¡ S rl )rM rP r6 r6 r7 rH s zNullPool._do_getc
|
C s6 | j d¡ | j| j| j| j| j| j| j| j | j
|
dS )NrQ )rS rT rU rV rR rW rX ) rY rZ r[ r\ r_ rT r` ra r^ rb rc rP r6 r6 r7 rd s øzNullPool.recreatec C s d S rl r6 rP r6 r6 r7 rh + s zNullPool.disposeN) rn ro rp rq re rC rH rd rh r6 r6 r6 r7 rw s rw c @ s e Zd ZdZdZd dddddd Zd d
|
d d Zd d
|
ddZd d
|
ddZdd
|
ddZ dd dddZ
|
dd
|
ddZ dd
|
d d Z d S ) ÚSingletonThreadPoola© A Pool that maintains one connection per thread.
|
|
Maintains one connection per each thread, never moving a connection to a
|
thread other than the one which it was created in.
|
|
.. warning:: the :class:`.SingletonThreadPool` will call ``.close()``
|
on arbitrary connections that exist beyond the size setting of
|
``pool_size``, e.g. if more unique **thread identities**
|
than what ``pool_size`` states are used. This cleanup is
|
non-deterministic and not sensitive to whether or not the connections
|
linked to those thread identities are currently in use.
|
|
:class:`.SingletonThreadPool` may be improved in a future release,
|
however in its current status it is generally used only for test
|
scenarios using a SQLite ``:memory:`` database and is not recommended
|
for production use.
|
|
|
Options are the same as those of :class:`_pool.Pool`, as well as:
|
|
:param pool_size: The number of threads in which to maintain connections
|
at once. Defaults to five.
|
|
:class:`.SingletonThreadPool` is used by the SQLite dialect
|
automatically when a memory-based database is used.
|
See :ref:`sqlite_toplevel`.
|
|
Fr r# r$ r )r' r( r, c K s6 t j| |f| t ¡ | _t ¡ | _t | _|| _d S rl ) r r. r2 ÚlocalÚ_connÚ_fairyÚsetÚ
|
_all_connsrJ )r5 r' r( r, r6 r6 r7 r. P s
|
|
|
z SingletonThreadPool.__init__rD c C s: | j d¡ | j| j| j| j| j| j| j| j | j
|
| j d S )NrQ )r( rS rT rR rU rV rW rX ) rY rZ r[ r\ rJ r_ rT r^ r` ra rb rc rP r6 r6 r7 rd \ s ÷z SingletonThreadPool.recreater8 c C s< | j D ]&}z | ¡ W q tk
|
r* Y qX q| j ¡ dS )zDispose of this pool.N)r} rA Ú ExceptionÚclearrf r6 r6 r7 rh j s
|
zSingletonThreadPool.disposec C s( t | j| jkr$| j ¡ }| ¡ q d S rl )Úlenr} rJ ÚpoprA ©r5 Úcr6 r6 r7 Ú_cleanupw s
|
z SingletonThreadPool._cleanupri c C s dt | t| jf S )Nz"SingletonThreadPool id:%d size: %d)Úidr r} rP r6 r6 r7 re | s þzSingletonThreadPool.statusr r9 c C s$ z
|
| j `W n tk
|
r Y nX d S rl )r{ ÚcurrentÚAttributeErrorrB r6 r6 r7 rC s
|
z#SingletonThreadPool._do_return_connc C s z.t rtt| j ¡ }n
|
| j ¡ }|r,|W S W n tk
|
rB Y nX | ¡ }t |¡| j_t | j
|
| j krr| ¡ | j
|
|¡ |S rl )r r r rz r r rM ÚweakrefÚrefr r} rJ r Úaddr r6 r6 r7 rH s
|
|
zSingletonThreadPool._do_getr c C sH zt t| j ¡ }W n tk
|
r( Y nX |d k r:| ¡ S t | | j¡S rl )r r r{ r r Z_checkout_existingZ _checkout©r5 Úrecr6 r6 r7 Úconnect s zSingletonThreadPool.connectN)r ) rn ro rp rq rr r. rd rh r re rC rH r r6 r6 r6 r7 rx / s ý rx c @ s e Zd ZdZejddddZddddZd dd
|
d Zd dd d Z d d dddZ
|
ddddZ dd dddZ ddddZ dS )Ú
|
StaticPoola A Pool of exactly one connection, used for all requests.
|
|
Reconnect-related functions such as ``recycle`` and connection
|
invalidation (which is also used to support auto-reconnect) are only
|
partially supported right now and may not yield good results.
|
|
|
r rD c C s t | S rl r rP r6 r6 r7 Ú
|
connection² s zStaticPool.connectionri c C s dS )Nr r6 rP r6 r6 r7 re ¶ s zStaticPool.statusr8 c C s, d| j kr(| jjd k r(| j ¡ | j d= d S ©Nr )Ú__dict__r Údbapi_connectionrA rP r6 r6 r7 rh ¹ s ÿ
|
þ
|
zStaticPool.disposec
|
C s6 | j d¡ | j| j| j| j| j| j| j| j | j
|
dS )NrQ )r' rS rV rR rT rU rW rX ) rY rZ r[ r\ r_ ra r^ rT r` rb rc rP r6 r6 r7 rd Á s øzStaticPool.recreate)Úother_static_poolr; c s ddd fdd }|| _ d S )Nr r )r r; c s j j}|d k st|S rl )r r ÚAssertionError)r rg ©r r6 r7 r' Ñ s z*StaticPool._transfer_from.<locals>.creator)Z_invoke_creator)r5 r r' r6 r r7 Ú_transfer_fromÎ s zStaticPool._transfer_fromr c C s
|
t d S rl )ÚNotImplementedErrorrP r6 r6 r7 rM Ø s z StaticPool._create_connectionr9 c C s d S rl r6 rB r6 r6 r7 rC Û s zStaticPool._do_return_connc C s | j }| ¡ r | jd= | j }|S r )r Z _is_hard_or_soft_invalidatedr r r6 r6 r7 rH Þ s
|
zStaticPool._do_getN)rn ro rp rq r Zmemoized_propertyr re rh rd r rM rC rH r6 r6 r6 r7 r § s
|
r c @ sz e Zd ZU dZded< ded< ddddd Zd
|
d d d ZdddddZdd ddZd d ddZ dd ddZ
|
dS )Ú AssertionPoola A :class:`_pool.Pool` that allows at most one checked out connection at
|
any given time.
|
|
This will raise an exception if more than one connection is checked out
|
at a time. Useful for debugging code that is using more connections
|
than desired.
|
|
z Optional[ConnectionPoolEntry]rz zOptional[List[str]]Ú_checkout_tracebackr )Úargsr, c O s6 d | _ d| _| dd¡| _d | _tj| f|| d S )NFZstore_tracebackT)rz Ú _checked_outr Ú_store_tracebackr r r. )r5 r r, r6 r6 r7 r. õ s
|
zAssertionPool.__init__ri rD c C s dS )Nr r6 rP r6 r6 r7 re ü s zAssertionPool.statusr r8 r9 c C s& | j stdd| _ || jks"td S )Nz connection is not checked outF)r r rz rB r6 r6 r7 rC ÿ s z AssertionPool._do_return_connc C s d| _ | jr| j ¡ d S r< )r rz rA rP r6 r6 r7 rh s zAssertionPool.disposec
|
C s6 | j d¡ | j| j| j| j| j| j| j| j | j
|
dS )NrQ )rT rR rS rV rU rW rX ) rY rZ r[ r\ rT r^ r_ ra r` rb rc rP r6 r6 r7 rd
|
s øzAssertionPool.recreatec C s^ | j r2| jr"dd t| j¡ }nd}td| | jsB| ¡ | _d| _ | jrXt ¡ | _| jS )Nz at:
|
%sÚ z!connection is already checked outT)
|
r r Újoinr r rz rM r Ú tracebackÚ format_stack)r5 Úsuffixr6 r6 r7 rH s ÿ
|
|
zAssertionPool._do_getN) rn ro rp rq rt r. re rC rh rd rH r6 r6 r6 r7 r ç s
|
r ))rq Ú
|
__future__r r2 r Útypingr r r r r r r r
|
r Úbaser r r r r r r r r r r r r r> Z util.typingr Zengine.interfacesr r ru rv rw rx r r r6 r6 r6 r7 Ú<module> sH N (x@
|
