,iddlmZmZddlmZddlmZddlZdZGddeZ Gdd eZ Gd d e Z Gd d ejZ Gdde Zy))ABCabstractmethod)Enum)CallableN<ceZdZdZdZdZy)Stateclosedopenz half-openN)__name__ __module__ __qualname__CLOSEDOPEN HALF_OPENZ/var/www/html/langgraph-service/venv/lib/python3.12/site-packages/redis/multidb/circuit.pyr r s F DIrr c"eZdZeedefdZejedefdZeedefdZ e jedefdZ eedZ e jed Z ed e deegd ffd Z y ) CircuitBreakerreturncy)z@The grace period in seconds when the circle should be kept open.Nrselfs r grace_periodzCircuitBreaker.grace_period rrcy)z Set the grace period in seconds.Nrrrs rrzCircuitBreaker.grace_periodsrcy)z!The current state of the circuit.Nrrs rstatezCircuitBreaker.staterrr cy)z!Set current state of the circuit.Nrrr s rr zCircuitBreaker.state"rrcy)z&Database associated with this circuit.Nrrs rdatabasezCircuitBreaker.database(rrcy)z*Set database associated with this circuit.Nrrr$s rr$zCircuitBreaker.database.rrcbNcyz6Callback called when the state of the circuit changes.Nrrr's ron_state_changedzCircuitBreaker.on_state_changed4 r) r r rpropertyrfloatrsetterr r r$rr+rrrrrs  e  /// u   \\ 5    __   8-=ue,Ld,R#S  rrc eZdZdZdej fdZedefdZ e jdefdZ ede fdZ e jd e fd Z ed Z e jd Z eded e e gdffdZy)BaseCircuitBreakerz; Base implementation of Circuit Breaker interface. r'c||_tj|jjtj|jj tj |jji|_d|_ yN) _cbr rcloserr r half_open_state_pb_mapper _databaser*s r__init__zBaseCircuitBreaker.__init__?sQ LL$((.. JJ OOTXX//!  rrc.|jjSr3r4 reset_timeoutrs rrzBaseCircuitBreaker.grace_periodHsxx%%%rrc&||j_yr3r;rs rrzBaseCircuitBreaker.grace_periodLs!-rcVt|jjjSN)value)r r4r namers rr zBaseCircuitBreaker.statePs488>>..//rr c,|j|yr3)r7r"s rr zBaseCircuitBreaker.stateTs$e$&rc|jSr3r8rs rr$zBaseCircuitBreaker.databaseXs ~~rc||_yr3rDr&s rr$zBaseCircuitBreaker.database\s !rrNcyr)rr*s rr+z#BaseCircuitBreaker.on_state_changed`r,r)r r r__doc__ pybreakerrr9r-r.rr/r r r$rrr+rrrr1r1:s933&e&&...0u00 \\'5''__"" 8-=ue,Ld,R#S  rr1c2eZdZdZdeeeegdffdZdZy) PBListenerzDWrapper for callback to be compatible with pybreaker implementation.r'Nc ||_||_y)z Initialize a PBListener instance. Args: cb: Callback function that will be called when the circuit breaker state changes. database: Database instance associated with this circuit breaker. N)r4r8)rr'r$s rr9zPBListener.__init__is!rct|}|j|_t|j}t|j}|j |||yr?)PBCircuitBreakerAdapterr8r$r rAr4)rr' old_state new_states r state_changezPBListener.state_changeysC $R (nn  /  /  Y *r) r r rrGrrr r9rPrrrrJrJfs(N" neU3T9 :" +rrJcTeZdZdejffd ZdedeegdffdZxZ S)rMr'c$t||y)a Initialize a PBCircuitBreakerAdapter instance. This adapter wraps pybreaker's CircuitBreaker implementation to make it compatible with our CircuitBreaker interface. Args: cb: A pybreaker CircuitBreaker instance to be adapted. N)superr9)rr' __class__s rr9z PBCircuitBreakerAdapter.__init__s rrNcft||j}|jj|yr3)rJr$r4 add_listener)rr'listeners rr+z(PBCircuitBreakerAdapter.on_state_changeds$b$--0 h'r) r r rrHrr9rr r+ __classcell__)rTs@rrMrMs3 933 (8-=ue,Ld,R#S(rrM)abcrrenumrtypingrrHDEFAULT_GRACE_PERIODr rr1CircuitBreakerListenerrJrMrrrr^s\#D ' S' T) ) X+11+6(0(r