o ݭh0@sddlZddlZddlmZmZmZmZmZmZddl m Z ddl m Z m Z mZmZddlmZmZmZmZddlmZGdddeZGd d d eZGd d d eZGd dde ZGdddZdS)N) AsyncIteratorIterableMappingSequenceTupleType)Redis) ConnectionConnectionPool EncodableT SSLConnection)ConnectionError ReadOnlyError ResponseError TimeoutError) str_if_bytesc@ eZdZdS)MasterNotFoundErrorN__name__ __module__ __qualname__rrV/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/aioredis/sentinel.pyrrc@r)SlaveNotFoundErrorNrrrrrrrrcsDeZdZfddZddZfddZddZfd d ZZS) SentinelManagedConnectionc sH|d|_|ddstt|jdi|dStjdi|dS)Nconnection_poolsslFr)poprsuperr __init__)selfkwargs __class__rrr!s  z"SentinelManagedConnection.__init__cCsD|j}|jjd|j}|jrd|jd|j}||7}|dS)N )rr%r service_namehostport)r"pools host_inforrr__repr__"s z"SentinelManagedConnection.__repr__cs\|\|_|_tIdH|jjr*|dIdHt|IdHdkr,t ddSdS)NPINGPONGz PING failed) r)r*r connectrcheck_connection send_commandr read_responser )r"addressr$rr connect_to*s z$SentinelManagedConnection.connect_toc sz|jrdS|jjr||jIdHIdHdS|j2z3dHW}z ||IdHWSty9Yqw6tN)_readerr is_masterr6get_master_address rotate_slavesr r)r"slaverrrr12s  z!SentinelManagedConnection.connectcsDz tIdHWSty!|jjr |IdHtdw)Nz"The previous master is now a slave)r r4rrr9 disconnectr r"r$rrr4?s z'SentinelManagedConnection.read_response) rrrr!r.r6r1r4 __classcell__rrr$rrs    rcs\eZdZdZfddZddZfddZdeffd d Zd d Z d e fddZ 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 sj|dt|d<|dd|_|dd|_tjdi|t||j d<||_ ||_ d|_ d|_ dS)Nconnection_classr9Tr2Frr)getrrr9r2r r!weakrefproxyconnection_kwargsr(sentinel_managermaster_addressslave_rr_counter)r"r(rFr#r$rrr!Vs zSentinelConnectionPool.__init__cCs&|jjd|jd|jrdpddS)Nr&(masterr<z)>)r%rr(r9r>rrrr.cs  zSentinelConnectionPool.__repr__cstd|_d|_dSr7)r resetrGrHr>r$rrrKis  zSentinelConnectionPool.reset connectioncs0|j p|jo|j|j|jfk}|ot|Sr7)r9rGr)r*r owns_connection)r"rLcheckr$rrrMnsz&SentinelConnectionPool.owns_connectioncsB|j|jIdH}|jr|j|kr||_|jddIdH|S)NF)inuse_connections)rFdiscover_masterr(r9rGr=)r"rGrrrr:ts z)SentinelConnectionPool.get_master_addressreturncCs|j|jIdH}|r8|jdurtdt|d|_tt|D]}|jdt||_||j}|Vq#z |IdHVWn t yKYnwt d|j)zRound-robin slave balancerNrzNo slave found for ) rFdiscover_slavesr(rHrandomrandintlenranger:rr)r"slaves_r<rrrr;~s   z$SentinelConnectionPool.rotate_slaves) rrr__doc__r!r.rKr rMr:rr;r?rrr$rr@Ns   r@c@seZdZdZ  dddZddZded ed efd d Z d efd dZ de e d e eeeffddZd ed e eeeffddZeefd edeedeefddZeefd edeedeefddZdS)Sentinela Redis Sentinel cluster client >>> from aioredis.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|dur dd|D}|_fdd|D_|_|_dS)NcSs i|] \}}|dr||qS)socket_) startswith).0kvrrr s z%Sentinel.__init__..cs&g|]\}}td||djqS))r)r*r)rsentinel_kwargs)r^hostnamer*r>rr sz%Sentinel.__init__..)itemsrb sentinelsmin_other_sentinelsrE)r"rfrgrbrErr>rr!s   zSentinel.__init__cCsNg}|jD]}||jjdd|jjdq|jjdd|dS)Nr):r*z )rfappendrrEr%rjoin)r"sentinel_addressessentinelrrrr.s  zSentinel.__repr__stater(rQcCs2|dr |ds |drdS|d|jkrdSdS)Nr9is_sdownis_odownFznum-other-sentinelsT)rg)r"rnr(rrrcheck_master_states zSentinel.check_master_statec st|jD]>\}}z |IdH}Wn ttfyYqw||}|rD|||rD||jd|jd<|j|<|d|dfSqtd|)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. Nripr*zNo master found for ) enumeraterfsentinel_mastersr rrBrqr)r"r( sentinel_normmastersrnrrrrPs zSentinel.discover_masterrXcCs:g}|D]}|ds|drq||d|dfq|S)z1Remove slaves that are in an ODOWN or SDOWN staterprorrr*)rj)r"rX slaves_aliver<rrr filter_slavess zSentinel.filter_slavesc sV|jD]$}z ||IdH}Wn tttfyYqw||}|r(|SqgS)z;Returns a list of alive slaves for service ``service_name``N)rfsentinel_slavesr rrrx)r"r(rmrXrrrrSs  zSentinel.discover_slaves redis_classconnection_pool_classcK4d|d<t|j}||||||fi|dS)a Returns a redis client instance for the ``service_name`` master. A :py:class:`~redis.sentinel.SentinelConnectionPool` class is used to retrive 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. Tr9rdictrEupdater"r(rzr{r#rErrr master_fors  zSentinel.master_forcKr|)a Returns redis client instance for the ``service_name`` slave(s). A SentinelConnectionPool class is used to retrive 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. Fr9r}r~rrrr slave_for*s  zSentinel.slave_for)rN)rrrrZr!r.rstrboolrqrPrrrrr rxrSrr@rrrrrrrr[sH      )r[)rTrCtypingrrrrrraioredis.clientraioredis.connectionr r r r aioredis.exceptionsr rrraioredis.utilsrrrrr@r[rrrrs   6B