YtjhdZddlmZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZmZm Z m!Z!m"Z"ddl#m$Z$m%Z%ddl&m'Z'ddlm(Z(m)Z)m*Z*ddl+m,Z,m-Z-m.Z.ddl/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5ddl/m6Z6m7Z7m8Z8e9Z:ej;dZ<Gddeee%Z=GddeeZ>GddeZ?Gdde?Z@Gd d!e?ZAGd"d#e?ZBGd$d%e?ZCGd&d'e?ZDdS)(z0 Psycopg connection pool module (sync version). ) annotationsN)ABCabstractmethod) monotonic) TracebackType)AnyGenericcast)ref)contextmanager)deque)Iterator) Connection)errors)TransactionStatus)CTConnectFailedCB ConnectionCB ConninfoParam KwargsParam)AttemptWithBackoffBasePool) Scheduler) PoolClosed PoolTimeoutTooManyRequests)PSYCOPG_VERSIONPoolConnectionSelf) ConditionEventLockQueueWorkercurrent_thread_name)gathersleepspawnz psycopg.poolceZdZUded< dqeeeedddddddddddd d d dd d drfd+Zdsd.Z dsfd/ Z dtdud0Z e dvdwd3Z dvdxd5Zdyd7Zdzd8Zd{d:Zd|d<Zdsd=Zd|d>Zdsd?Zd}dAZd~dBZdddDZdsdEZdsdFZdsdGZddudIZddKZddMZddTZdvddUZ dsdVZ!e"d|dWZ#dsdXZ$dd[Z%dd]Z&e'dd`Z(dvdxdaZ)ddcZ*ddeZ+ dddiZ,d}djZ-d|dkZ.d|dlZ/d|dmZ0dsdnZ1dfdp Z2xZ3S)ConnectionPoolz deque[CT]_poolNF>@rg @g@gr@)connection_classkwargsmin_sizemax_sizeopen configurecheckresetname close_returnstimeout max_waiting max_lifetimemax_idlereconnect_timeoutreconnect_failed num_workersconninforr1type[CT]r2KwargsParam | Noner3intr4 int | Noner5 bool | Noner6ConnectionCB[CT] | Noner7r8r9 str | Noner:boolr;floatr<r=r>r?r@ConnectFailedCB | NonerAc J| rItdkr>|tur&tttt }nt d||_||_||_ ||_ ||_ | |_ ||_ |||ttt|_d|_d|_g|_t)||| | | | |||| t-|_| dx}|_|r|dSdS)N)r0r0zUsing 'close_returns=True' and a non-standard 'connection_class' requires psycopg 3.3 or newer. Please check the docs at https://www.psycopg.org/psycopg3/docs/advanced/pool.html#pool-sqlalchemy for a workaround.) r3r4r9r:r;r<r=r>r?rAT)rrr typerr TypeErrorrBr2r1_check _configure_reset_reconnect_failedr WaitingClient_waiting_pool_full_event _sched_runner_workerssuper__init__r#_lock_open_implicit_open)selfrBr1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rA __class__s O/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/psycopg_pool/pool.pyrZzConnectionPool.__init__+sK,  _v55:--#'R.#A#A  j!  0 # !1   +mB/022 /3,0&(  '#%/#  VV <)- -D4&   JJLLLLL  returnNonecrt|ddrdS|}d}t|d|ddS)N_closedTzNyou can try to call 'close()' explicitly or to use the pool as context manager@)r; timeout_hint)getattr_signal_stop_workerr')r^workershints r`__del__zConnectionPool.__del__vsN 4D ) )  F**,,_4888888ract|jrNd|_tjdt |jdt |jdtdSdS)NFzthe default for the z 'open' parameter will become 'False' in a future release. Please use open={True|False} explicitly, or use the pool as context manager using: `with z(...) as pool: ...`)rY_check_open_getconnr\warningswarnrN__name__DeprecationWarning)r^r_s r`rnz"ConnectionPool._check_open_getconns ##%%%   "'D  M{tDzz':{{SWX\S]S]Sf{{{"       racZ||j5|jrJt|j|jkr ddddSt |_dddn #1swxYwYtd|j |j |s'| td|d|j5|jsJd|_dddn #1swxYwYtd|j dS)a- Wait for the pool to be full (with `min_size` connections) after creation. Close the pool, and raise `PoolTimeout`, if not ready within *timeout* sec. Calling this method is not mandatory: you can try and use the pool immediately after its creation. The first client will be served as soon as a connection is ready. You can use this method if you prefer your program to terminate in case the environment is not configured properly, rather than trying to stay up the hardest it can. Nz"waiting for pool %r initializationz%pool initialization incomplete after  seczpool %r is ready to use) rnr[rVlenr, _min_sizer"loggerinfor9waitcloser)r^r;s r`ryzConnectionPool.waits   """ Z , ,, , ,,4:$.00 , , , , , , , ,%*GGD !  , , , , , , , , , , , , , , ,  8$)DDD$))'22 U JJLLLSgSSSTT T Z ) )( ( (($(D ! ) ) ) ) ) ) ) ) ) ) ) ) ) ) )  -ty99999s)(A0A00A47A4#DDD float | None Iterator[CT]c #K||} t}|5|Vdddn #1swxYwY||t}|j|jxxt d||z zz cc<dS#||t}|j|jxxt d||z zz cc<wxYw)a&Context manager to obtain a connection from the pool. Return the connection immediately if available, otherwise wait up to *timeout* or `self.timeout` seconds and throw `PoolTimeout` if a connection is not available in time. Upon context exit, return the connection to the pool. Apply the normal :ref:`connection context behaviour ` (commit/rollback the transaction in case of success/error). If the connection is no more in working state, replace it with a new one. r;N@@)getconnrputconn_stats _USAGE_MSrE)r^r;connt0t1s r` connectionzConnectionPool.connections=||G|,, CB                   LL   B K ' ' '3vb/A+B+B B ' ' ' ' ' LL   B K ' ' '3vb/A+B+B B ' ' ' ' ' ' 's)B; B?B?BAC*rcD||j}t|z}td|j|j|jxxdz cc<| ||S#t$rtd|dddwxYw)aObtain a connection from the pool. You should preferably use `connection()`. Use this function only if it is not possible to use the connection as context manager. After using this function you *must* call a corresponding `putconn()`: failing to do so will deplete the pool. A depleted pool is a sad pool: you don't want a depleted pool. Nzconnection requested from %rr couldn't get a connection after .2frt) r;rrwrxr9r _REQUESTS_NUMrn_getconn_with_check_loopr)r^r;deadlines r`rzConnectionPool.getconns ?lG;;( 2DI>>> D&'''1,'''   """ 00:: :   D7DDDD  s *A?? Brcd} ||tz } ||td|j|S#t $r||dYnwxYwt}|st||z }n| || |rtt|j )NTzconnection given by %r from_getconnr~)_getconn_uncheckedr_check_connectionrwrxr9CLIENT_EXCEPTIONS_putconnr update_delaytime_to_give_uprr(delay)r^rattemptrnows r`rz'ConnectionPool._getconn_with_check_loops -1 %**8ikk+ABBD &&t,,, 4di@@@ % 7 7 7 d 66666 7++C *,X^DDD$$S)))&&s++ %!mm#gm$$$+ %sA !BBc *|j5||x}sdt}t}|j||j|jxxdz cc<|dddn #1swxYwY|s | |}nZ#t$rM|j r*| t||j d|j|jxxdz cc<wxYw t}|j|jxxt!d||z zz cc<n@#t}|j|jxxt!d||z zz cc<wxYw||_|S)Nrr~Trr)r[_get_ready_connectionrrTrUappendr_REQUESTS_QUEUED_maybe_grow_poolryrrrun_taskReturnConnection_REQUESTS_ERRORS_REQUESTS_WAIT_MSrEr,)r^r;rrposrs r`rz!ConnectionPool._getconn_uncheckeds Z ( ( 66w???D ([[)6 $$S))) D1222a7222%%''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( O Oxxx00$   8WMM"24PT"U"U"UVVV D1222a7222  [[ D2333s6R"W;M7N7NN3333[[ D2333s6R"W;M7N7NN3333333   s1A>--  : :%%''D4:!111#&tz??   #dm"4"48H"H"H K- . . .! 3 . . .![DI[[c$-6H6H[[[  rarc|jsdS ||dS#t$r!}td|d}~wwxYw)Nzconnection failed check: %s)rPrrwrx)r^res r`rz ConnectionPool._check_connection,se{  F  KK          KK5q 9 9 9  s" A AA c|j|jks|jrdS|xjdz c_td|j|jd|_|t|ddS)Nrgrowing pool %r to %sTgrowing)_nconns _max_size_growingrwrxr9r AddConnectionr^s r`rzConnectionPool._maybe_grow_pool5sv <4> ) )T] ) F   +TY EEE  mD$77788888rac||td|j||rdS||ddS)zReturn a connection to the loving hands of its pool. Use this function only paired with a `getconn()`. You don't need to use it if you use the much more comfortable `connection()` context manager. zreturning connection to %rNFr)_check_pool_putconnrwrxr9_maybe_close_connectionrr^rs r`rzConnectionPool.putconn?sf   &&& 0$)<<<  ' ' - -  F d /////rac<|j5t|j}|jt |_dddn #1swxYwY|D]9}|||t|:dS)ak Remove all the connections from the pool and create new ones. If a connection is currently out of the pool it will be closed when returned to the pool and replaced with a new one. This method is useful to force a connection re-configuration, for example when the adapters map changes after the pool was created. N) r[listr,clearr _drained_at_close_connectionrrr^connsrs r`drainzConnectionPool.drainNsZ + +$$E J     ({{D  + + + + + + + + + + + + + + +  / /D  " "4 ( ( ( MM--- . . . . / /sAAAArc|jr'|t|||dS|||dSNr)rRrr_return_connectionr^rrs r`rzConnectionPool._putconnbsT ; E MM*4LQQQ R R R R R  # #D| # D D D D DracB|jsdS||dS)zgClose a returned connection if necessary. Return `!True if the connection was closed. FT)rerrs r`rz&ConnectionPool._maybe_close_connectionis,| 5 t$$$traryc|j5|dddn #1swxYwY|r||dSdS)a`Open the pool by starting connecting and and accepting clients. If *wait* is `!False`, return immediately and let the background worker fill the pool if `min_size` > 0. Otherwise wait up to *timeout* seconds for the requested number of connections to be ready (see `wait()` for details). It is safe to call `!open()` again on a pool already open (because the method was already called, or because the pool context was entered, or because the pool was initialized with *open* = `!True`) but you cannot currently re-open a closed pool. Nr~)r[r]ry)r^ryr;s r`r5zConnectionPool.openvsZ   JJLLL                 ' IIgI & & & & & ' 's )--c|jsdS|t|_t |_d|_d|_||dS)NFT) re _check_openr$_tasksr_sched_opened_start_workers_start_initial_tasksrs r`r]zConnectionPool._openss|  F gg kk     !!#####ract|jj|jd|_|jrJt |jD]C}t|j|j f|jd|}|j |DdS)Nz -scheduler)r9z-worker-)argsr9) r)rrunr9rWrXrangerAworkerrr)r^its r`rzConnectionPool._start_workerss"4;?DI9Q9Q9QRRR=   t'(( $ $Adk =V=VST=V=VWWWA M  # # # # $ $ract|jD]$}|t|%|t |t ||jdSN)rrrrSchedule ShrinkPoolr>)r^rs r`rz#ConnectionPool._start_initial_taskssht|$$ / /A MM--- . . . . htZ%5%5t}EEFFFFFrarfc,|jrdS|j5d|_td|jt |j}|jt |j}|j| }dddn #1swxYwYt|d|i|D]}| ||D]-}| td|jd.dS)aClose the pool and make it unavailable to new clients. All the waiting and future clients will fail to acquire a connection with a `PoolClosed` exception. Currently used connections will not be closed until returned to the pool. Wait *timeout* seconds for threads to terminate their job, if positive. If the timeout expires the pool is closed anyway, although it may raise some warnings on exit. NTzpool %r closedr;rz is closed)rer[rwdebugr9rrUrr,rir'rfailr)r^r;waiting connectionsrjrrs r`rzzConnectionPool.closesp <  F Z 1 1DL LL)49 5 5 54=))G M   ! ! !tz**K J     ..00G 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1" ))))  ) )D  " "4 ( ( ( ( F FC HHZ CDI C C CDD E E E E F FsBB33B7:B7 list[Worker]c|jdd|jddgc}|_|D]$}|t |%|jr!||jd|_|S)Nr)renterrXr StopWorkerrWr)r^rj_s r`riz"ConnectionPool._signal_stop_workers !T"""#'-"2B , ,A MM*T** + + + +   & NN4- . . .!%D rar c<d|_||SNF)r\r5rs r` __enter__zConnectionPool.__enter__s#  raexc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | Nonec.|dSr)rz)r^rrrs r`__exit__zConnectionPool.__exit__s rac|||\}}td||jz }td|j|||j5||_||_|xj|z c_dddn #1swxYwYt|D]$}| t|%dS)z+Change the size of the pool during runtime.rz&resizing %r to min_size=%s max_size=%sN) _check_sizemaxrvrwrxr9r[rrrrr)r^r3r4ngrowrs r`resizezConnectionPool.resizes!--hAA(Ax$.011 4di8   Z " "%DN%DN LLE !LL " " " " " " " " " " " " " " " u / /A MM--- . . . . / /sBB  B c|j5t|j}|j|dddn #1swxYwY|r|}|jtkrSt d|| || t| | |||ng#t$rZ|j|jxxdz cc<td|| t|YnwxYw|dSdS)zVerify the state of the connections currently in the pool. Test each connection: if it works return it to the pool, otherwise dispose of it and create a new one. Nz discarding expired connection %srz discarding broken connection: %s)r[rr,rrpop _expire_atrrwrxrrrcheck_connection _add_to_poolrr_CONNECTIONS_LOSTwarningrs r`r7zConnectionPool.checks Z $ $$$E J       ! ! # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ (99;;D)++-- >EEE&&t,,, mD11222 (%%d+++ !!$'''' % 3 3 3 D2333q8333A4HHH mD1122222 3 ( ( ( ( (s%AAAA#DA!E21E2c|jr|ddSd|_ |dd|_dS#d|_wxYw)a> A simple check to verify that a connection is still working. Return quietly if the connection is still working, otherwise raise an exception. Used internally by `check()`, but also available for client usage, for instance as `!check` callback when a pool is created. r-TFN) autocommitexecute)rs r`rzConnectionPool.check_connection$sb ? ( LL     "DO ( R   "'%''''s A AcB|jsdS||dS)zV Called when reconnection failed for longer than `reconnect_timeout`. N)rSrs r`r@zConnectionPool.reconnect_failed8s.%  F t$$$$$rataskMaintenanceTaskc:|j|dS)z#Run a maintenance task in a worker.N)r put_nowait)r^rs r`rzConnectionPool.run_taskAs t$$$$$rarcF|j||jdS)z1Run a maintenance task in a worker in the future.N)rrtick)r^rrs r` schedule_taskzConnectionPool.schedule_taskEs" %+++++raqQueue[MaintenanceTask]cR t|x}tr)tdt dS |n>#t$r1}td||j j |Yd}~nd}~wwxYw)zRunner to execute pending maintenance task. The function is designed to run as a task. Block on the queue *q*, run a task received. Finish running if a StopWorker is received. Tzterminating working task %sNztask run %s failed: %s: %s) isinstancegetrrwrr&rrrr_rq)clsrrexs r`rzConnectionPool.workerIs 15577?4Z88  :(>F$ % [[ I04(0DDVDDD B K, - - -VrBw5G1H1H H - - - - !    K0 1 1 1Q 6 1 1 1   ?  OOD ! ! !+88=N=SSS)&116(kkktkkk ((... s C&C=strcdt|jr|S|jpdS)zCResolve conninfo (static string, sync callable, or async callable).r-)callablerBrs r`rz ConnectionPool._resolve_conninfos/ DM " " #==?? "}""radict[str, Any]cr|jsiSt|jr|S|jS)z?Resolve kwargs (static dict, sync callable, or async callable).)r2r!rs r`rzConnectionPool._resolve_kwargss9{ I DK  !;;== {rarAttemptWithBackoff | NonercFt}|st|j} |}n#t$r}t d|j|||rzt d|j|j|j 5|xj dzc_ |r|j rd|_ dddn #1swxYwY| n@| ||t||||jYd}~dSd}~wwxYwt d|||r|j 5|j |jks|j |jkrb|jr[|xj dz c_ t d |j|j |t|d nd|_ ddddS#1swxYwYdSdS) zTry to connect and add the connection to the pool. If failed, reschedule a new attempt in the future for a few times, then give up, decrease the pool connections number and call `self.reconnect_failed()`. r~zerror connecting in %r: %sz3reconnection attempt in pool %r failed after %s secrFNrz!adding new connection to the poolrT)rrr?rrrwrr9rr[rrr@rrrrrxrrvrrUr)r^rrrrr s r`_add_connectionzConnectionPool._add_connectionskk I(1GHHHG ==??DD     NN7B G G G&&s++ II* Z..LLA%LL.4=.(- ............... %%''''$$S)))""!$AAA7= FFFFF' *  7888 $  * * *<$.00L4>11dm1LLA%LLKK 7DLQQQMM-d"C"C"CDDDD$)DM * * * * * * * * * * * * * * * * * * * *sN= EA#D=*!C D=C D=C AD==E=B HHHc|||rs|jjtjkrX|j|jxxdz cc<td| t|dSns|jjtjkrY|j|j xxdz cc<t d|| t|dS|j |jks|jt!krStd||| t|dS||dS)z> Return a connection to the pool after usage. rz#not serving connection found brokenNz discarding closed connection: %szdiscarding expired connection)_reset_connectionrrrUNKNOWNrrrwrxrr _RETURNS_BADr _created_atrrrrrrs r`rz!ConnectionPool._return_connections{ t$$$  {-1B1JJJ D2333q8333 ABBB mD11222 K [ +/@/H H H K) * * *a / * * * NN=t D D D MM--- . . . F  t/ / /4?ikk3Q3Q KK7 8 8 8  " "4 ( ( ( MM--- . . . F $racd|_|jr||dS|j5|jr#|| ddddS|jr4|j|rn^|j4|j||jr6t|j|j kr|jddddS#1swxYwYdS)z Add a connection to the pool. The connection can be a fresh one or one already used in the pool. If a client is already waiting for a connection pass it on, otherwise put it back into the pool N) r,rerr[rUrsetrrVrurvrs r`rzConnectionPool._add_to_poolsx <   " "4 ( ( ( FZ 0 0 | &&t,,, 0 0 0 0 0 0 0 0- 0=((**..t44 - 0 !!$'''(0S__-N-N)--///+ 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0sC7BC77C;>C;c|jjx}tjkrn|tjkrdS|tjks|tjkrtd| | n#t$rF}td|j j ||| |Yd}~nHd}~wwxYw|tjkr0td|| ||jr |||jjx}tjkr4t|j}t#jd|d|jddS#t$r;}td|| |Yd}~dSd}~wwxYwdS) z? Bring a connection to IDLE state or close it. Nz$rolling back returned connection: %sz1rollback failed: %s: %s. Discarding connection %szclosing returned connection: %sr z by reset function rzerror resetting connection: %s)rrrrr)INTRANSINERRORrwrrollbackrr_rqrACTIVErRr9rr)r^rrr rs r`r(z ConnectionPool._reset_connectionsk4 4F9J9O O O  (0 0 0 F (0 0 0F>O>W4W4W NNA4 H H H - $ - - -GL)  &&t,,,,,,,, -(/ / / NNr)r^to_close nconns_mins r` _shrink_poolzConnectionPool._shrink_pool:s-" Z & &)J"4:D |dn,,aDJ:--// !   A%   & & & & & & & & & & & & & & &  - KKZ        " "8 , , , , , - -sA7B  BBdict[str, int]ct}t|j||j<|Sr)rY _get_measuresrurU_REQUESTS_WAITING)r^rvr_s r`r:zConnectionPool._get_measuresRs4 WW " " $ $%(%7%74 !" ra)r-)$rBrr1rCr2rDr3rEr4rFr5rGr6rHr7rHr8rHr9rIr:rJr;rKr<rEr=rKr>rKr?rKr@rLrArErbrc)r/)r;rKrbrcr)r;r{rbr|)r;r{rbr)rrKrbrr;rKrbr)r;r{rbr)rrrbrc)rrrrJrbrcrrrbrJ)Fr/)ryrJr;rKrbrc)rf)rbr)rbr )rrrrrrrbrc)r3rEr4rFrbrc)rrrbrc)rrrrKrbrc)rrrbrcrbr)rbr")F)rr$rrJrbrc)rbr8)4rq __module__ __qualname____annotations__r rNrrrZrlrnryr rrrrrrrrrrrr5r]rrrzrirrrr7 staticmethodrr@rr classmethodrrrrr&rrr(rr7r: __classcell__r_s@r`r+r+(s;#%I&*T$r(J%?%?%)# -1)-)-#'##+37)IIIIIIIIV9999      :::::@CCCC^C,8%%%%4    D$9999 0 0 0 0////(EEEE    '''''($$$$"$$$$GGGG'F'F'F'F'FR     /////"!(!(!(!(F(((\(&%%%%%%%%,,,,[*@####CH2*2*2*2*2*h    8-0-0-0-0^%-%-%-%-N----0rar+c6eZdZdZdZddZdd Zdd ZddZdS)rTz|j5|jsZ|jsS |j|st d|dd|_n#t $r}||_Yd}~nd}~wwxYwdddn #1swxYwY|jr|j|jsJ|jS)zWait for a connection to be set and return it. Raise an exception if the wait times out or if fail() is called. rrrtN)rJrrIryrr)r^r;r s r`ryzWaitingClient.waiths  Z $ $I $ $$:??733%0PwPPPP&& )$$$!#DJJJJJJ$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ : * 9  99 s:A63A  A6 A'A"A6"A''A66A:=A:rrJc|j5|js|jr ddddS||_|j ddddS#1swxYwYdS)zSignal the client waiting that a connection is ready. Return True if the client has "accepted" the connection, False otherwise (typically because wait() has timed out). NFTrJrrI notify_allrs r`r-zWaitingClient.set}s Z  y DJ         DI J ! ! # # #                   A!AAArI Exceptionc|j5|js|jr ddddS||_|j ddddS#1swxYwYdS)zSignal the client that, alas, they won't have a connection today. Return True if the client has "accepted" the error, False otherwise (typically because wait() has timed out). NFTrN)r^rIs r`rzWaitingClient.fails Z  y DJ         DJ J ! ! # # #                   rPNr=r>r?)rIrQrbrJ) rqrArB__doc__ __slots__rZryr-rrar`rTrTXsoFF*I ! ! ! !*          rarTcJeZdZdZd dZddZdd Zdd Zedd Z d S)rz8A task to run asynchronously to maintain the pool state.poolConnectionPool[Any]c.t||_dSr)r rWr^rWs r`rZzMaintenanceTask.__init__sII rarbrc|}|rt|jnd}d|jjd|dt |ddS)Nz< z at 0xx>)rWreprr9r_rqid)r^rWr9s r`__repr__zMaintenanceTask.__repr__sUyy{{"&*FFTFFDFFFFFrarcc|}|r|jrtd|dStdt |||dS)zRun the task. This usually happens in a worker. Call the concrete _run() implementation, if the pool is still alive. ztask run discarded: %sNztask running in %s: %s)rWclosedrwrr&_runrZs r`rzMaintenanceTask.runsm yy{{ t{  LL14 8 8 8 F -/B/D/DdKKK $rac|}|r|jrtd|dS||dS)zRun the scheduled task This function is called by the scheduler task. Use a worker to run the task for real in order to free the scheduler immediately. ztask tick discarded: %sN)rWrdrwrrrZs r`rzMaintenanceTask.ticksT yy{{ t{  LL2D 9 9 9 F dracdSrrUrZs r`rezMaintenanceTask._runs7:sraN)rWrXr@r=rWrXrbrc) rqrArBrSrZrbrrrrerUrar`rrsBBGGGG        :::^:::rarceZdZdZddZdS) rz+Signal the maintenance worker to terminate.rWrXrbrccdSrrUrZs r`rezStopWorker._runs raNrhrqrArBrSrerUrar`rrs.55      rarc.eZdZ d dfd Zdd ZxZS)rNFrWrXrr$rrJcft|||_||_dSr)rYrZrr)r^rWrrr_s r`rZzAddConnection.__init__s/   rarbrccH||j|jdS)Nr)r&rrrZs r`rezAddConnection._runs% T\4<@@@@@rar)rWrXrr$rrJrh)rqrArBrZrerFrGs@r`rrsf .2 AAAAAAAArarc,eZdZdZd fd Zd d ZxZS)rz-Clean up and return a connection to the pool.rWrXrrrrJcft|||_||_dSr)rYrZrr)r^rWrrr_s r`rZzReturnConnection.__init__s0  (rarbrccH||j|jdSr)rrrrZs r`rezReturnConnection._runs&  8IJJJJJra)rWrXrrrrJrhrqrArBrSrZrerFrGs@r`rrs_77)))))) KKKKKKKKrarceZdZdZddZdS) rzIf the pool can shrink, remove one connection. Re-schedule periodically and also reset the minimum number of connections in the pool. rWrXrbrccd|||j|dSr)rr>r7rZs r`rezShrinkPool._runs4 4/// raNrhrkrUrar`rrs2 rarc,eZdZdZd fd Zd d ZxZS)rzSchedule a task in the pool scheduler. This task is a trampoline to allow to use a sync call (pool.run_task) to execute an async one (pool.schedule_task). It is pretty much no-op in sync code. rWrXrrrrKcft|||_||_dSr)rYrZrr)r^rWrrr_s r`rZzSchedule.__init__s-   rarbrccF||j|jdSr)rrrrZs r`rez Schedule._runs" 49dj11111ra)rWrXrrrrKrhrrrGs@r`rrs[ 22222222rar)ErS __future__rloggingroabcrrtimertypesrtypingrr r weakrefr contextlibr collectionsr collections.abcrpsycopgrrr psycopg.pqrrrrrrbaserrschedrrrr_compatrrr _acompatr!r"r#r$r%r&r'r(r)rQr getLoggerrwr+rTrrrrrrrUrar`rs #"""""########%%%%%%%%%%%%%%%%$$$$$$((((((NNNNNNNNNNNNNN........<<<<<<<<<<::::::::::PPPPPPPPPPPPPPPP**********  > * *m m m m m WR[(m m m `?????GBK???D);););););c);););X         A A A A AO A A A K K K K K K K K        2222222222ra