ng\jZSSKrSSKrSSKrSSKJrJrJrJrJr SSK J r J r SSK J r SSKJr SSKJr SSKJrJr SSKJr SS KJr SS KJr SS KJr SS KJrJrJ r J!r! SS K"J#r#J$r$J%r% SSK&J'r'J(r(J)r) SSK*J+r+ SSK,J-r- SSK.J/r/ SSK0J1r1J2r2J3r3 SSK4J5r5 \Rl"\75r8\5"SS\\55r9S\4Sjr:"SS\\5r;"SS5r'-'D'D$!'!8!8$22 335K4MN 6"55oo--"55$66!00!33#'#?#? ! !02!(  cFUR5 g![a gf=fN)close ExceptionrKs rL__del__MultiDBClient.__del__Us$  JJL       cURRUR5 URRURUR 5 SnUR Houp#URRUR5 URR[R:XdMTU(aM]X RlSnMq U(d [S5eSUlg)zD Perform initialization of databases to define their initial state. FTz4Initial connection failed - no active database foundN)rF run_coro_sync_perform_initial_health_checkrun_recurring_coror6_check_databases_healthr1circuiton_state_changed!_on_circuit_state_change_callbackstateCBStateCLOSEDrD_active_databaserrE)rKis_active_db_founddatabaseweights rL initializeMultiDBClient.initialize^s (()K)KL --  ' '  ( ( # $ H    - -d.T.T U%%7@R@R:B%%6%)"!0"*F  rOreturncUR$)z5 Returns a sorted (by weight) list of all databases. )r1rTs rL get_databasesMultiDBClient.get_databasessrOreNcSnURHup4X1:XdM Sn O U(d [S5eURRURU5 UR R [R:XaBURRS5SupTU[R4URl g[S5e)z< Promote one of the existing databases to become an active. NT/Given database is not a member of database listrz1Cannot set active database, database is unhealthy)r1 ValueErrorrFrY_check_db_healthr]r`rarb get_top_nrMANUALrDactive_databaser)rKreexists existing_db_highest_weighted_dbs rLset_active_database!MultiDBClient.set_active_databases"ooNK&. NO O (()>)>I    ! !W^^ 3%)__%>%>q%A!%D " !((5D ! ! 1 & ?  rOskip_initial_health_checkc[S[5S9URS'SUR;a[SS9URS'UR(a<UR R R"UR40URD6nOUR(aYURR[S[5S95 UR R RURS9nO&UR R "S 0URD6nURcUR5O URn[UUURURS 9nURRUR U5 UR$R'S 5SupgUR$R)XUR5 UR+XV5 g!["a U(deNkf=f) z Adds a new database to the database list. Args: config: DatabaseConfig object that contains the database configuration. skip_initial_health_check: If True, adds the database even if it is unhealthy. r)retriesbackoffretrymaint_notifications_configF)enabled)connection_poolN)clientr]rfhealth_check_urlro)rr client_kwargsrfrom_urlrJ client_class from_pool set_retryr]default_circuit_breakerrrfrrFrYrqrr1rradd_change_active_database)rKr'r{rr]rerxhighest_weights rL add_databaseMultiDBClient.add_databases).a(MW% (v/C/C C(7  != > ??\\..77#)#7#7F      & &uQ 'L M\\..88 & 0 09F\\..F1E1EFF~~%  * * ,  ==#44       , ,T-B-BH M /3oo.G.G.J1.M+ Hoo6 $$XC * ,- s-&G**G?>G? new_databasehighest_weight_databasecURUR:aKURR[R:Xa"U[ R 4URlgggrQ) rfr]r`rarbr AUTOMATICrDrt)rKrrs rLr%MultiDBClient._change_active_databasesZ   "9"@"@ @$$**gnn<!++5D ! ! 1= ArOcURRU5nURRS5Sup4XB::aKURR[ R :Xa"U[R4URl ggg)z, Removes a database from the database list. rorN) r1removerrr]r`rarbrrsrDrt)rKrerfrxrs rLremove_databaseMultiDBClient.remove_databases~''1.2oo.G.G.J1.M+  $#++11W^^C$!((5D ! ! 1D %rOrfc SnURHupEXA:XdM Sn O U(d [S5eURRS5SupgURRX5 X!lUR X5 g)z, Updates a database from the database list. NTrnror)r1rprr update_weightrfr)rKrerfrurvrwrxrs rLupdate_database_weight$MultiDBClient.update_database_weightsz"ooNK&. NO O.2oo.G.G.J1.M+ %%h7  $$XCrOfailure_detectorc:URRU5 g)z. Adds a new failure detector to the database. N)r;append)rKrs rLadd_failure_detector"MultiDBClient.add_failure_detector s &&'78rO healthcheckcUR URRU5 SSS5 g!,(df  g=f)z* Adds a new health check to the database. N)rIr4r)rKrs rLadd_health_checkMultiDBClient.add_health_checks)]]    & &{ 3]]s 2 Ac|UR(dUR5 URR"U0UD6$)z2 Executes a single command and return its result. )rErgrDexecute_commandrKargsoptionss rLrMultiDBClient.execute_commands3 OO $$44dFgFFrOc[U5$)z* Enters into pipeline mode of the client. )PipelinerTs rLpipelineMultiDBClient.pipeline#s~rOfuncrcUR(dUR5 URR"U/UQUQ76$)z# Executes callable as transaction. )rErgrDexecute_transaction)rKrwatchesrs rL transactionMultiDBClient.transaction)s8 OO $$88RR'RRrOc \UR(dUR5 [U40UD6$)z Return a Publish/Subscribe object. With this object, you can subscribe to channels and listen for messages that get published to them. )rErgPubSub)rKkwargss rLpubsubMultiDBClient.pubsub2s'  OO d%f%%rOc # UR [UR5nSSS5 URR WU5IShvN nU(dIUR R [R:wa[RUR lU$U(aGUR R [R:wa[RUR lU$!,(df  N=fN7f)z? Runs health checks on the given database until first failure. N) rIlistr4r9executer]r`raOPENrb)rKrer2 is_healthys rLrqMultiDBClient._check_db_health=s]] !4!45M 44<<]HUU %%5)0  &  H,,22gnnD%,^^H   "]Vs( DC1'D D B$D1 C?;Dc# 0n/UlURHIup#[R"UR U55nX!U'URR U5 MK [R "URSS06IShvN n[URU5VVs0sH upFXU_M nnnUR5Hiup&[U[5(dMURn[RURl[ R#SUR$S9 SXx'Mk U$Ns snnf7f)zS Runs health checks as a recurring task. Runs health checks against all databases. return_exceptionsTNz%Health check failed, due to exception)exc_infoF) _hc_tasksr1asyncio create_taskrqrgatherzipitems isinstancerrerarr]r`loggerdebugoriginal_exception) rK task_to_dbrerwtaskresultsresult db_results unhealthy_dbs rLr\%MultiDBClient._check_databases_healthPs# 46 ??KH&&t'<' > @ 1   s/B'' B43B4)r?rFrArJr1r@r=r;rIrr6r9r4rDrE)T)'__name__ __module__ __qualname____firstlineno____doc__rrMrUrgrrkrryrboolrrrrfloatrrrrrrrrrrrqdictr\rZrrar_rR__static_attributes__rrOrLr%r%$s, (}(T$ Ly  L T :IM6D$6DAE6Dp ( CO  D|DUD&9_9 4K4G S*t); <S &|& tHdN/C D0L%L29LFML*ArOr%r]c.[RUlgrQ)rarr`)r]s rLrrs%%GMrOc\rSrSr%SrSr\S\S'S\4Sjr SSjr S r S r S\ 4S jrS\4S jrSSjrSSjrSSjrSrS\\4SjrSrg )riz? Pipeline implementation for multiple logical Redis databases. F_is_async_clientrc/UlXlgrQ)_command_stack_client)rKrs rLrMPipeline.__init__s  rOricU$rQrrTs rL __enter__Pipeline.__enter__ rOc$UR5 grQreset)rKexc_type exc_value tracebacks rL__exit__Pipeline.__exit__  rOcFUR5 g![a gf=frQrrSrTs rLrUPipeline.__del__s"  JJL   rWc,[UR5$rQ)rrrTs rL__len__Pipeline.__len__s4&&''rOcg)z1Pipeline instances should always evaluate to TrueTrrTs rL__bool__Pipeline.__bool__srONc/UlgrQ)rrTs rLrPipeline.resets  rOc$UR5 g)zClose the pipelineNrrTs rLrRPipeline.closes  rOc>URRX45 U$)a: Stage a command to be executed when execute() is next called Returns the current Pipeline object back so commands can be chained together, such as: pipe = pipe.set('foo', 'bar').incr('baz').decr('bang') At some other point, you can then run: pipe.execute(), which will execute all commands queued in the pipe. )rrrs rLpipeline_execute_command!Pipeline.pipeline_execute_commands ""D?3 rOc&UR"U0UD6$)zAdds a command to the stack)rrKrrs rLrPipeline.execute_commands,,d=f==rOc(URR(dURR5 URRR [ UR 55UR5 $!UR5 f=f)z0Execute all the commands in the current pipeline)rrErgrDexecute_pipelinetuplerrrTs rLrPipeline.executes_||'' LL # # % <<00AAd))* JJLDJJLs 7A??B)rr)rirriN)rrrrrrr__annotations__r%rMrrrUintrrrrrRrrrrrrrrOrLrrso(-gen,} (($! > c rOrc 2\rSrSrSrS\4SjrS SjrS!SjrS!S jr S!S jr \ S\ 4S j5r S rS \\-S\SS4SjrSrS \\-S\SS4SjrSrS \\-S\SS4SjrSrS"S\ S\4SjjrS"S\ S\4SjjrS#S\S\ S\\S\ SS4 SjjrSrg)$riz* PubSub object for multi database client. rc \XlURRR"S0UD6 g)zInitialize the PubSub object for a multi-database client. Args: client: MultiDBClient instance to use for pub/sub operations **kwargs: Additional keyword arguments to pass to the underlying pubsub implementation Nr)rrDr)rKrrs rLrMPubSub.__init__s$  %%,,6v6rOricU$rQrrTs rLrPubSub.__enter__ rrONcFUR5 g![a gf=frQr rTs rLrUPubSub.__del__s$  JJL   rWcLURRRS5$)NrrrDexecute_pubsub_methodrTs rLr PubSub.resets||,,BB7KKrOc$UR5 grQrrTs rLrR PubSub.closer rOcVURRRR$rQ)rrD active_pubsub subscribedrTs rLr3PubSub.subscribeds||,,::EEErOcPURRR"S/UQ76$)Nrr,rKrs rLrPubSub.execute_command!s*||,,BB  $  rOrrcVURRR"S/UQ70UD6$)a Subscribe to channel patterns. Patterns supplied as keyword arguments expect a pattern name as the key and a callable as the value. A pattern's callable will be invoked automatically when a message is received on that pattern rather than producing a message via ``listen()``. psubscriber,rs rLr9PubSub.psubscribe&4||,,BB   #)  rOcPURRR"S/UQ76$)zR Unsubscribe from the supplied patterns. If empty, unsubscribe from all patterns. punsubscriber,r6s rLr=PubSub.punsubscribe4, ||,,BB  !  rOcVURRR"S/UQ70UD6$)a" Subscribe to channels. Channels supplied as keyword arguments expect a channel name as the key and a callable as the value. A channel's callable will be invoked automatically when a message is received on that channel rather than producing a message via ``listen()`` or ``get_message()``. subscriber,rs rLrAPubSub.subscribe=s4||,,BB   "(  rOcPURRR"S/UQ76$)zQ Unsubscribe from the supplied channels. If empty, unsubscribe from all channels unsubscriber,r6s rLrDPubSub.unsubscribeKs% ||,,BB=XSWXXrOcVURRR"S/UQ70UD6$)aJ Subscribes the client to the specified shard channels. Channels supplied as keyword arguments expect a channel name as the key and a callable as the value. A channel's callable will be invoked automatically when a message is received on that channel rather than producing a message via ``listen()`` or ``get_sharded_message()``. ssubscriber,rs rLrGPubSub.ssubscribeRr;rOcPURRR"S/UQ76$)z] Unsubscribe from the supplied shard_channels. If empty, unsubscribe from all shard_channels sunsubscriber,r6s rLrJPubSub.sunsubscribe`r?rOignore_subscribe_messagestimeoutcLURRRSUUS9$)z Get the next message if one is available, otherwise None. If timeout is specified, the system will wait for `timeout` seconds before returning. Timeout should be specified as a floating point number, or None, to wait indefinitely. get_messagerLrMr,rKrLrMs rLrOPubSub.get_messageis0||,,BB &?C  rOcLURRRSUUS9$)z Get the next message if one is available in a sharded channel, otherwise None. If timeout is specified, the system will wait for `timeout` seconds before returning. Timeout should be specified as a floating point number, or None, to wait indefinitely. get_sharded_messagerPr,rQs rLrTPubSub.get_sharded_messageys0||,,BB !&?C  rO sleep_timedaemonexception_handlersharded_pubsubr cPURRRUUUUUS9$)N)rWrXrrY)rrDexecute_pubsub_run)rKrVrWrXrYs rL run_in_threadPubSub.run_in_threads6||,,?? /) @  rO)r)rirr!)F)r^FNF)rrrrrr%rMrrUrrRpropertyrr3rr r"r!r9r=rArDrGrJrrOrTrrr\rrrOrLrrsF 7} 7LFDFF   ,  8E        ,  8E    Y   ,  8E     IL )- @E "IL )- @E $ 04$      $H-       rOr)=rloggingrGtypingrrrrr!redis.asyncio.multidb.healthcheckrr redis.backgroundr redis.backoffr redis.clientr redis.commandsr rredis.maint_notificationsrredis.multidb.circuitrrraredis.multidb.command_executorrredis.multidb.configrrrrredis.multidb.databaserrrredis.multidb.exceptionrrrredis.multidb.failure_detectorrredis.observability.attributesr redis.retryr redis.typingr r!r" redis.utilsr# getLoggerrrr%rrrrrOrLrss99L0#+<>02A ED ;<>>$   8 $JA'JAJAZ &&B"LBJ[ [ rO