U g7@sddlZddlZddlZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZmZmZmZddlmZddlmZmZmZmZddlmZGdd d eZGd d d eZGd d d eZGdddeeZGdddeZGdddeZ dS)N) AsyncIteratorIterableMappingOptionalSequenceTupleType)Redis) ConnectionConnectionPool EncodableT SSLConnection)AsyncSentinelCommands)ConnectionError ReadOnlyError ResponseError TimeoutError) str_if_bytesc@s eZdZdS)MasterNotFoundErrorN__name__ __module__ __qualname__rr:/tmp/pip-unpacked-wheel-f3sx1i9r/redis/asyncio/sentinel.pyrsrc@s eZdZdS)SlaveNotFoundErrorNrrrrrrsrcsneZdZfddZddZfddZddZd d Zdd d dee e e e e edfddZ Z S)SentinelManagedConnectionc s|d|_tjf|dS)Nconnection_pool)poprsuper__init__)selfkwargs __class__rrr s z"SentinelManagedConnection.__init__cCsBd|jjd|jj}|jr:d|jd|j}||7}|dS)N<.z,host=z,port=z)>)r$rrhostport)r!sZ host_inforrr__repr__s z"SentinelManagedConnection.__repr__csV|\|_|_tIdH|jjrR|dIdHt|IdHdkrRt ddS)NZPINGZPONGz PING failed) r'r(rconnectrcheck_connectionZ send_commandr read_responser)r!addressr#rr connect_to&s  z$SentinelManagedConnection.connect_toc s|jr dS|jjr0||jIdHIdHnN|j2z<3dHW}z||IdHWStk rtYq:Yq:Xq:6tdSN)Z_readerr is_masterr/get_master_address rotate_slavesrr)r!slaverrr_connect_retry.sz(SentinelManagedConnection._connect_retrycs|j|jddIdHS)NcSs tdS)Nr)asynciosleep)errorrrr>z3SentinelManagedConnection.connect..)retryZcall_with_retryr5r!rrrr+;sz!SentinelManagedConnection.connectFNT)disconnect_on_error push_requestdisable_decodingtimeoutr=r>csVztj||||dIdHWStk rP|jjrJ|IdHtdYnXdS)Nr?z"The previous master is now a slave)rr-rrr1 disconnectr)r!r@rAr=r>r#rrr-Asz'SentinelManagedConnection.read_response)FN) rrrr r*r/r5r+boolrfloatr- __classcell__rrr#rrs    rc@s eZdZdS)SentinelManagedSSLConnectionNrrrrrrF\srFcs\eZdZdZfddZddZfddZedfd d Zd d Z e d ddZ Z 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 sv|d|ddrtnt|d<|dd|_|dd|_tjf|t ||j d<||_ ||_ d|_ d|_dS)NZconnection_classsslFr1Tr,r)getrrFrr1r,rr weakrefproxyconnection_kwargs service_namesentinel_managermaster_addressslave_rr_counter)r!rMrNr"r#rrr hs zSentinelConnectionPool.__init__c Cs2d|jjd|jjd|jd|jr(dp*dd S)Nr%r&z (service=(Zmasterr4z))>)r$rrrMr1r<rrrr*zs0zSentinelConnectionPool.__repr__cstd|_d|_dSr0)rresetrOrPr<r#rrrRs zSentinelConnectionPool.reset) connectioncs0|j p|jo|j|j|jfk}|o.t|Sr0)r1rOr'r(rowns_connection)r!rScheckr#rrrTsz&SentinelConnectionPool.owns_connectioncs@|j|jIdH}|jr<|j|kr<||_|jddIdH|S)NF)Zinuse_connections)rNdiscover_masterrMr1rOrB)r!rOrrrr2s  z)SentinelConnectionPool.get_master_address)returncCs|j|jIdH}|rn|jdkr8tdt|d|_tt|D](}|jdt||_||j}|VqDz|IdHVWnt k rYnXt d|jdS)zRound-robin slave balancerNrzNo slave found for ) rNdiscover_slavesrMrPrandomrandintlenranger2rr)r!slaves_r4rrrr3s  z$SentinelConnectionPool.rotate_slaves) rrr__doc__r r*rRr rTr2rr3rErrr#rrG`s   rGc@seZdZdZdddZddZdd Zeee d d d Z ed ddZ e e eeeefdddZeeeeefdddZeefeeeeedddZeefeeeeedddZdS)Sentinela~ 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) >>> await master.set('foo', 'bar') >>> slave = sentinel.slave_for('mymaster', socket_timeout=0.1) >>> await 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. rNc sD|dkrdd|D}|_fdd|D_|_|_dS)NcSs i|]\}}|dr||qS)Zsocket_) startswith).0kvrrr s z%Sentinel.__init__..cs&g|]\}}tf||djqS))r'r()r sentinel_kwargs)rchostnamer(r<rr sz%Sentinel.__init__..)itemsrg sentinelsmin_other_sentinelsrL)r!rkrlrgrLrr<rr s  zSentinel.__init__csptdd}dkr&d|rFt|jjIdHn&fdd|jD}tj |IdHdS)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. onceFNcsg|]}t|jqSr)r6ZTaskexecute_command)rcsentinelargsr"rrrisz,Sentinel.execute_command..T) rCrIkeysrrZchoicerkrnr6Zgather)r!rqr"rmZtasksrrprrns   zSentinel.execute_commandcCsXg}|jD](}||jjdd|jjdq d|jd|jjdd|dS) Nr':r(r%r&z (sentinels=[,z])>)rkappendrrLr$rjoin)r!Zsentinel_addressesrorrrr*s "zSentinel.__repr__)staterMrWcCs2|dr|ds|drdS|d|jkr.dSdS)Nr1is_sdownis_odownFznum-other-sentinelsT)rl)r!rxrMrrrcheck_master_states zSentinel.check_master_state)rMc st}t|jD]\}}z|IdH}WnBttfk rl}z ||d|WYqW5d}~XYnX||}|r|||r||jd|jd<|j|<|d|dfSqd}t |dkrdd |}t d ||dS) 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. Nz - ripr(z : z, zNo master found for ) list enumeraterkZsentinel_mastersrrrvrIr{r\rwr) r!rMZcollected_errorsZ sentinel_noroZmasterserx error_inforrrrVs"  zSentinel.discover_master)r^rWcCs:g}|D],}|ds|drq||d|dfq|S)z1Remove slaves that are in an ODOWN or SDOWN staterzryr|r()rv)r!r^Z slaves_aliver4rrr filter_slavess zSentinel.filter_slaves)rMrWc sZ|jD]N}z||IdH}Wntttfk r<YqYnX||}|r|SqgS)z;Returns a list of alive slaves for service ``service_name``N)rkZsentinel_slavesrrrr)r!rMror^rrrrY*s    zSentinel.discover_slaves)rM redis_classconnection_pool_classcKs4d|d<t|j}|||||f|}||S)a Returns a redis client instance for the ``service_name`` master. 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. Tr1dictrLupdateZ from_poolr!rMrrr"rLrrrr master_for8s   zSentinel.master_forcKs4d|d<t|j}|||||f|}||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. Fr1rrrrr slave_for]s   zSentinel.slave_for)rN)rrrr`r rnr*rstrrCr{rVrrrrr rrYr rGrrrrrrrras8     (ra)!r6rZrJtypingrrrrrrrZredis.asyncio.clientr Zredis.asyncio.connectionr r r r Zredis.commandsrZredis.exceptionsrrrrZ redis.utilsrrrrrFrGrarrrrs$   BG