AHj ;ddlZddlZddlmZmZddlmZddlmZddl m Z m Z m Z ddl mZmZmZmZddlmZGdd eZGd d eZGd d e ZGddee ZGddZGdde ZGddeZy)N)OptionalUnion)Redis)SentinelCommands) ConnectionConnectionPool SSLConnection)ConnectionError ReadOnlyError ResponseError TimeoutError)SENTINELc eZdZy)MasterNotFoundErrorN__name__ __module__ __qualname__F/root/tools/cai/cai_env/lib/python3.12/site-packages/redis/sentinel.pyrrrrc eZdZy)SlaveNotFoundErrorNrrrrrrrrrc reZdZfdZdZdZdZdZ d edddde e e fd e e d e e ffd ZxZS) SentinelManagedConnectionc P|jd|_t| di|y)Nconnection_poolr)poprsuper__init__)selfkwargs __class__s rr!z"SentinelManagedConnection.__init__s%%zz*;< "6"rc|j}dt|jdt|jd|jd}|j r!d|j d|j }||z}|S)N<. (service=z%s)>z,host=z,port=)rtyperr service_namehostport)r"pools host_infos r__repr__z"SentinelManagedConnection.__repr__s{##T %%&aT (;(;'<))*$ 0  99  6$))=II Arcp|\|_|_|j|jjdy)NF) check_healthretry_socket_connect)r+r,connect_check_healthrcheck_connection)r"addresss r connect_toz$SentinelManagedConnection.connect_to)s5& 49 !!-->>!& " rc*|jry|jjr*|j|jj y|jj D]} |j|cSt#t $rY)wxYwN)_sockr is_masterr7get_master_address rotate_slavesr r)r"slaves r_connect_retryz(SentinelManagedConnection._connect_retry1s ::     ) ) OOD00CCE F--;;= ??511  % $'s,B BBcP|jj|jdS)Ncyr9r)errors rz3SentinelManagedConnection.connect..?sr)retrycall_with_retryr?r"s rconnectz!SentinelManagedConnection.connect>s zz))$*=*=?QRRrF)timeoutdisconnect_on_error push_requestrHrIrJc t|||||S#t$r3|jjr|j t dwxYw)N)disable_decodingrHrIrJz"The previous master is now a slave)r read_responser rr; disconnectr )r"rLrHrIrJr$s rrMz'SentinelManagedConnection.read_responseAsf 7(!1$7) )   ##-- !%&JKK  s >t?P?PQ >>d11^C"0D #668O***U*Crc#K|jj|j}|r|j't j dt |dz |_tt |D]5}|jdzt |z|_||j}|7 |jtd|j#t$rY#wxYww)NrzNo slave found for ) r\discover_slavesr*r`randomrandintlenranger<rr)r"slaves_r>s rr=z)SentinelConnectionPoolProxy.rotate_slavess&&66t7H7HI $$,(.q#f+/(J%3v;' )-)>)>)Bc&k(Q%t445    ))+ +!#6t7H7H6K!LMM#   s*B'C#*C<C# C C#C  C#N)rrrr!r]r<r=rrrrWrW`s % NrrWcXeZdZdZfdZdZfdZedZfdZ dZ dZ xZ S) SentinelConnectionPoolz Sentinel backed connection pool. If ``check_connection`` flag is set to True, SentinelManagedConnection sends a PING command right after establishing the connection. c |jd|jddrtnt|d<|jdd|_|jdd|_t ||j|j |||_t|$d i||j|jd<||_ ||_ y) Nconnection_classsslFr;Tr5)rr;r5r*r\rr) getrrUrr;r5rWproxyr r!connection_kwargsr*r\)r"r*r\r#r$s rr!zSentinelConnectionPool.__init__s%+ZZ ::eU+-. & !" K6 & +=u E0 nn!22%-    "6"48JJ01( 0rc |jrdnd}dt|jdt|jd|jd|d S)Nmasterr>r&r'r((z))>)r;r)rrr*)r"roles rr0zSentinelConnectionPool.__repr__sR>>xwT %%&aT (;(;'<))*!D6 6 rcVt||jjyr9)r r]rs)r"r$s rr]zSentinelConnectionPool.resets   rc.|jjSr9)rsr_rFs rr_z%SentinelConnectionPool.master_addressszz(((rc|j xs3|jxr%|j|j|jfk(}t }|xr|j |Sr9)r;r_r+r,r owns_connection)r" connectioncheckparentr$s rr|z&SentinelConnectionPool.owns_connectionsZNN" NN Xt22z 6XX ;// ;;rc6|jjSr9)rsr<rFs rr<z)SentinelConnectionPool.get_master_addressszz,,..rc6|jjS)zRound-robin slave balancer)rsr=rFs rr=z$SentinelConnectionPool.rotate_slavesszz''))r) rrr__doc__r!r0r]propertyr_r|r<r=rRrSs@rrnrns:1. ))</*rrncZeZdZdZ d dZdZdZdZdZdZ d Z e e fd Z e e fd Zy) Sentinelar Redis Sentinel cluster client >>> from redis.sentinel import Sentinel >>> sentinel = Sentinel([('localhost', 26379)], socket_timeout=0.1) >>> master = sentinel.master_for('mymaster', socket_timeout=0.1) >>> master.set('foo', 'bar') >>> slave = sentinel.slave_for('mymaster', socket_timeout=0.1) >>> slave.get('foo') b'bar' ``sentinels`` is a list of sentinel nodes. Each node is represented by a pair (hostname, port). ``min_other_sentinels`` defined a minimum number of peers for a sentinel. When querying a sentinel, if it doesn't meet this threshold, responses from that sentinel won't be considered valid. ``sentinel_kwargs`` is a dictionary of connection arguments used when connecting to sentinel instances. Any argument that can be passed to a normal Redis connection can be specified here. If ``sentinel_kwargs`` is not specified, any socket_timeout and socket_keepalive options specified in ``connection_kwargs`` will be used. ``connection_kwargs`` are keyword arguments that will be used when establishing a connection to a Redis server. Nc |5|jDcic]\}}|jds||}}}||_|D cgc]\}} t|| fi|jc} }|_||_||_||_ycc}}wcc} }w)Nsocket_)items startswithsentinel_kwargsr sentinelsmin_other_sentinelsrt_force_master_ip) r"rrrforce_master_iprtkvhostnamer,s rr!zSentinel.__init__s  "!2!8!8!:Aall9>U1O /#, $ (D 9D$8$8 9 $7 !2 /  sBB!B clt|jdd}t|jdd}|r:tj|jj |i|}|r|gS|rdSdSg}|jD]#}|j |j |i|%|r|St|S)z Execute Sentinel command in sentinel nodes. once - If set to True, then execute the resulting command on a single node at random, rather than across the entire sentinel cluster. onceFreturn_responsesT)rQrrgchoicerexecute_commandappendall)r"argsr#rrresponse responsessentinels rrzSentinel.execute_commands FJJvu-.  +=u EF Dv}}T^^4DDdUfUH z!'t2U2  HH   5X55tFvF G H  9~rcg}|jD]6}|jdj|jj8dt |j dt |jddj|dS)Nz {host}:{port}r&r'z (sentinels=[,z])>) rr format_maprrtr)rrjoin)r"sentinel_addressesrs rr0zSentinel.__repr__s H  % %**8+C+C+U+UV   T %%&aT (;(;'<388$678 = rcJ|dr |ds|dry|d|jkryy)Nr;is_sdownis_odownFznum-other-sentinelsT)r)r"stater*s rcheck_master_statezSentinel.check_master_state)s5[!U:%6% :K & '$*B*B Brc.t}t|jD]\}} |j}|j|}|s+|j||s>||jdc|jd<|j|<|j |jn|d}||dfcSd} t|dkDrddj|} td || #tt f$r }|j |d|Yd}~d}~wwxYw) z Asks sentinel servers for the Redis master's address corresponding to the service labeled ``service_name``. Returns a pair (address, port) or raises MasterNotFoundError if no master is found. z - Nripr,z : z, zNo master found for ) list enumeratersentinel_mastersr r rrrrrrirr) r"r*collected_errors sentinel_normasterserr error_infos rrczSentinel.discover_master1s7 6%.t~~%> ) !K "335KK -E00 ENN1%?q!4>>+#>,,8))t 5=((' )*  1 $tyy)9:; sC%%D4DDc`g}|D]&}|ds|dr|j|d|df(|S)z1Remove slaves that are in an ODOWN or SDOWN staterrrr,)r)r"rk slaves_aliver>s r filter_slaveszSentinel.filter_slavesTsK  >EZ E*$5   teFm < = >rc|jD]+} |j|}|j |}|s)|cSgS#tttf$rYFwxYw)z;Returns a list of alive slaves for service ``service_name``)rsentinel_slavesr r r r)r"r*rrks rrfzSentinel.discover_slaves]sf H !11,?''/F   $]LA  s=AAc d|d<t|j}|j||j|||fi|S)a Returns a redis client instance for the ``service_name`` master. Sentinel client will detect failover and reconnect Redis clients automatically. A :py:class:`~redis.sentinel.SentinelConnectionPool` class is used to retrieve the master's address before establishing a new connection. NOTE: If the master's address has changed, any cached connections to the old master are closed. By default clients will be a :py:class:`~redis.Redis` instance. Specify a different class to the ``redis_class`` argument if you desire something different. The ``connection_pool_class`` specifies the connection pool to use. The :py:class:`~redis.sentinel.SentinelConnectionPool` will be used by default. All other keyword arguments are merged with any connection_kwargs passed to this class and passed to the connection pool as keyword arguments to be used to initialize Redis connections. Tr;dictrtupdate from_poolr"r* redis_classconnection_pool_classr#rts r master_forzSentinel.master_forisP>#{ !7!78  ($$ !, J8I J  rc d|d<t|j}|j||j|||fi|S)a Returns redis client instance for the ``service_name`` slave(s). A SentinelConnectionPool class is used to retrieve the slave's address before establishing a new connection. By default clients will be a :py:class:`~redis.Redis` instance. Specify a different class to the ``redis_class`` argument if you desire something different. The ``connection_pool_class`` specifies the connection pool to use. The SentinelConnectionPool will be used by default. All other keyword arguments are merged with any connection_kwargs passed to this class and passed to the connection pool as keyword arguments to be used to initialize Redis connections. Fr;rrs r slave_forzSentinel.slave_forsP0${ !7!78  ($$ !, J8I J  r)rNN)rrrrr!rr0rrcrrfrrnrrrrrrrsR> 008  !WF 4 $ R4  rr)rgrYtypingrr redis.clientrredis.commandsrredis.connectionrrr redis.exceptionsr r r r redis.utilsrrrrrUrWrnrrrrrs "+FF ! /   @ @F #