+ ~j^RIt^RIt^RIt^RIHtHtHt^RIHt^RI H t H t ^RI H t ^RIHt] !R4t!RR ] ],4tR#) N)AsyncGenerator AwaitableCallable)asynccontextmanager)GenericTypeVar)logger)aioTc a]tRt^toRtRRRRRRRRRR /V3R lR lltV3R lR ltV3RlRlt]V3RlRl4t V3RlRlt V3RlRlt V3RlRlt V3RlRlt V3RlRltV3RlRltV3RlRltR tVtR#)!ConnectionPoolzHelper class to manage persistent connections like websockets. Handles connection pooling and reconnection after max duration. Can be used as an async context manager to automatically return connections to the pool. max_session_durationNmark_refreshed_on_getF connect_cbclose_cbconnect_timeoutg$@c <V^8dQhRS[R,RS[RS[S[.S[S[,3,R,RS[S[.S[R,3,R,RS[RR/#)rNrrrrreturn)floatboolrrr )format __classdict__s"y/Users/mitch_tango/dev/rabbit-r1-livekit/agent/.venv/lib/python3.14/site-packages/livekit/agents/utils/connection_pool.py __annotate__ConnectionPool.__annotate__sw!2!2$dl!2 $ !2 eWil23d: !2 A3 $/047 !2!2 !2c WnW nW0nW@n/Vn\ 4VnWPn\P!4Vn \ 4Vn RVn RVn RVnR#)aInitialize the connection wrapper. Args: max_session_duration: Maximum duration in seconds before forcing reconnection mark_refreshed_on_get: If True, the session will be marked as fresh when get() is called. only used when max_session_duration is set. connect_cb: Optional async callback to create new connections close_cb: Optional async callback to close connections NF)_max_session_duration_mark_refreshed_on_get _connect_cb _close_cb _connectionsset _available_connect_timeoutasyncioLock _connect_lock _to_close _prewarm_tasklast_acquire_timelast_connection_reused)selfrrrrrs&$$$$$r__init__ConnectionPool.__init__se"&:"&;#%!,."%% /$\\^"%EI),,1#rc&<V^8dQhRS[RS[/#rtimeoutrrr )rrs"rrr8s  e  rc "VPf \R4hVPV4GRjxL p\P!4VPV&V#L(5i)zCreate a new connection. Returns: The new connection object Raises: NotImplementedError: If no connect callback was provided Nz.Must provide connect_cb or implement connect())r"NotImplementedErrortimer$)r/r4 connections&& r_connectConnectionPool._connect8sS    #%&VW W++G44 (, *%5s.AA)Ac<V^8dQhRR/#rrN)rrs"rrrGsPPtPrc "VP'd7VPP4pVPV4GRjxL KHR#L \d'p\P !RT 2TR7Rp?KxRp?ii;i5i)z7Drain and close all the connections queued for closing.Nzerror closing connection: )exc_info)r+pop_maybe_close_connection Exceptionr warning)r/connes& r_drain_to_closeConnectionPool._drain_to_closeGsnnnn>>%%'D P2248889 P!;D6BQOO PsEBBAA A B A BA;5B;BBc:<V^8dQhRS[RS[S[R3,/#)rr4rN)rrr )rrs"rrrQs$  5 ^AtG5L rc "VPVR7GRjxL pV5xVPV4R#L \dTPT4hi;i5i)zoGet a connection from the pool and automatically return it when done. Yields: An active connection object r4N)getput BaseExceptionremove)r/r4rEs&$ rr9ConnectionPool.connectionPsUXXgX.. J HHTN/  KK   s"A6A8AAAc&<V^8dQhRS[RS[/#r3r5)rrs"rrr`sEarc 2"VP;_uu_4GRjxL VP4GRjxL \P!4pVP'dVPP 4pVP e)W P V,, VP 8:dCVP'dW P V&RVnRVn TuuRRR4GRjxL #VPV4K\P!4pVPV4GRjxL p\P!4V, VnRVn TuuRRR4GRjxL #ELSEL>LLHL +GRjxL 'giR#;i5i)zeGet an available connection or create a new one if needed. Returns: An active connection object NrTF) r*rGr8r&rAr r$r!r-r.rO perf_counterr:)r/r4nowrEt0s&$ rrLConnectionPool.get`s$ %%%%&&( ( ())+C///**,..6..t448R8RR22225))$/-0D*26D/&%%" D!""$Bw//D%)%6%6%82%=D "*/D '/&%% (&(0)&%%%sFE.FE:E1*E:A!E:E: F+E4,F1> %..& & & & &s #0. 0c$<V^8dQhRS[RR/#rXrY)rrs"rrrs . .1 . .rc VPPV4WP9d:VPP V4VPP VR4R#R#)zRemove a specific connection from the pool. Marks the connection to be closed during the next drain cycle. Args: conn: The connection to reset N)r&discardr$r+r[rAr\s&&rrOConnectionPool.removesO % $$ $ NN  t $    ! !$ - %rc<V^8dQhRR/#r=r>)rrs"rrrs  D rc \VPP44FpVPP V4K VPP 4VP P 4R#)ziClear all existing connections. Marks all current connections to be closed during the next drain cycle. N)listr$keysr+r[clearr&r\s& r invalidateConnectionPool.invalidatesV **//12D NN  t $3 ! rc<V^8dQhRR/#r=r>)rrs"rrrs///rc aSPfSP'dR#RV3Rllp\P!V!44p\P !V4SnR#)zInitiate prewarming of the connection pool without blocking. This method starts a background task that creates a new connection if none exist. The task automatically cleans itself up when the connection pool is closed. NcV^8dQhRR/#r=r>)rs"rr,ConnectionPool.prewarm..__annotate__s . .T .rcR<"SP;_uu_4GRjxL SP'g@SPSPR7GRjxL pSPP V4RRR4GRjxL R#LgL3L +GRjxL 'giR#;i5i)NrK)r*r$r:r'r&r[)rEr/s r _prewarm_impl-ConnectionPool.prewarm.._prewarm_implsh))))(((!%t7L7L!MMDOO''-*))M*)))s\B'BB'2B BB 2 B'=B>B'B B' B$ B B$ B$  B')r,r$r( create_taskweakrefref)r/rqtasksf rprewarmConnectionPool.prewarmsN    )T->->->  . . ""=?3$[[.rc<V^8dQhRR/#r=r>)rrs"rrrs%%d%rc "VPe7VP4pV'd\P!V4GRjxL VP4VP 4GRjxL R#L.L5i)z@Close all connections, draining any pending connection closures.N)r,r gracefully_cancelrjrG)r/rvs& racloseConnectionPool.acloses\    )%%'D++D111 ""$$$2 %s(&A5A5A1'A5*A3+A53A5) r&r#r"r*r'r$r!r r,r+r-r.)__name__ __module__ __qualname____firstlineno____doc__r0r:rGrr9rLrMrBrOrjrwr|__static_attributes____classdictcell__)rs@rr r s !2.2!2', !2 >B !2 ;? !2"&!2!2F  PP  > & &'' . .  //$%%rr )r(r8rtcollections.abcrrr contextlibrtypingrrlogr r r r r>rrrs; ??*# CLv%WQZv%r