3fiZdZddlmZddlmZddlmZddlmZmZddl m Z m Z ddl m Zdd l mZerdd lmZGd d ZGd dZy)z9 psycopg connection pool base class and functionalities. ) annotations) monotonicrandom) TYPE_CHECKINGAny)Counterdeque)errors) PoolClosed)BaseConnectionceZdZUdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZded< d"dZd#dZed$dZed$dZed%dZd&dZd'dZd'dZd(dZd)dZd)dZd)dZ e!d*dZ"d(d Z#y!)+BasePoolrpool_minpool_max pool_sizepool_availablerequests_waiting requests_numrequests_queuedrequests_wait_msrequests_errorsusage_ms returns_badconnections_numconnections_msconnections_errorsconnections_lostz deque[Any]_poolc |j||\}}|s#tjdzx} t_d| }| dkr td||_||_||_||_||_||_ | |_ ||_ ||_ | |_ ||_t|_t#t$|_d|_||_d|_d|_d|_d|_y)Nr zpool-znum_workers must be at least 1FT) _check_sizer _num_pool ValueErrorname close_returns _min_size _max_sizetimeout max_waitingreconnect_timeout max_lifetimemax_idle num_workers_nconnsr r r str_stats _drained_at _nconns_min_growing_opened_closed_open_implicit) selfmin_sizemax_sizer&r'r*r+r-r.r,r/nums O/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/psycopg_pool/base.py__init__zBasePool.__init__-s"--hA('/'9'9A'= =C($3%=D ?=> > *!! &!2(  & W cln $    #c d|jjd|jjd|jdt |dd S)N<. z at 0xx>) __class__ __module____name__r&idr9s r=__repr__zBasePool.__repr__esH))*!DNN,C,C+D }F2d8A,a 1 r?c|jSN)r(rJs r=r:zBasePool.min_sizek ~~r?c|jSrM)r)rJs r=r;zBasePool.max_sizeorNr?c|jS)z`!True` if the pool is closed.)r7rJs r=closedzBasePool.closedss||r?c||}|dkr td||kr td||cxk(rdk(rtd||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%)r9r:r;s r=r#zBasePool._check_sizexsi  H a<:; ; h NO O x $1 $RS S %!!x!!r?c`|jr"|jrtjdyy)Nz8pool has already been opened/closed and cannot be reused)r7r6eOperationalErrorrJs r= _check_openzBasePool._check_opens, <|jrtd|jdtd|jdy)Nz the pool z is already closedz is not open yet)r7r6r r&rJs r=_check_open_getconnzBasePool._check_open_getconnsG <<|| 9TYYM9K!LMM 9TYYM9I!JKK r?ct|ddx}|ury|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%)r9connpoolmsgs r=_check_pool_putconnzBasePool._check_pool_putconnsZD'40 0DT 9  ' }5C1C.tyym2cU"TF K  r?cnt|j}|j|j|S)z< Return current stats about the pool usage. )dictr2update _get_measures)r9rvs r= get_statszBasePool.get_statss,$++  $$$&' r?c|jtc}|_t|}|j|j |S)zy Return current stats about the pool usage. After the call, all the counters are reset to zero. )r2r r`rarb)r9statsrcs r= pop_statszBasePool.pop_statss; "[[')t{ %[ $$$&' r?c |j|j|j|j|j|j |j t|jiS)zG Return immediate measures of the pool (not counters). ) _POOL_MINr( _POOL_MAXr) _POOL_SIZEr0_POOL_AVAILABLElenr rJs r=rbzBasePool._get_measuressH NNDNN NNDNN OOT\\  #djj/   r?c4|d||z tzz|zzS)zV Add a random value to *value* between *min_pc* and *max_pc* percent. ?r)clsvaluemin_pcmax_pcs r=_jitterzBasePool._jitters$ 68;rKpropertyr:r;rQr#rVrXr^rdrgrb classmethodrtryr?r=rrsLIIIJ&O*"M(*(I L(&O.* 6$6$ 6$  6$  6$6$6$6$6$!6$6$p  " L     GG Jr?rc4eZdZdZdZdZdZd dZd dZd dZ y) AttemptWithBackoffzR Keep the state of a repeated operation attempt with exponential backoff. rog?g@c.||_d|_d|_y)Nr")r*delay give_up_at)r9r*s r=r>zAttemptWithBackoff.__init__s  r?cv|jdk(rP||jz|_tj |j |j |j |_n|xj|jzc_|j|z|jkDrtd|j|z |_yy)z7Calculate how long to wait for a new connection attemptr"N) rr*rrrt INITIAL_DELAY DELAY_JITTER DELAY_BACKOFFmaxr9nows r= update_delayzAttemptWithBackoff.update_delays :: !DLL0DO!))""T%6%6$68I8IDJ JJ$,, ,J :: doo -S$//C"78DJ .r?cB|jdkDxr||jk\S)z8Return True if we are tired of trying this attempt. Meh.r")rrs r=time_to_give_upz"AttemptWithBackoff.time_to_give_ups$?)??r?N)r*r})rr}r~r)rr}r~r|) rHrGr__doc__rrrr>rrrr?r=rrs(MLM 9@r?rN)r __future__rtimerrtypingrr collectionsr r psycopgr rTr psycopg._connection_baserrrrr?r=rsB #%&7sJsJl@@r?