o FDi9$@sddlZddlZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZddlmZdd lmZeeZGd d d ZdS) N)deque) AsyncIterator)asynccontextmanagersuppress)Optional) Connection)AsynchPoolError) constants) PoolStatusc@seZdZejejfdedefddZd6ddZd7d d Z de fd d Z e de fd dZe de fddZe de fddZe defddZe defddZe defddZe defddZe defddZd7ddZdefdd Zdeefd!d"Zdefd#d$Zd%eddfd&d'Zd(d)d*ed+e ddfd,d-Zd(d)d+e ddfd.d/Zede efd0d1Z!d6d2d3Z"d7d4d5Z#dS)8PoolminsizemaxsizecKs|dkrtd|dkrtd||krtd||_||_||_t||_t|_t |d|_ t |d|_ d|_ d|_ dS)Nz+maxsize is expected to be greater than zerorz2minsize is expected to be greater or equal to zerozminsize is greater than maxsize)maxlenF) ValueError_maxsize_minsize_connection_kwargsasyncio Semaphore_semLock_lockr_acquired_connections_free_connections_opened_closed)selfr r kwargsrD/var/www/Datamplify/venv/lib/python3.10/site-packages/asynch/pool.py__init__s     z Pool.__init__returncs|IdH|SN)startuprrrr __aenter__'szPool.__aenter__Ncs|IdHdSr#)shutdown)rexc_typeexc_valexc_tbrrr __aexit__+szPool.__aexit__c Cs>t|j}|j}d|d|jd|jdt|dd|d S)N)type__name__statusrrid)rcls_namer1rrr __repr__.s z Pool.__repr__cC|jS)ziReturns True if the pool is opened. :returns: the pool open status :rtype: bool )rr%rrr opened6z Pool.openedcCr5)ziReturn True if the pool is closed. :returns: the pool close status :rtype: bool )rr%rrr closed@r7z Pool.closedcCsD|js |js tjS|jr|jstjS|jr|jstjSt|d)zReturn the status of the pool. :raise AsynchPoolError: an unresolved pool state. :return: the Pool object status :rtype: str (PoolStatus StrEnum) z is in an unknown state)rrr createdr6r8rr%rrr r1Js   z Pool.statuscC t|jS)zReturn the number of connections acquired from the pool. A connection is acquired when the `pool.connection()` is invoked. :return: the number of connections requested from the pool :rtype: int )lenrr%rrr acquired_connections[s zPool.acquired_connectionscCr:)zReturn the number of free connections in the pool. :return: the number of free connections in the pool :rtype: int )r;rr%rrr free_connectionsgs zPool.free_connectionscCs |j|jS)amReturn the number of connections associated with the pool. This number is the sum of the acquired and free connections. So this sum may be interpreted as the current size of the pool or the number of connections associated with the pool and so on. :return: the number of connections related to the pool :rtype: int )r<r=r%rrr _pool_sizeqs zPool._pool_sizecCr5r#)rr%rrr r z Pool.maxsizecCr5r#)rr%rrr r r?z Pool.minsizec s|j|jkrt|d|j|jkrt|dtdi|j}|IdHz|IdH|j|WdSt yS}z d|d|}t||d}~ww)Nz is already fullz is overburdenzfailed to create a z for r) r>rrrrconnectpingrappendConnectionErrorrconnemsgrrr _create_connections   zPool._create_connectioncCs|js td||jS)Nzno free connection in )rrpopleftr%rrr _pop_connections zPool._pop_connectioncsT|jr(|}tt|IdH|WdS1s wY|jsdSr#)rrJrrC_refreshrrErrr _get_fresh_connections  zPool._get_fresh_connectioncsJ|IdH}r|j||S|IdH|}|j||Sr#)rMrrBrHrJrLrrr _acquire_connections  zPool._acquire_connectionrEc s~||jvrtd|d||j|z |IdHWnty6}z d|d}t||d}~ww|j|dS)Nzthe connection z does not belong to zthe z is invalidated)rrremoverKrCrrBrDrrr _release_connections    zPool._release_connectionFstrictnrRcs|dkrd|d}t|j|jkr-djd|djd}t||s1dSfdd t|D}|r\td d tj|d d iIdHDr^d|d}t|dSdS)Nrz!cannot create a negative number (z) of connections for z has the z connections, adding z will exceed its maxsize ()csg|] }tqSr)r create_taskrH).0_r%rr sz*Pool._init_connections..css|] }t|tr|VqdSr#) isinstance Exception)rVirrr s z)Pool._init_connections..return_exceptionsTzfailed to create the z connection(s) for the )rr>r rrangeanyrgather)rrSrRrGtasksrr%r _init_connectionss.  zPool._init_connectionscs2|j|j}dkr|j||dIdHdSdS)NrrQ)r r>rb)rrRgaprrr _ensure_minsize_connectionssz Pool._ensure_minsize_connectionscCs|j4IdH|j4IdH|IdH}WdIdHn 1IdHs(wYzM|VW|j4IdH6z ||IdHWnty[}z t|WYd}~nd}~ww|jddIdHWdIdHn 1IdHsuwYnI|j4IdH6z ||IdHWnty}z t|WYd}~nd}~ww|jddIdHWdIdHw1IdHswYwWdIdHdS1IdHswYdS)aEGet a connection from the pool. If requested more connections than the pool can provide, the pool gets blocked until a connection comes back. :raises AsynchPoolError: if a connection cannot be acquired or released :return: a free connection from the pool :rtype: Connection NTrQ)rrrNrPrloggerwarningrd)rrErFrrr connections6 ((,.zPool.connectionc s|j4IdH/|jr|WdIdHS|j|jddIdHd|_|jr,d|_WdIdH|S1IdHs=wY|S)zInitialise the pool. When entering the context, the pool get filled with connections up to the pool `minsize` value. :return: a pool object with `minsize` opened connections :rtype: Pool NTrQF)rrrbr rr%rrr r$s   z Pool.startupc s|j4IdH6|jr|j}|IdH|js |jr-|j}|IdH|jsd|_d|_WdIdHdS1IdHsDwYdS)zClose the pool. This method closes consequently free connections first. Then it does the same for the acquired connections. Then the pool is marked closed. NFT)rrrIcloserrrrLrrr r' s  .z Pool.shutdown)r"r )r"N)$r0 __module__ __qualname__r POOL_MIN_SIZE POOL_MAX_SIZEintr!r&r+strr4propertyboolr6r8r1r<r=r>r r rHrrJrrMrNrPrbrdrrrgr$r'rrrr r sL           r )rlogging collectionsrcollections.abcr contextlibrrtypingrasynch.connectionr asynch.errorsr asynch.protor asynch.proto.models.enumsr getLoggerr0rer rrrr s