U @f<@sddlmZddlmZddlZddlZddlmZddlm Z ddlm Z ddl m Z ddl mZeZGd d d eZd d ZGd ddeZGdddeZGdddeZeZGdddZejD] ZefddZeeeeeqe`[[GdddeZGdddZdS))deque)contextmanagerN)Pool)timeout)hubs)Timer) GreenThreadc@s eZdZdS)ConnectTimeoutN)__name__ __module__ __qualname__r r J/var/www/html/chatgpt/venv/lib/python3.8/site-packages/eventlet/db_pool.pyr sr cCs |dSN)rollback)connr r rcleanup_rollbacksrcseZdZdddddeffdd Zdd Zd d Zd d ZddZdddZ fddZ e ffdd Z e e fddZddZddZZS)BaseConnectionPoolr c sN|st||_||_| |_||_||_||_d|_||_t j ||dddS)a Constructs a pool with at least *min_size* connections and at most *max_size* connections. Uses *db_module* to construct new connections. The *max_idle* parameter determines how long pooled connections can remain idle, in seconds. After *max_idle* seconds have elapsed without the connection being used, the pool closes the connection. *max_age* is how long any particular connection is allowed to live. Connections that have been open for longer than *max_age* seconds are closed, regardless of idle time. If *max_age* is 0, all connections are closed on return to the pool, reducing it to a concurrency limiter. *connect_timeout* is the duration in seconds that the pool will wait before timing out on connect() to the database. If triggered, the timeout will raise a ConnectTimeout from get(). The remainder of the arguments are used as parameters to the *db_module*'s connection constructor. NT)min_sizemax_sizeZorder_as_stack) AssertionError _db_module_args_kwargsmax_idlemax_ageconnect_timeout_expiration_timercleanupsuper__init__) self db_modulerrrrr r"argskwargs __class__r rr$szBaseConnectionPool.__init__c Cs|jdks|jdkrdS|jdk r4t|jdds4dSzZt}|||jdd||j}tdd|jD}|||j}t||}Wn tt fk rd|_YdSX|dkrt |t t jj|jgi|_|jdS)aSets up a timer that will call _expire_old_connections when the oldest connection currently in the free pool is ready to expire. This is the earliest possible time that a connection could expire, thus, the timer will be running as infrequently as possible without missing a possible expiration. If this function is called when a timer is already scheduled, it does nothing. If max_age or max_idle is 0, _schedule_expiration likewise does nothing. rNZcalledFcSsg|] }|dqS)r ).0tr r r Zsz;BaseConnectionPool._schedule_expiration..)rrr!getattrtime_expire_old_connections free_itemsmin IndexError ValueErrorrrrZget_hubZgreenletswitch_schedule_expirationZschedule)r%nowZ idle_delayZoldestZ age_delayZ next_delayr r rr8>s.    z'BaseConnectionPool._schedule_expirationcstj}fddjD}fddjD}jj|j|tj8_|D]}j|ddqhdS)aIterates through the open connections contained in the pool, closing ones that have remained idle for longer than max_idle seconds, or have been in existence for longer than max_age seconds. *now* is the current time, as returned by time.time(). cs$g|]\}}}||r|qSr  _is_expiredr- last_used created_atrr9r%r rr/qsz>BaseConnectionPool._expire_old_connections..cs*g|]"\}}}||s|||fqSr r:r<r?r rr/vsTquietN)lenr3clearextend current_size _safe_close)r%r9Zoriginal_countZexpiredZnew_freerr r?rr2is     z*BaseConnectionPool._expire_old_connectionscCs8|jdks0|jdks0|||jks0|||jkr4dSdS)z@Returns true and closes the connection if it's expired. rTF)rr)r%r9r=r>r r rr;s  zBaseConnectionPool._is_expiredcCs:d}z|r|j}|nd}Wntk r4YnX|S)a$If the connection was wrapped by a subclass of BaseConnectionWrapper and is still functional (as determined by the __nonzero__, or __bool__ in python3, method), returns the unwrapped connection. If anything goes wrong with this process, returns None. N)_base_destroyAttributeError)r%rbaser r r_unwrap_connections z%BaseConnectionPool._unwrap_connectionFcCsPz |Wn>tk r Yn,tk rJ|sFtdtdYnXdS)zVCloses the (already unwrapped) connection, squelching any exceptions. zConnection.close raised: %sr,N)closerI Exceptionprintsysexc_info)r%rrAr r rrFs zBaseConnectionPool._safe_closecsvt}|dkrDz |}Wn$tk rB|jd8_YnXt|trZ|\}}}nt}t||}||_ |S)Nr,) r#getcreaterMrE isinstancetupler1PooledConnectionWrapper_db_pool_created_at)r%rZ _last_usedr>wrappedr)r rrQs     zBaseConnectionPool.getc st|dd}t}||}||||r@|j|ddd}nn|dk r|tkrV|j}z|rd||WnFtk r}ztd||fd}W5d}~XYnd}YnX|dk rt |||fn(| dkrt dn|j d8_ | dS)NrVrFr@zWARNING: cleanup %s raised: %sr,)r0r1rKr;rF_MISSINGr"rMrNr#putZwaitingrEr8)r%rr"r>r9er)r rrYs0    zBaseConnectionPool.putc cs(|}z |VW5|j||dXdS)N)r")rQrY)r%r"rr r ritems zBaseConnectionPool.itemcCs`|jr|j|jt}|_|D]6}t|tr:|dn|}|j|dd|jd8_q$dS)zuClose all connections that this pool still holds a reference to, and removes all references to them. Tr@r,N)r!cancelr3rrSrTrFrE)r%r3r[rr r rrCs zBaseConnectionPool.clearcCs |dSr)rCr%r r r__del__szBaseConnectionPool.__del__)F)r r r rr$r8r2r;rKrFrQrXrYrr[rCr_ __classcell__r r r)rrs$%+  & rc@s$eZdZdZddZeddZdS)TpooledConnectionPoolzZA pool which gives out :class:`~eventlet.tpool.Proxy`-based database connections. cCs*t}|||j|j|jf|j|jfSrr1connectrr rrr%r9r r rrR szTpooledConnectionPool.createcOsPt|t}z2ddlm}|j|jf||}|j|ddWS|XdS)Nr)tpool)cursor)Zautowrap_names) rTimeoutr r]eventletreexecutercZProxy)clsr&r r'kwr.rerr r rrcs  zTpooledConnectionPool.connectNr r r __doc__rR classmethodrcr r r rrasrac@s$eZdZdZddZeddZdS)RawConnectionPoolz7A pool which gives out plain database connections. cCs*t}|||j|j|jf|j|jfSrrbrdr r rrR!szRawConnectionPool.createcOs.t|t}z|j||WS|XdSr)rrgr r]rc)rjr&r r'rkr.r r rrc&szRawConnectionPool.connectNrlr r r rrosroc@s0eZdZddZddZddZddZd Zd S) GenericConnectionWrappercCs ||_dSr)rG)r%baseconnr r rr$4sz!GenericConnectionWrapper.__init__cCs |jSr)rG __enter__r^r r rrr=sz"GenericConnectionWrapper.__enter__cCs|j|||Sr)rG__exit__)r%excvaluetbr r rrs@sz!GenericConnectionWrapper.__exit__cCs |jSr)rG__repr__r^r r rrwCsz!GenericConnectionWrapper.__repr__)!Z affected_rowsZ autocommitbeginZ change_userZcharacter_set_namerLcommitrfZdump_debug_infoerrnoerrorZ errorhandlerZget_server_infoZ insert_idliteralZpingqueryrZ select_dbZserver_capabilitiesZset_character_setZset_isolation_levelZset_server_optionZ set_sql_modeZ show_warningsshutdownZsqlstatestatZ store_resultZstring_literal thread_idZ use_resultZ warning_countN)r r r r$rrrsrw _proxy_funcsr r r rrp3s  rpcs&fdd}|_|_d|_|S)Ncst|j||Sr)r0rG)r%r'r( _proxy_funr r _proxy_methodnsz_wrapper.._proxy_methodzGenericConnectionWrapper.) func_namer r )rrr rr_wrapperms   rcsDeZdZdZfddZddZeZddZdd Zd d Z Z S) rUzA connection wrapper where: - the close method returns the connection to the pool instead of closing it directly - ``bool(conn)`` returns a reasonable value - returns itself to the pool if it gets garbage collected cst|||_dSr)r#r$_pool)r%rqpoolr)r rr$s z PooledConnectionWrapper.__init__cCst|dot|jS)NrG)hasattrboolrGr^r r r __nonzero__sz#PooledConnectionWrapper.__nonzero__cCs(d|_z|`Wntk r"YnXdSr)rrGrIr^r r rrHs z PooledConnectionWrapper._destroycCs"|r|jr|j||dS)zReturn the connection to the pool, and remove the reference to it so that you can't use it again through this wrapper object. N)rrYrHr^r r rrLs  zPooledConnectionWrapper.closecCsdSrr r^r r rr_szPooledConnectionWrapper.__del__) r r r rmr$r__bool__rHrLr_r`r r r)rrUzs  rUc@s*eZdZdZd ddZddZddZdS) DatabaseConnectorzr This is an object which will maintain a collection of database connection pools on a per-host basis. NcOs@|st||_|jdkrt|_||_||_||_||_i|_dS)zconstructor *module* Database module to use. *credentials* Mapping of hostname to connect arguments (e.g. username and password) N)r_conn_pool_classConnectionPool_modulerr _credentials _databases)r%module credentialsZ conn_poolr'r(r r rr$s zDatabaseConnector.__init__cCs&||jkr|j|S|jddSdS)Ndefault)rrQ)r%hostr r rcredentials_fors  z!DatabaseConnector.credentials_forcCsf||f}||jkr\|j}||d<||d<||||j|jf|j|}||j|<|j|S)z@Returns a ConnectionPool to the target host and schema. dbr)rrcopyupdaterrrr)r%rZdbnamekeyZ new_kwargsZdbpoolr r rrQs   zDatabaseConnector.get)N)r r r rmr$rrQr r r rrs  r) collectionsr contextlibrrOr1Zeventlet.poolsrrhrrZeventlet.hubs.timerrZeventlet.greenthreadrobjectrXrMr rrrarorrprrrsetattrrUrr r r rs2       q8  &