o LjJ@sddlZddlZddlmZmZddlmZddlmZddl m Z m Z m Z ddl mZmZmZmZddlmZGdd d eZGd d d eZeZGd d d e ZGdddee ZGdddZGddde ZGdddeZdS)N)OptionalUnion)Redis)SentinelCommands) ConnectionConnectionPool SSLConnection)ConnectionError ReadOnlyError ResponseError TimeoutError)SENTINELc@ eZdZdS)MasterNotFoundErrorN__name__ __module__ __qualname__rrL/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/redis/sentinel.pyrrc@r)SlaveNotFoundErrorNrrrrrrrrc steZdZfddZddZddZddZd d Z ded d d d e e e fde e de e ffddZZS)SentinelManagedConnectionc s"|d|_tjdi|dS)Nconnection_poolr)poprsuper__init__)selfkwargs __class__rrrs z"SentinelManagedConnection.__init__cCsR|j}dt|jdt|jd|jd}|jr'd|jd|j}||}|S)N<. (service=z%s)>z,host=z,port=)rtyperr service_namehostport)rpools host_inforrr__repr__!sz"SentinelManagedConnection.__repr__cCs"|\|_|_|j|jjdddS)NF) check_healthretry_socket_connect)r&r'connect_check_healthrcheck_connection)raddressrrr connect_to,s  z$SentinelManagedConnection.connect_toc Cs\|jrdS|jjr||jdS|jD]}z||WSty+YqwtN)_sockr is_masterr1get_master_address rotate_slavesr r)rslaverrr_connect_retry4s z(SentinelManagedConnection._connect_retrycCs|j|jddS)NcSsdSr2r)errorrrrBsz3SentinelManagedConnection.connect..)retrycall_with_retryr8rrrrconnectAsz!SentinelManagedConnection.connectF)timeoutdisconnect_on_error push_requestr?r@rAcs@z tj||||dWSty|jjr|tdw)N)disable_decodingr?r@rAz"The previous master is now a slave)r read_responser rr4 disconnectr )rrBr?r@rArrrrCDs z'SentinelManagedConnection.read_response)F)rrrrr+r1r8r>r rfloatobjectrboolrC __classcell__rrrrrs$    rc@r)SentinelManagedSSLConnectionNrrrrrrI_rrIc@s4eZdZddZddZddZddZd d Zd S) SentinelConnectionPoolProxycCs0t||_||_||_||_||_|dSr2)weakrefrefconnection_pool_refr4r/r%sentinel_managerreset)rrr4r/r%rNrrrrds  z$SentinelConnectionPoolProxy.__init__cCsd|_d|_dSr2)master_addressslave_rr_counterr=rrrrOss z!SentinelConnectionPoolProxy.resetcCsD|j|j}|jr |j|kr ||_|}|dur |jdd|S)NF)inuse_connections)rNdiscover_masterr%r4rPrMrD)rrPrrrrr5ws z.SentinelConnectionPoolProxy.get_master_addressccs|j|j}|r5|jdurtdt|d|_tt|D]}|jdt||_||j}|Vq z|VWn t yEYnwt d|j)NrzNo slave found for ) rNdiscover_slavesr%rQrandomrandintlenranger5rr)rslaves_r7rrrr6s   z)SentinelConnectionPoolProxy.rotate_slavescC|SzRound-robin replica balancer. This is an alias for :py:meth:`rotate_slaves`, using the preferred Redis 5.0+ terminology. r6r=rrrrotate_replicasz+SentinelConnectionPoolProxy.rotate_replicasN)rrrrrOr5r6r_rrrrrJcs  rJcsdeZdZdZfddZddZfddZedd Zfd d Z d d Z ddZ 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 s|d|ddr tnt|d<|dd|_|dd|_t||j|j||d|_tj d i||j|j d<||_ ||_ dS) Nconnection_classsslFr4Tr/)rr4r/r%rNrr) getrrIrr4r/rJproxyrrconnection_kwargsr%rN)rr%rNrrrrrs&   zSentinelConnectionPool.__init__c Cs:|jrdnd}dt|jdt|jd|jd|d S)Nmasterr7r!r"r#(z))>)r4r$rrr%)rrolerrrr+szSentinelConnectionPool.__repr__cst|jdSr2)rrOrer=rrrrOs zSentinelConnectionPool.resetcCs|jjSr2)rerPr=rrrrPsz%SentinelConnectionPool.master_addresscs4|j p|jo|j|j|jfk}t}|o||Sr2)r4rPr&r'rowns_connection)r connectioncheckparentrrrrjs z&SentinelConnectionPool.owns_connectioncC |jSr2)rer5r=rrrr5s z)SentinelConnectionPool.get_master_addresscCrn)zRound-robin slave balancer)rer6r=rrrr6s z$SentinelConnectionPool.rotate_slavescCr\r]r^r=rrrr_r`z&SentinelConnectionPool.rotate_replicas)rrr__doc__rr+rOpropertyrPrjr5r6r_rHrrrrras    rac@seZdZdZ   d#ddZddZdd Zd$d d Zd dZddZ ddZ ddZ ddZ ddZ ddZddZeefddZeefdd Zeefd!d"ZdS)%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. rNc sJ|dur dd|D}|_fdd|D_|_|_|_dS)NcSs i|] \}}|dr||qS)socket_) startswith).0kvrrr s z%Sentinel.__init__..cs$g|]\}}t||fijqSr)rsentinel_kwargs)rthostnamer'r=rr sz%Sentinel.__init__..)itemsrx sentinelsmin_other_sentinelsrf_force_master_ip)rr|r}rxforce_master_iprfrr=rrs   zSentinel.__init__cOst|dd}t|dd}|r)t|jj|i|}|r#|gS|r'dSdSg}|jD] }||j|i|q.|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)rGrrVchoicer|execute_commandappendall)rargsrrrresponse responsessentinelrrrrs  zSentinel.execute_commandcCsNg}|jD] }|d|jjqdt|jdt|jdd|dS)Nz {host}:{port}r!r"z (sentinels=[,z])>) r|r format_maprrfr$rrjoin)rsentinel_addressesrrrrr+1s  zSentinel.__repr__returnc CsXd}|jD] }z|Wqty%}z |dur|}WYd}~qd}~ww|r*|dS)au Close all sentinel clients created by this Sentinel and their connection pools. Each client is closed independently: if one raises, the remaining clients are still closed and the first error is re-raised afterwards. Clients returned by ``master_for``/``slave_for`` are owned by the caller and are not closed here. N)r|close Exception)rexcrerrrr<s  zSentinel.closecCs|Sr2rr=rrr __enter__QszSentinel.__enter__cCs |dSr2)r)rexc_type exc_value tracebackrrr__exit__Ts zSentinel.__exit__cCs2|dr |ds |drdS|d|jkrdSdS)Nr4is_sdownis_odownFznum-other-sentinelsT)r})rstater%rrrcheck_master_stateWs zSentinel.check_master_statec Cst}t|jD]X\}}z|}Wnttfy0}z||d|WYd}~qd}~ww||}|r`|||r`||jd|jd<|j|<|j durT|j n|d}||dfSqd} t |dkrqdd |} t d || ) 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 enumerater|sentinel_mastersr r rrdrr~rXrr) rr%collected_errors sentinel_normastersrrr error_inforrrrS_s0     zSentinel.discover_mastercCs:g}|D]}|ds|drq||d|dfq|S)z1Remove slaves that are in an ODOWN or SDOWN staterrrr')r)rrZ slaves_aliver7rrr filter_slavess zSentinel.filter_slavescC ||S)zRemove replicas that are in an ODOWN or SDOWN state. This is an alias for :py:meth:`filter_slaves`, using the preferred Redis 5.0+ terminology. )r)rreplicasrrrfilter_replicas zSentinel.filter_replicasc CsN|jD]!}z||}Wn tttfyYqw||}|r$|SqgS)z;Returns a list of alive slaves for service ``service_name``)r|sentinel_slavesr r r r)rr%rrZrrrrUs  zSentinel.discover_slavescCr)zReturns a list of alive replicas for service ``service_name``. This is an alias for :py:meth:`discover_slaves`, using the preferred Redis 5.0+ terminology. )rU)rr%rrrdiscover_replicasrzSentinel.discover_replicascK4d|d<t|j}||||||fi|S)a9 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. HIMPORT note: the returned client exposes the HIMPORT command family (``himport_prepare``, ``himport_set``, ``himport_discard``, ``himport_discard_all``). Fieldsets are declared at runtime with ``himport_prepare`` on the client and live on that client's shared registry, so they survive Sentinel failover automatically: the pool re-points to the new master and the fieldset is re-prepared lazily on the next ``himport_set``. Each call to the current method returns a *new* client with its *own* empty registry, so call ``himport_prepare`` on the long-lived client you reuse rather than creating a fresh one per operation; otherwise ``himport_set`` fails with ``no such fieldset``. (``himport_set`` is a write and is served by the master.) Tr4dictrfupdate from_poolrr% redis_classconnection_pool_classrrfrrr master_fors +  zSentinel.master_forcKr)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. HIMPORT note: the returned client exposes the HIMPORT command family (``himport_prepare``, ``himport_set``, ``himport_discard``, ``himport_discard_all``). Fieldsets are declared at runtime with ``himport_prepare`` on the client and live on that client's shared registry, so they survive Sentinel failover automatically: the pool re-points to the new master and the fieldset is re-prepared lazily on the next ``himport_set``. Each call to current method returns a *new* client with its *own* empty registry, so call ``himport_prepare`` on the long-lived client you reuse rather than creating a fresh one per operation; otherwise ``himport_set`` fails with ``no such fieldset``. (``himport_set`` is a write and is served by the master.) Fr4rrrrr slave_fors $  zSentinel.slave_forcKs|j|f||d|S)z Returns redis client instance for the ``service_name`` replica(s). This is an alias for :py:meth:`slave_for`, using the preferred Redis 5.0+ terminology. )rr)r)rr%rrrrrr replica_fors zSentinel.replica_for)rNN)rN)rrrrorrr+rrrrrSrrrUrrrarrrrrrrrqs4  #    5 .rq)rVrKtypingrr redis.clientrredis.commandsrredis.connectionrrrredis.exceptionsr r r r redis.utilsr rrReplicaNotFoundErrorrrIrJrarqrrrrs    C8E