!ni<jddlmZddlmZddlZddlZddlmZddlm Z ddlm Z ddl m Z ddl mZeZGd d eZd ZGd d eZGddeZGddeZeZGddZej2D]ZefdZeeeeee`[[GddeZGddZy))deque)contextmanagerN)Pool)timeout)hubs)Timer) GreenThreadc eZdZy)ConnectTimeoutN)__name__ __module__ __qualname__P/home/homepc/tiktok-worker/venv/lib/python3.12/site-packages/eventlet/db_pool.pyr r srr c$|jyN)rollback)conns rcleanup_rollbackrs MMOrceZdZdddddeffd ZdZdZd Zd Zdd Z fd Z e ffd Z e e fdZdZdZxZS)BaseConnectionPoolr c|sJ||_||_| |_||_||_||_d|_||_t |%||dy)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_sizeorder_as_stack) _db_module_args_kwargsmax_idlemax_ageconnect_timeout_expiration_timercleanupsuper__init__) self db_modulerrr$r%r&r(argskwargs __class__s rr*zBaseConnectionPool.__init__s[4y#     .!%  (XdSrc|jdk(s|jdk(ry|jt|jddsy t j}|j ||j dd|z |jz}t|j Dcgc]}|d c}}||z |jz}t||}|dkDrht|ttjjj|j gi|_|jj#yycc}w#ttf$r d|_YywxYw)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. rNcalledF)r%r$r'getattrtime_expire_old_connections free_itemsmin IndexError ValueErrorrr rget_hubgreenletswitch_schedule_expirationschedule)r+now idle_delaytoldest age_delay next_delays rr>z'BaseConnectionPool._schedule_expiration>s9 <<1   2   " " . 6 6%H  ))+C  ( ( -//"-a036$--GJ81!A$89F#5IZ3J >%*:{4<<>CZCZ7[7b7b+/+D+Db"&ND "  " " + + - 9J' %)D "  s%AE D>+%E>EEEc t|j}|jDcgc]\}}}|j|||r|}}}}|jDcgc]\}}}|j|||s|||f }}}}|jj|jj ||xj |t|jz zc_|D]}|j |dycc}}}wcc}}}w)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(). TquietN)lenr7 _is_expiredclearextend current_size _safe_close)r+r@original_count last_used created_atrexpirednew_frees rr6z*BaseConnectionPool._expire_old_connectionsisT__-04==+ :tY ; ==04AA+ :t##CJ? D )AA  x( ^c$//.BBB /D   T  . /!= As C<#Dc|jdks3|jdks$||z |jkDs||z |jkDryy)z@Returns true and closes the connection if it's expired. rTF)r$r%)r+r@rPrQs rrJzBaseConnectionPool._is_expiredsB MMQ $,,!"3?T]]2#dll2rcrd} |r|j}|j|Sd} |S#t$rY|SwxYw)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+rbases r_unwrap_connectionz%BaseConnectionPool._unwrap_connectionsQ zz        s)) 66c |jy#t$rYyt$r*|s%tdt j dzYyYywxYw)zVCloses the (already unwrapped) connection, squelching any exceptions. zConnection.close raised: %sr3N)closerX Exceptionprintsysexc_info)r+rrHs rrNzBaseConnectionPool._safe_closesN K JJL   K3s||~a7HIJ Ks A-AAc t|}| |j}t |t r|\}}}ntj}t||}||_ |S#t$r|xjdzc_wxYw)Nr3) r)getcreater]rM isinstancetupler5PooledConnectionWrapper_db_pool_created_at)r+r _last_usedrQwrappedr/s rrbzBaseConnectionPool.getsw{} < {{} dE "+/ (J DJ*$5'1#)  !!Q&!  s A## Bct|dd}tj}|j|}|j|||r|j |dd}n!||t ur |j } |r|||t|)|||fn8|jdkDrt|)dn|xjdzc_ |jy#t$r}td|d|d}Yd}~d}~wd}xYw)NrgrFrGzWARNING: cleanup z raised: r3)r4r5rZrJrN_MISSINGr(r]r^r)putwaitingrMr>)r+rr(rQr@er/s rrlzBaseConnectionPool.putsT#8!< iik&&t,   Cj 1   T  /D  (",, DM   GKj$/ 0||~! D!!!Q&! !!#' '1EF s4 C D%C==Dc#K|j} ||j||y#|j||wxYww)N)r()rbrl)r+r(rs ritemzBaseConnectionPool.items;xxz ,J HHT7H +DHHT7H +sA,AAAc|jr|jj|jtc}|_|D]A}t |t r|dn|}|j |d|xjdzc_Cy)zuClose all connections that this pool still holds a reference to, and removes all references to them. TrGr3N)r'cancelr7rrdrerNrM)r+r7rprs rrKzBaseConnectionPool.clearsz  ! !  " " ) ) +&*oouw# DO #D(u5474D   T  .    "   #rc$|jyr)rKr+s r__del__zBaseConnectionPool.__del__s  r)F)r r rrr*r>r6rJrZrNrbrkrlrrprKrv __classcell__r/s@rrrsfab!") #TJ).V/6$ K<!)$$L#,, #rrc&eZdZdZdZedZy)TpooledConnectionPoolzZA pool which gives out :class:`~eventlet.tpool.Proxy`-based database connections. ctj}|||j|j|jg|ji|j fSrr5connectr!r&r"r#r+r@s rrczTpooledConnectionPool.create UiikC OOT11P48JJPBF,,PP Prctj|t} ddlm}|j |j g|i|}|j|d|jS#|jwxYw)Nr)tpool)cursor)autowrap_names) rTimeoutr eventletrexecuter}Proxyrs)clsr,r&r-kwrBrrs rr}zTpooledConnectionPool.connects` OOO^-= >  & 5==!2!2@T@R@D;;tK;@ HHJAHHJs 8A))A;Nr r r__doc__rc classmethodr}rrrrzrzs!P rrzc&eZdZdZdZedZy)RawConnectionPoolz7A pool which gives out plain database connections. ctj}|||j|j|jg|ji|j fSrr|r~s rrczRawConnectionPool.create!rrctj|t} |j|i||j S#|j wxYwr)rrr r}rs)rr,r&r-rrBs rr}zRawConnectionPool.connect&sB OOO^-= > $9$$d1b1 HHJAHHJs AANrrrrrrs!P rrc(eZdZdZdZdZdZdZy)GenericConnectionWrapperc||_yr)rV)r+baseconns rr*z!GenericConnectionWrapper.__init__4s  rc6|jjSr)rV __enter__rus rrz"GenericConnectionWrapper.__enter__=szz##%%rc<|jj|||Sr)rV__exit__)r+excvaluetbs rrz!GenericConnectionWrapper.__exit__@szz""3r22rc6|jjSr)rV__repr__rus rrz!GenericConnectionWrapper.__repr__Cszz""$$r)! affected_rows autocommitbegin change_usercharacter_set_namer\commitrdump_debug_infoerrnoerror errorhandlerget_server_info insert_idliteralpingqueryr select_dbserver_capabilitiesset_character_setset_isolation_levelset_server_option set_sql_mode show_warningsshutdownsqlstatestat store_resultstring_literal thread_id use_result warning_countN)r r rr*rrr _proxy_funcsrrrrr3s&3%"LrrcBfd}|_|_dz|_|S)Nc<t|j|i|Sr)r4rV)r+r-r. _proxy_funs r _proxy_methodz_wrapper.._proxy_methodns 274::z2DCFC CrzGenericConnectionWrapper.) func_namer r)rrs` r_wrapperrms, D", !+ %@:%M "rc>eZdZdZfdZdZeZdZdZdZ xZ S)rfzA 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 c2t||||_yr)r)r*_pool)r+rpoolr/s rr*z PooledConnectionWrapper.__init__s " rcHt|dxrt|jS)NrV)hasattrboolrVrus r __nonzero__z#PooledConnectionWrapper.__nonzero__sg&;4 +;||f}||jvr}|jj}||d<||d<|j|j ||j |j g|ji|}||j|<|j|S)z@Returns a ConnectionPool to the target host and schema. dbr)rr#copyupdaterrrr")r+rdbnamekey new_kwargsdbpools rrbzDatabaseConnector.getsVn doo %**,J%Jt !%Jv    d2248 9*T** 9#zz9-79F#)DOOC s##rr)r r rrr*rrbrrrrrs  &: $rr) collectionsr contextlibrr_r5eventlet.poolsrrrreventlet.hubs.timerreventlet.greenthreadr objectrkr]r rrrzrrrrrrsetattrrfrrrrrs% %, 8 Y mm`.**&'55p+77 HJ& $j(:2FG H) !6!L,$,$r