o i:@sddlZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZGddde ZGd d d e ZGd d d e ZGd ddee ZGdddZGddde ZGdddeZdS)N)Optional)Redis)SentinelCommands) ConnectionConnectionPool SSLConnection)ConnectionError ReadOnlyError ResponseError TimeoutErrorc@ eZdZdS)MasterNotFoundErrorN__name__ __module__ __qualname__rrE/home/app/Keep/.python/lib/python3.10/site-packages/redis/sentinel.pyr r c@r )SlaveNotFoundErrorNrrrrrrrrcsfeZdZfddZddZddZddZd d Z dd d d d ee dee ffddZ Z S)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)rpoolsZ host_inforrr__repr__sz"SentinelManagedConnection.__repr__cCs"|\|_|_|j|jjdddS)NF)Z check_healthZretry_socket_connect)r$r%Zconnect_check_healthrcheck_connection)raddressrrr connect_to(s  z$SentinelManagedConnection.connect_toc Cs\|jrdS|jjr||jdS|jD]}z||WSty+YqwtN)_sockr is_masterr+get_master_address rotate_slavesrr)rslaverrr_connect_retry0s z(SentinelManagedConnection._connect_retrycCs|j|jddS)NcSsdSr,r)errorrrr>sz3SentinelManagedConnection.connect..)retryZcall_with_retryr2rrrrconnect=sz!SentinelManagedConnection.connectF)disconnect_on_error push_requestr8r9cs>z tj|||dWSty|jjr|tdw)N)disable_decodingr8r9z"The previous master is now a slave)r read_responser rr. disconnectr)rr:r8r9rrrr;@s z'SentinelManagedConnection.read_response)F) rrrrr(r+r2r7rboolr; __classcell__rrrrrs   rc@r )SentinelManagedSSLConnectionNrrrrrr?Yrr?c@s,eZdZddZddZddZddZd S) SentinelConnectionPoolProxycCs0t||_||_||_||_||_|dSr,)weakrefrefconnection_pool_refr.r)r#sentinel_managerreset)rrr.r)r#rDrrrr^s  z$SentinelConnectionPoolProxy.__init__cCsd|_d|_dSr,)master_addressslave_rr_counterr6rrrrEms z!SentinelConnectionPoolProxy.resetcCsD|j|j}|jr |j|kr ||_|}|dur |jdd|S)NF)Zinuse_connections)rDdiscover_masterr#r.rFrCr<)rrFrrrrr/qs 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 ) rDdiscover_slavesr#rGrandomrandintlenranger/r r)rslaves_r1rrrr0|s   z)SentinelConnectionPoolProxy.rotate_slavesN)rrrrrEr/r0rrrrr@]s  r@cs\eZdZdZfddZddZfddZedd Zfd d Z d d Z 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) NZconnection_classsslFr.Tr))rr.r)r#rDrr) getrr?rr.r)r@proxyrrconnection_kwargsr#rD)rr#rDrrrrrs&   zSentinelConnectionPool.__init__c Cs:|jrdnd}dt|jdt|jd|jd|d S)NZmasterr1rr r!(z))>)r.r"rrr#)rZrolerrrr(szSentinelConnectionPool.__repr__cst|jdSr,)rrErTr6rrrrEs zSentinelConnectionPool.resetcCs|jjSr,)rTrFr6rrrrFsz%SentinelConnectionPool.master_addresscs4|j p|jo|j|j|jfk}t}|o||Sr,)r.rFr$r%rowns_connection)r connectioncheckparentrrrrWs z&SentinelConnectionPool.owns_connectioncC |jSr,)rTr/r6rrrr/s z)SentinelConnectionPool.get_master_addresscCr[)zRound-robin slave balancer)rTr0r6rrrr0s z$SentinelConnectionPool.rotate_slaves) rrr__doc__rr(rEpropertyrFrWr/r0r>rrrrrQs    rQc@sleZdZdZ   dddZddZdd Zd d Zd d ZddZ ddZ e e fddZ e e fddZdS)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)Zsocket_) startswith).0kvrrr s z%Sentinel.__init__..cs$g|]\}}t||fijqSr)rsentinel_kwargs)r`hostnamer%r6rr sz%Sentinel.__init__..)itemsrd sentinelsmin_other_sentinelsrU_force_master_ip)rrhrirdZforce_master_iprUrr6rrs   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)r=rrKchoicerhexecute_commandappendall)rargsrrkrlresponse responsessentinelrrrrns  zSentinel.execute_commandcCsNg}|jD] }|d|jjqdt|jdt|jdd|dS)Nz {host}:{port}rr z (sentinels=[,z])>) rhro format_maprrUr"rrjoin)rZsentinel_addressesrtrrrr(s  zSentinel.__repr__cCs2|dr |ds |drdS|d|jkrdSdS)Nr.is_sdownis_odownFznum-other-sentinelsT)ri)rstater#rrrcheck_master_state&s 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 enumeraterhZsentinel_mastersrr rorSr{rjrMrwr ) rr#Zcollected_errorsZ sentinel_nortZmasterserzr| error_inforrrrH.s0     zSentinel.discover_mastercCs:g}|D]}|ds|drq||d|dfq|S)z1Remove slaves that are in an ODOWN or SDOWN stateryrxr|r%)ro)rrOZ slaves_aliver1rrr filter_slavesQs zSentinel.filter_slavesc CsN|jD]!}z||}Wn tttfyYqw||}|r$|SqgS)z;Returns a list of alive slaves for service ``service_name``)rhZsentinel_slavesrr r r)rr#rtrOrrrrJZs  zSentinel.discover_slavescK4d|d<t|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.dictrUupdateZ from_poolrr#Z redis_classZconnection_pool_classrrUrrr master_forfs   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. Fr.rrrrr slave_fors   zSentinel.slave_for)rNN)rrrr\rrnr(r{rHrrJrrQrrrrrrr^s$  #  )r^)rKrAtypingrZ redis.clientrZredis.commandsrZredis.connectionrrrZredis.exceptionsrr r r r rrr?r@rQr^rrrrs   A0=