o Represents an async listener for given event object. rcdSrrrrrrrz"AsyncEventListenerInterface.listenNrrrrrr#r"r#c@s^eZdZdZedefddZedefddZedee ee e e e fffdd Zd S) EventDispatcherInterfacezf Represents a dispatcher that dispatches events to listeners associated with given event. rcCrrrrrrrdispatch*rz!EventDispatcherInterface.dispatchcr$rrrrrrdispatch_async.r%z'EventDispatcherInterface.dispatch_asyncmappingscCsdS)zRegister additional listeners.Nr)rr)rrrregister_listeners2s z+EventDispatcherInterface.register_listenersN)rrrr rr!r'r(rrrr rr#r*rrrrr&$sr&cs*eZdZdZdedeffdd ZZS)EventExceptionzM Exception wrapper that adds an event object into exception context. exceptionrcs||_||_t|dSr)r,rsuper__init__)rr,r __class__rrr.CszEventException.__init__)rrrr Exceptionr!r. __classcell__rrr/rr+>sr+c@steZdZ d deeeeeeffddZ defddZ defdd Z d eeeee ee fffd d ZdS)EventDispatcherNevent_listenersc Cs\ttgttgttgttgtt gt t gi|_ t |_d|_|r,||dSdS)z] Dispatcher that dispatches events to listeners associated with given event. N)AfterConnectionReleasedEventReAuthConnectionListener(AfterPooledConnectionsInstantiationEvent"RegisterReAuthForPooledConnections'AfterSingleConnectionInstantiationEvent!RegisterReAuthForSingleConnection'AfterPubSubConnectionInstantiationEventRegisterReAuthForPubSub#AfterAsyncClusterInstantiationEvent"RegisterReAuthForAsyncClusterNodes!AsyncAfterConnectionReleasedEventAsyncReAuthConnectionListener_event_listeners_mapping threadingLock_lock _async_lockr*)rr4rrrr.Ks*  zEventDispatcher.__init__rcCsR|j|jt|g}|D]}||qWddS1s"wYdSr)rDrAgettyperrr listenerslistenerrrrr'ms  "zEventDispatcher.dispatchc s|jdur t|_|j4IdH"|jt|g}|D] }||IdHqWdIdHdS1IdHs:wYdSr)rEasynciorCrArFrGrrHrrrr(ts  .zEventDispatcher.dispatch_asyncr)cCsp|j+|D]}||jvrtt|j||||j|<q|||j|<qWddS1s1wYdSr)rDrAlistset)rr) event_typerrrr*~s  "z"EventDispatcher.register_listenersr)rrrrrrr!rrr.r'r(r r#r*rrrrr3Is  " r3c@s$eZdZdZddZeddZdS)r5zA Event that will be fired before each command execution. cCs ||_dSr _connection)r connectionrrrr. z%AfterConnectionReleasedEvent.__init__cC|jSrrOrrrrrQz'AfterConnectionReleasedEvent.connectionN)rrrr r.propertyrQrrrrr5s r5c@ eZdZdS)r?Nrrrrrrrr?r?c@seZdZdZdZdS) ClientType)sync)asyncN)rrrSYNCASYNCrrrrrZsrZc@sfeZdZdZ ddededeefddZe dd Z e d efd d Z e d e edffd dZ dS)r7zQ Event that will be fired after pooled connection instances was created. Nconnection_pools client_typecredential_providercC||_||_||_dSr)_connection_pools _client_type_credential_provider)rr_r`rarrrr. z1AfterPooledConnectionsInstantiationEvent.__init__cCrSr)rcrTrrrr_rUz9AfterPooledConnectionsInstantiationEvent.connection_poolsreturncCrSrrdrTrrrr`rUz4AfterPooledConnectionsInstantiationEvent.client_typecCrSrrerTrrrrarUzeZdZddZdefddZdefddZd efd d Z d S) r>cCrrrrTrrrr.rRz+RegisterReAuthForAsyncClusterNodes.__init__rcCs6t|jtr||_|j|j|j|jdSdSr)rrar rrrrrrrrrrs z)RegisterReAuthForAsyncClusterNodes.listenrcs,|jjD]}|jj||IdHqdSr)rrrr)rrkeyrrrrs z+RegisterReAuthForAsyncClusterNodes._re_authrcrrrrrrrrrz2RegisterReAuthForAsyncClusterNodes._raise_on_errorN) rrrr.r=rr rr1rrrrrr>s r>c@sZeZdZddZdefddZdefddZdefd d Zd e fd d Z d e fddZ dS)r<cCs"d|_d|_d|_d|_d|_dSr)rPrordrkrrTrrrr.s  z RegisterReAuthForPubSub.__init__rcCst|jjtrLt|jdrN||_|j|_|j|_ |j |_ |j |_ |j tjkr:|jj|j|jj|jdS|jj|j|jj|jdSdSdS)N)rrprar r get_protocolrrPrqror`rdrjrkrZr]rrrrrrrrrrrs$ zRegisterReAuthForPubSub.listenrcCsZ|j|jd|d||jWdn1s wY|j|dSrrkrPrrrrrorrrrrrs z RegisterReAuthForPubSub._re_authc s|j4IdH#|jd|d|IdH|jIdHWdIdHn 1IdHs1wY|j|IdHdSrrrrrrrs (z&RegisterReAuthForPubSub._re_auth_asyncrcCrrrrrrrrrz'RegisterReAuthForPubSub._raise_on_errorcrrrrrrrrrz-RegisterReAuthForPubSub._raise_on_error_asyncN) rrrr.r;rr rrr1rrrrrrr<s  r<c@s*eZdZdZeddddefddZdS) &InitializeConnectionCountObservabilityzC Listener that initializes connection count observability. z{Connection count is now tracked via record_connection_count(). This functionality will be removed in the next major versionz7.4.0)reasonversionrcCstt|jdSr)r rr_rrrrrs z-InitializeConnectionCountObservability.listenN)rrrr rr7rrrrrrsr).rKrBabcrrenumrtypingrrrrr redis.auth.tokenr redis.credentialsr r redis.observability.recorderr r redis.utilsrrrr#r&r1r+r3r5r?rZr7r9r;r=rwr|r6r@r8r:r>r<rrrrrs<     I   #-4