o ^fu@sdZddlmZddlmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZmZe r8dd lmZGd d d ZGd d d ZdS)z9 psycopg connection pool base class and functionalities. ) monotonicrandom)AnyDictOptionalTuple TYPE_CHECKING)errors) PoolClosed)CounterDeque)BaseConnectionc@seZdZUdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZeded< dBdedeeeefdedeedeededededededefdd Zd!efd"d#Zed!efd$d%Zed!efd&d'Zed!e fd(d)Z!dedeed!e"eeffd*d+Z#dCd-d.Z$dCd/d0Z%dDd3d4Z&d!eeeffd5d6Z'd!eeeffd7d8Z(d!eeeffd9d:Z)e*d;edd?Z+dDd@dAZ,d,S)EBasePoolrpool_minpool_max pool_sizepool_availablerequests_waiting requests_numrequests_queuedrequests_wait_msrequests_errorsusage_ms returns_badconnections_numconnections_msconnections_errorsconnections_lostr_poolconninfokwargsmin_sizemax_sizenametimeout max_waiting max_lifetimemax_idlereconnect_timeout num_workersc Cs|||\}}|stjd} t_d| }| dkrtd||_|p%i|_||_||_||_||_ ||_ | |_ ||_ | |_ | |_||_t|_tt|_||_d|_d|_d|_d|_dS)Nr zpool-znum_workers must be at least 1FT) _check_sizer _num_pool ValueErrorr"r#r& _min_size _max_sizer'r(r+r)r*r,_nconnsrr r str_stats _nconns_min_growing_opened_closed_open_implicit) selfr"r#r$r%r&r'r(r)r*r+r,numr<J/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg_pool/base.py__init__+s2    zBasePool.__init__returnc Cs.d|jjd|jjd|jdt|dd S)N<. z at 0xx>) __class__ __module____name__r&idr:r<r<r=__repr__ds zBasePool.__repr__cC|jSN)r0rIr<r<r=r$jzBasePool.min_sizecCrKrL)r1rIr<r<r=r%nrMzBasePool.max_sizecCrK)z`!True` if the pool is closed.)r8rIr<r<r=closedrszBasePool.closedcCsX|dur|}|dkrtd||krtd||kr#dkr(td||fS||fS)Nrzmin_size cannot be negativez/max_size must be greater or equal than min_sizez3if min_size is 0 max_size must be greater or than 0)r/)r:r$r%r<r<r=r-wszBasePool._check_sizeNcCs|jr |jr tddSdS)Nz8pool has already been opened/closed and cannot be reused)r8r7eOperationalErrorrIr<r<r= _check_opens zBasePool._check_opencCs4|jr|jrtd|jdtd|jddS)Nz the pool z is already closedz is not open yet)r8r7r r&rIr<r<r=_check_open_getconns zBasePool._check_open_getconnconnBaseConnection[Any]cCsJt|dd}||ur dS|rd|j}nd}td|jd|d|)Nr zit comes from pool zit doesn't come from any poolz can't return connection to pool z, z: )getattrr&r/)r:rSpoolmsgr<r<r=_check_pool_putconns zBasePool._check_pool_putconncCst|j}|||S)z< Return current stats about the pool usage. )dictr4update _get_measures)r:rvr<r<r= get_statss zBasePool.get_statscCs*|jt}|_t|}|||S)zy Return current stats about the pool usage. After the call, all the counters are reset to zero. )r4r rYrZr[)r:statsr\r<r<r= pop_statsszBasePool.pop_statsc Cs(|j|j|j|j|j|j|jt|jiS)zG Return immediate measures of the pool (not counters). ) _POOL_MINr0 _POOL_MAXr1 _POOL_SIZEr2_POOL_AVAILABLElenr rIr<r<r=r[s  zBasePool._get_measuresvaluemin_pcmax_pccCs|d||t|S)zV Add a random value to *value* between *min_pc* and *max_pc* percent. ?r)clsrerfrgr<r<r=_jitterszBasePool._jittercCst||jdd|_dS)zeSet an expiry date on a connection. Add some randomness to avoid mass reconnection. gN)rrjr) _expire_at)r:rSr<r<r=_set_connection_expiry_datesz$BasePool._set_connection_expiry_date)r!)r?N)rSrTr?N)-rGrF __qualname__r.r`rarbrc_REQUESTS_WAITING _REQUESTS_NUM_REQUESTS_QUEUED_REQUESTS_WAIT_MS_REQUESTS_ERRORS _USAGE_MS _RETURNS_BAD_CONNECTIONS_NUM_CONNECTIONS_MS_CONNECTIONS_ERRORS_CONNECTIONS_LOSTr__annotations__r3rrrintfloatr>rJpropertyr$r%boolrNrr-rQrRrXr]r_r[ classmethodrjrmr<r<r<r=rsv       9"     rc@sNeZdZdZdZdZdZdefddZded d fd d Z ded e fd dZ d S)AttemptWithBackoffzR Keep the state of a repeated operation attempt with exponential backoff. rhg?g@r'cCs||_d|_d|_dS)Nrk)r'delay give_up_at)r:r'r<r<r=r>s zAttemptWithBackoff.__init__nowr?NcCsj|jdkr||j|_t|j|j |j|_n|j|j9_|j||jkr3td|j||_dSdS)z7Calculate how long to wait for a new connection attemptrkN) rr'rrrj INITIAL_DELAY DELAY_JITTER DELAY_BACKOFFmaxr:rr<r<r= update_delays  zAttemptWithBackoff.update_delaycCs|jdko ||jkS)z8Return True if we are tired of trying this attempt. Meh.rk)rrr<r<r=time_to_give_upsz"AttemptWithBackoff.time_to_give_up) rGrFrn__doc__rrrr|r>rr~rr<r<r<r=rs rN)rtimerrtypingrrrrr psycopgr rOr _compatr rpsycopg._connection_baserrrr<r<r<r=s     8