Ë
|
|
çúh^G ã ój d dl Z d dlZd dlmZ d dlmZ d dlmZmZm Z m
|
Z
|
d dl m Z d dl mZ d dlmZmZ d dlmZ d d lmZ d d
|
lmZ d d lmZmZ d d lm Z m Z m Z d d l m Z m!Z! d dl"m#Z# d dl$m%Z%m&Z& d dl'm(Z( e jR e*« Z+e( G d dee« « Z,defdZ- G d dee« Z. G d d« Z/y)é N)Ú as_completed)ÚThreadPoolExecutor)ÚAnyÚCallableÚListÚOptional)ÚBackgroundScheduler)ÚPubSubWorkerThread)Ú CoreCommandsÚRedisModuleCommands)ÚCircuitBreaker)ÚState)ÚDefaultCommandExecutor)ÚDEFAULT_GRACE_PERIODÚ MultiDbConfig)ÚDatabaseÚ DatabasesÚ SyncDatabase)ÚNoValidDatabaseExceptionÚUnhealthyDatabaseException)ÚFailureDetector)Ú HealthCheckÚHealthCheckPolicy)Ú experimentalc óü e Zd ZdZdefdZd ZdefdZde ddfd Z
|
de fd
|
Z d e d e fd Z de fdZde defdZdefdZdefdZd Zd Zdedgdf fdZd Zde defdZd#d ee gdf fd Z d e d e d e fd!Z d" Z!y)$Ú MultiDBClientzx
|
Client that operates on multiple logical Redis databases.
|
Should be used in Active-Active database setups.
|
Úconfigc
|
óX |j « | _ |j « | _ |j %| j j |j « |j | _ |j j |j |j « | _ |j « | _ |j %| j j |j « |j |j# « n |j | _ | j$ j' | j « |j( | _ |j, | _ |j0 | _ | j2 j5 t6 f« t9 | j | j | j2 | j$ |j: |j< | j. | j* ¬« | _ d| _ tC jD « | _# tI « | _% || _& y )N)Úfailure_detectorsÚ databasesÚ command_retryÚfailover_strategyÚfailover_attemptsÚfailover_delayÚevent_dispatcherÚauto_fallback_intervalF)'r Ú
|
_databasesÚdefault_health_checksÚ_health_checksÚ health_checksÚextendÚhealth_check_intervalÚ_health_check_intervalÚhealth_check_policyÚvalueÚhealth_check_probesÚhealth_check_probes_delayÚ_health_check_policyÚdefault_failure_detectorsÚ_failure_detectorsr r" Údefault_failover_strategyÚ_failover_strategyÚ set_databasesr& Ú_auto_fallback_intervalr% Ú_event_dispatcherr! Ú_command_retryÚupdate_supported_errorsÚConnectionRefusedErrorr r# r$ Úcommand_executorÚ initializedÚ threadingÚRLockÚ_hc_lockr Ú _bg_schedulerÚ_config)Úselfr s õg H:\项ç®\archive\æµè¯ç»\èæ¬\Change_password\venv_build\Lib\site-packages\redis/multidb/client.pyÚ__init__zMultiDBClient.__init__ sÀ Ø ×*Ñ*Ó,Ø $× :Ñ :Ó <Ôà × Ñ Ð +Ø × Ñ × &Ñ & v×';Ñ';Ô <à&,×&BÑ&BÔ#Ø7=×7QÑ7Q×7WÑ7WØ × &Ñ &¨×(HÑ(Hó8
|
Ô!ð #)×"BÑ"BÓ"DÔ à × #Ñ #Ð /Ø × #Ñ #× *Ñ *¨6×+CÑ+CÔ Dð ×'Ñ'Ð/ð × ,Ñ ,Ô .à×)Ñ)ð Ô ð
|
× Ñ ×-Ñ-¨d¯o©oÔ>Ø'-×'DÑ'DÔ$Ø!'×!8Ñ!8Ô Ø $× 2Ñ 2ÔØ ×Ñ×3Ñ3Ô5KÐ4MÔNÜ 6Ø "× 5Ñ 5ØooØ ×-Ñ-Ø "× 5Ñ 5Ø $× 6Ñ 6Ø!×0Ñ0Ø !× 3Ñ 3Ø#'×#?Ñ#?ô !
|
Ô ð !ÔÜ!Ó) Ü 0Ó 2ÔØ ó c ó¨ d }| j |¬« | j j | j | j « d}| j D ]h \ }}|j
|
j | j « |j
|
j t j k( sS|rV|| j _ d}j |s t d« d| _ y)zT
|
Perform initialization of databases to define their initial state.
|
c ó | ©N© )Úerrors rE Ú raise_exception_on_failed_hcz>MultiDBClient.initialize.<locals>.raise_exception_on_failed_hcL s ØKrG )Úon_errorFTz4Initial connection failed - no active database foundN)Ú_check_databases_healthrB Ú run_recurringr- r' ÚcircuitÚon_state_changedÚ!_on_circuit_state_change_callbackÚstateÚCBStateÚCLOSEDr= Úactive_databaser r> )rD rM Úis_active_db_foundÚdatabaseÚweights rE Ú
|
initializezMultiDBClient.initializeG sÍ ò
|
ð ×$Ñ$Ð.JÐ$ÔKð ×Ñ×(Ñ(Ø × 'Ñ 'Ø × (Ñ (ô
|
ð
|
#Ðà $§¡ò *Ñ Hfà × Ñ × -Ñ -¨d×.TÑ.TÔ Uð × Ñ ×%Ñ%¬¯ © Ó7Ò@RØ8@×%Ñ%Ô5Ø%)Ñ"ð *ñ "Ü*ØFóð ð ÕrG Úreturnc ó | j S )zE
|
Returns a sorted (by weight) list of all databases.
|
)r' ©rD s rE Ú get_databaseszMultiDBClient.get_databasesj s ð Ð rG rY Nc óF d}| j D ] \ }}||k( s d} n |s t d« | j |« |j j t
|
j k( r3| j j d« d \ }}|| j _ yt d« )zL
|
Promote one of the existing databases to become an active.
|
NTú/Given database is not a member of database listé r z1Cannot set active database, database is unhealthy) r' Ú
|
ValueErrorÚ_check_db_healthrQ rT rU rV Ú get_top_nr= rW r )rD rY ÚexistsÚ existing_dbÚ_Úhighest_weighted_dbs rE Úset_active_databasez!MultiDBClient.set_active_databasep s§ ð à "ooò NKØhÓ&Ø Ùð ñ
|
Ü Ð NÓOÐ Oà × Ñ hÔ'à × Ñ × !Ñ !¤W§^¡^Ò 3Ø%)§_¡_×%>Ñ%>¸qÓ%AÀ!Ñ%DÑ "Ð Ø4<D× !Ñ !Ô 1Ø ä&Ø ?ó
|
ð
|
rG c ó | j D ] \ }}||k( s t d« | j |« | j j d« d \ }}| j j ||j
|
« | j ||« y)z;
|
Adds a new database to the database list.
|
z Given database already existsrb r N)r' rc rd re ÚaddrZ Ú_change_active_database)rD rY rg rh ri Úhighest_weights rE Ú add_databasezMultiDBClient.add_database s ð #ooò BNKØhÓ&Ü Ð!@ÓAÐAð Bð × Ñ hÔ'à.2¯o©o×.GÑ.GÈÓ.JÈ1Ñ.MÑ+Ð^Ø ×ÑH h§o¡oÔ6Ø ×$Ñ$ XÐ/BÕCrG Ú new_databaseÚhighest_weight_databasec óª |j |j kD r:|j j t j k( r|| j
|
_ y y y rJ )rZ rQ rT rU rV r= rW )rD rp rq s rE rm z%MultiDBClient._change_active_database sJ ð × Ñ Ð"9×"@Ñ"@Ò @Ø ×$Ñ$×*Ñ*¬g¯n©nÒ<à4@D× !Ñ !Õ 1ð =ð ArG c óú | j j |« }| j j d« d \ }}||k r:|j j t
|
j k( r|| j _ yyy)z<
|
Removes a database from the database list.
|
rb r N) r' Úremovere rQ rT rU rV r= rW )rD rY rZ ri rn s rE Úremove_databasez MultiDBClient.remove_database sr ð ×'Ñ'¨Ó1Ø.2¯o©o×.GÑ.GÈÓ.JÈ1Ñ.MÑ+Ð^ð fÒ $Ø#×+Ñ+×1Ñ1´W·^±^ÒCà4GD× !Ñ !Õ 1ð Dð %rG rZ c ó d}| j D ] \ }}||k( s d} n |s t d« | j j d« d \ }}| j j ||« ||_ | j ||« y)z<
|
Updates a database from the database list.
|
NTra rb r )r' rc re Ú update_weightrZ rm )rD rY rZ rf rg rh ri rn s rE Úupdate_database_weightz$MultiDBClient.update_database_weight s ð à "ooò NKØhÓ&Ø Ùð ñ
|
Ü Ð NÓOÐ Oà.2¯o©o×.GÑ.GÈÓ.JÈ1Ñ.MÑ+Ð^Ø ×%Ñ% h°Ô7Ø Ø ×$Ñ$ XÐ/BÕCrG Úfailure_detectorc ó: | j j |« y)z>
|
Adds a new failure detector to the database.
|
N)r4 Úappend)rD ry s rE Úadd_failure_detectorz"MultiDBClient.add_failure_detectorÀ s ð × Ñ ×&Ñ&Ð'7Õ8rG Ú healthcheckc ó| | j 5 | j j |« ddd« y# 1 sw Y yxY w)z:
|
Adds a new health check to the database.
|
N)rA r) r{ )rD r} s rE Úadd_health_checkz MultiDBClient.add_health_checkÆ s4 ð ]]ñ 4Ø × Ñ × &Ñ & {Ô 3÷ 4÷ 4ñ 4ús 2²;c ór | j s| j « | j j |i |¤S )zB
|
Executes a single command and return its result.
|
)r> r[ r= Úexecute_command©rD ÚargsÚoptionss rE r z MultiDBClient.execute_commandÍ s5 ð × Ò Ø OOÔ à4t×$Ñ$×4Ñ4°dÐF¸gÑFÐFrG c ó t | « S )z:
|
Enters into pipeline mode of the client.
|
)ÚPipeliner^ s rE ÚpipelinezMultiDBClient.pipelineÖ s ô ~Ð rG Úfuncr c óx | j s| j « | j j |g|¢|¢ S )z3
|
Executes callable as transaction.
|
)r> r[ r= Úexecute_transaction)rD r Úwatchesr s rE Ú transactionzMultiDBClient.transactionÜ s: ð × Ò Ø OOÔ à8t×$Ñ$×8Ñ8¸ÐRÀÐRÈ'ÒRÐRrG c óR | j s| j « t | fi |¤S )z¨
|
Return a Publish/Subscribe object. With this object, you can
|
subscribe to channels and listen for messages that get published to
|
them.
|
)r> r[ ÚPubSub)rD Úkwargss rE ÚpubsubzMultiDBClient.pubsubå s' ð × Ò Ø OOÔ ädÑ%fÑ%Ð%rG c óv | j j | j |« }|sH|j j t
|
j k7 r t
|
j |j _ |S |rF|j j t
|
j k7 r t
|
j |j _ |S )zO
|
Runs health checks on the given database until first failure.
|
)r2 Úexecuter) rQ rT rU ÚOPENrV )rD rY Ú
|
is_healthys rE rd z MultiDBClient._check_db_healthð s ð
|
×.Ñ.×6Ñ6°t×7JÑ7JÈHÓU
|
áØ× Ñ ×%Ñ%¬¯ © Ò5Ü)0¯ © × Ñ Ô&Ø Ð Ù H× ,Ñ ,× 2Ñ 2´g·n±nÒ DÜ%,§^¡^H× Ñ Ô "àÐrG rN c ób t t | j « ¬« 5 }| j D ch c]! \ }}|j | j |« # }}} t || j ¬« D ] } |j « ddd« yc c}}w # t $ rj}|j }t j |j _ t j d|j ¬« |r ||j « Y d}~d}~ww xY w# t" $ r t# d« w xY w# 1 sw Y yxY w)zk
|
Runs health checks as a recurring task.
|
Runs health checks against all databases.
|
)Ú max_workers)Útimeoutz%Health check failed, due to exception)Úexc_infoNz4Health check execution exceeds health_check_interval)r Úlenr' Úsubmitrd r r- Úresultr rY rU r rQ rT ÚloggerÚ exceptionÚoriginal_exceptionÚ TimeoutError) rD rN ÚexecutorrY rh ÚfuturesÚfutureÚeÚ unhealthy_dbs rE rO z%MultiDBClient._check_databases_health s ô
|
¬C°·±Ó,@Ô Að ÀXð $(§?¡?÷á Hað × 5Ñ 5°xÕ@ðGñ ð
|
Ü *Ø T×%@Ñ%@ô ò ;Fð ;Ø ñ ;÷ ð ùóøô 6ò
|
;Ø'(§z¡z Ü5<·\±\ ×,Ñ,Ô2ä ×(Ñ(Ø CØ%&×%9Ñ%9ð )ô ñ
|
$Ù $ Q×%9Ñ%9Ô :ÿøð
|
;ûô ò Ü"ØJóð ð ú÷1 ð úsY D%°&BÁD%ÁD Á5BÂD ÂD%Â D
|
 A DÄ D ÄD
|
Ä
|
D Ä D"Ä"D%Ä%D.rQ Ú old_stateÚ new_statec óü |t j k( r | j |j « y |t j k( r:|t j
|
k( r&| j j t t |« y y y rJ )
|
rU Ú HALF_OPENrd rY rV r rB Úrun_oncer Ú_half_open_circuit)rD rQ r¥ r¦ s rE rS z/MultiDBClient._on_circuit_state_change_callback" se ð ×)Ñ)Ò )Ø × !Ñ ! '×"2Ñ"2Ô 3Ø à Ò &¨9¼¿ ¹ Ò+DØ × Ñ × 'Ñ 'Ü$Ô&8¸'õ ð ,EÐ &rG c ó` | j j j j « y rJ )r= rW ÚclientÚcloser^ s rE r zMultiDBClient.close. s Ø × Ñ ×-Ñ-×4Ñ4×:Ñ:Õ<rG rJ )"Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r rF r[ r r_ r rj ro rm r ru Úfloatrx r r| r r r r r r r Úboolrd Ú ExceptionrO r rU rS r rK rG rE r r s ñð
|
' }ó ' òR! ðF yó ð
|
¨Lð
|
¸Tó
|
ð2 D \ó Dð AØ (ðAØCOóAð H¨ó HðD¨|ð DÀUó Dð&9°_ó 9ð 4¨Kó 4òGò ð S ¨*¨ °tÐ);Ñ <ó Sò &ð¨ ð ¸$ó ñ °¸)¸ÀdÐ9JÑ0Kó ðD
|
Ø%ð
|
Ø29ð
|
ØFMó
|
ó=rG r rQ c ó. t j | _ y rJ )rU r¨ rT )rQ s rE rª rª 2 s Ü×%Ñ%G
MrG c óx e Zd ZdZdefdZddZd Zd Zde fdZ
|
de fd Z dd Z dd Zdd Zd Zdee fdZy
|
)r zG
|
Pipeline implementation for multiple logical Redis databases.
|
r¬ c ó g | _ || _ y rJ )Ú_command_stackÚ_client)rD r¬ s rE rF zPipeline.__init__; s Ø ÔØ rG r\ c ó | S rJ rK r^ s rE Ú __enter__zPipeline.__enter__? ó Ø rG c ó$ | j « y rJ ©Úreset)rD Úexc_typeÚ exc_valueÚ tracebacks rE Ú__exit__zPipeline.__exit__B ó Ø
|
|
rG c óD | j « y # t $ r Y y w xY wrJ ©r¿ r´ r^ s rE Ú__del__zPipeline.__del__E s" ð Ø JJLøÜò Ù ð úó c ó, t | j « S rJ )r r¸ r^ s rE Ú__len__zPipeline.__len__K s Ü4×&Ñ&Ó'Ð'rG c ó y)z1Pipeline instances should always evaluate to TrueTrK r^ s rE Ú__bool__zPipeline.__bool__N s àrG Nc ó g | _ y rJ )r¸ r^ s rE r¿ zPipeline.resetR s
|
Ø ÕrG c ó$ | j « y)zClose the pipelineNr¾ r^ s rE r zPipeline.closeU s à
|
|
rG c ó@ | j j ||f« | S )ar
|
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.
|
)r¸ r{ r s rE Úpipeline_execute_commandz!Pipeline.pipeline_execute_commandY s! ð ×Ñ×"Ñ" D¨' ?Ô3Ø rG c ó& | j |i |¤S )zAdds a command to the stack)rÐ ©rD r r s rE r zPipeline.execute_commandh s à,t×,Ñ,¨dÐ=°fÑ=Ð=rG c ó | j j s| j j « | j j j t | j « « | j « S # | j « w xY w)z0Execute all the commands in the current pipeline)r¹ r> r[ r= Úexecute_pipelineÚtupler¸ r¿ r^ s rE r zPipeline.executel s_ à||×'Ò'Ø LL× #Ñ #Ô %ð Ø<<×0Ñ0×AÑAÜd×)Ñ)Ó*óð JJLøDJJLús ²7A: Á:B )r\ r ©r\ N)r® r¯ r° r± r rF r» rà rÇ ÚintrÊ r³ rÌ r¿ r rÐ r r r r rK rG rE r r 6 s^ ñð }ó óòòð (ó (ð$ó ó!óó ò >ð
|
cô
|
rG r c óÐ e Zd ZdZdefdZd dZd dZd dZd d Z e
|
de fd
|
« Z d Z d Zd Zd Zd Zd Zd Z d de defdZ d de defdZ d dede dee de ddf
|
dZy) r z2
|
PubSub object for multi database client.
|
r¬ c ó^ || _ | j j j di |¤ y)zýInitialize 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
|
NrK )r¹ r= r )rD r¬ r s rE rF zPubSub.__init__~ s( ð Ø, ×%Ñ%×,Ñ,Ñ6¨vÓ6rG r\ c ó | S rJ rK r^ s rE r» zPubSub.__enter__ r¼ rG Nc óD | j « y # t $ r Y y w xY wrJ rÆ r^ s rE rÇ zPubSub.__del__ s$ ð ð JJLøÜò Ù ð úrÈ c óL | j j j d« S )Nr¿ ©r¹ r= Úexecute_pubsub_methodr^ s rE r¿ z PubSub.reset s Ø||×,Ñ,×BÑBÀ7ÓKÐKrG c ó$ | j « y rJ r¾ r^ s rE r z PubSub.close rÄ rG c óV | j j j j S rJ )r¹ r= Ú active_pubsubÚ
|
subscribedr^ s rE râ zPubSub.subscribed s à||×,Ñ,×:Ñ:×EÑEÐErG c óP | j j j dg|¢ S )Nr rÝ ©rD r s rE r zPubSub.execute_command s, ØBt||×,Ñ,×BÑBØ ð
|
Ø $ò
|
ð
|
rG c óV | j j j dg|¢i |¤S )aE
|
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Ý rÒ s rE ræ zPubSub.psubscribe¤ ó7 ð Ct||×,Ñ,×BÑBØ ð
|
Ø ò
|
Ø#)ñ
|
ð
|
rG c óP | j j j dg|¢ S )zj
|
Unsubscribe from the supplied patterns. If empty, unsubscribe from
|
all patterns.
|
Ú punsubscriberÝ rä s rE ré zPubSub.punsubscribe° ó/ ð
|
Ct||×,Ñ,×BÑBØ ð
|
Ø !ò
|
ð
|
rG c óV | j j j dg|¢i |¤S )aR
|
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Ý rÒ s rE rì zPubSub.subscribe¹ s7 ð Ct||×,Ñ,×BÑBØ ð
|
Ø ò
|
Ø"(ñ
|
ð
|
rG c óP | j j j dg|¢ S )zi
|
Unsubscribe from the supplied channels. If empty, unsubscribe from
|
all channels
|
Ú unsubscriberÝ rä s rE rî zPubSub.unsubscribeÅ s( ð
|
Ct||×,Ñ,×BÑBÀ=ÐXÐSWÒXÐXrG c óV | j j j dg|¢i |¤S )az
|
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Ý rÒ s rE rð zPubSub.ssubscribeÌ rç rG c óP | j j j dg|¢ S )zu
|
Unsubscribe from the supplied shard_channels. If empty, unsubscribe from
|
all shard_channels
|
Ú sunsubscriberÝ rä s rE rò zPubSub.sunsubscribeØ rê rG Úignore_subscribe_messagesr c óR | j j j d||¬« S )a
|
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_message©ró r rÝ ©rD ró r s rE rõ zPubSub.get_messageá s0 ð ||×,Ñ,×BÑBØ Ø&?Øð Có
|
ð
|
rG c óR | j j j d||¬« S )a&
|
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_messagerö rÝ r÷ s rE rù zPubSub.get_sharded_messageñ s0 ð ||×,Ñ,×BÑBØ !Ø&?Øð Có
|
ð
|
rG Ú
|
sleep_timeÚdaemonÚexception_handlerÚsharded_pubsubr
|
c óV | j j j |||| |¬« S )N)rû rü r rý )r¹ r= Úexecute_pubsub_run)rD rú rû rü rý s rE Ú run_in_threadzPubSub.run_in_thread s6 ð ||×,Ñ,×?Ñ?Ø ØØ /ØØ)ð @ó
|
ð
|
rG )r\ r rÖ )Fç )r FNF)r® r¯ r° r± r rF r» rÇ r¿ r Úpropertyr³ râ r ræ ré rì rî rð rò r² rõ rù r r r rK rG rE r r y sê ñð 7}ó 7óóóLóð ðFDò Fó ðFò
|
ò
|
|
|
ò
|
ò
|
|
òYò
|
|
ò
|
ð ILñ
|
Ø)-ð
|
Ø@Eó
|
ð" ILñ
|
Ø)-ð
|
Ø@Eó
|
ð$ Ø Ø04Ø $ñ
|
àð
|
ð ð
|
ð $ HÑ-ð
|
ð
|
ð
|
ð
|
ô
|
rG r )0Úloggingr? Úconcurrent.futuresr Úconcurrent.futures.threadr Útypingr r r r Úredis.backgroundr Ú redis.clientr
|
Úredis.commandsr r Úredis.multidb.circuitr r rU Ú redis.multidb.command_executorr Úredis.multidb.configr r Úredis.multidb.databaser r r Úredis.multidb.exceptionr r Ú redis.multidb.failure_detectorr Úredis.multidb.healthcheckr r Ú redis.utilsr Ú getLoggerr® r r rª r r rK rG rE ú<module>r s ðÛ Û Ý +Ý 8ß 0Ó 0å 0Ý +ß <Ý 0Ý 2Ý Aß Dß DÑ Dß XÝ :ß DÝ $à × Ñ 8Ó $ð ôW=Ð'¨ ó W=ó ðW=ðt& ó &ô@Ð" Lô @÷FU
|
ò U
|
rG
|
