File: //usr/lib/python3.9/site-packages/redis/asyncio/__pycache__/sentinel.cpython-39.pyc
a
��bG4 � @ s� d dl Z d dlZd dlZd dlmZmZmZmZmZm Z d dl
m Z d dl m
Z
mZmZmZ d dlmZ d dlmZmZmZmZ d dlmZ G dd � d e�ZG d
d � d e�ZG d d
� d
e
�Z G dd� de e�Z G dd� de�Z G dd� de�Z dS )� N)�
AsyncIterator�Iterable�Mapping�Sequence�Tuple�Type)�Redis)�
Connection�ConnectionPool�
EncodableT�
SSLConnection)�AsyncSentinelCommands)�ConnectionError�
ReadOnlyError�
ResponseError� TimeoutError)� str_if_bytesc @ s e Zd ZdS )�MasterNotFoundErrorN��__name__�
__module__� __qualname__� r r �:/usr/lib/python3.9/site-packages/redis/asyncio/sentinel.pyr s r c @ s e Zd ZdS )�SlaveNotFoundErrorNr r r r r r s r c sL e Zd Z� fdd�Zdd� Z� fdd�Zdd� Zd
ed
�� fd d �
Z� Z S )�SentinelManagedConnectionc s" |� d�| _t� jf i |�� d S )N�connection_pool)�popr �super�__init__)�self�kwargs�� __class__r r r s z"SentinelManagedConnection.__init__c C sD | j }| jj� d|j� �}| jr<d| j� d| j� �}||7 }|d S )N� <service=z,host=z,port=�>)r r# r � service_name�host�port)r Zpool�sZ host_infor r r �__repr__ s z"SentinelManagedConnection.__repr__c � sV |\| _ | _t� �� I d H | jjrR| �d�I d H t| �� I d H �dkrRt d��d S )NZPINGZPONGz PING failed)
r' r( r �connectr �check_connectionZ send_commandr �
read_responser )r Zaddressr" r r �
connect_to' s z$SentinelManagedConnection.connect_toc � s� | j r
d S | jjr0| �| j�� I d H �I d H nL| j�� 2 z:3 d H W }z| �|�I d H W S tyr Y q:Y q:0 q:6 t�d S �N)Z_readerr � is_masterr. �get_master_address�
rotate_slavesr r )r �slaver r r r+ / s z!SentinelManagedConnection.connectF��disable_decodingc � sN zt � j|d�I d H W S tyH | jjrB| �� I d H td��� Y n0 d S )Nr4 z"The previous master is now a slave)r r- r r r0 �
disconnectr )r r5 r" r r r- < s z'SentinelManagedConnection.read_response)F)
r r r r r* r. r+ �boolr- �
__classcell__r r r" r r s
r c @ s e Zd ZdS )� SentinelManagedSSLConnectionNr r r r r r9 K s r9 c s\ e Zd ZdZ� fdd�Zdd� Z� fdd�Zed�� fd d
� Zd d � Z e
d
�dd�Z � 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 sz |� d|�dd�rtnt�|d< |�dd�| _|�dd�| _t� jf i |�� t� | �| j
d< || _ || _ d | _
d | _d S )NZconnection_classZsslFr0 Tr, r )�getr r9 r r0 r, r r �weakref�proxy�connection_kwargsr& �sentinel_manager�master_address�slave_rr_counter)r r& r? r! r" r r r W s
��z SentinelConnectionPool.__init__c C s& | j j� d| j� d| jr dp d� d�S )Nr$ �(Zmasterr3 z)>)r# r r& r0 �r r r r r* g s
� ��z SentinelConnectionPool.__repr__c s t � �� d | _d | _d S r/ )r �resetr@ rA rC r" r r rD m s
z SentinelConnectionPool.reset)�
connectionc s0 | j p | j o | j|j|jfk}|o.t� �|�S r/ )r0 r@ r' r( r �owns_connection)r rE �checkr" r r rF r s �z&SentinelConnectionPool.owns_connectionc � s@ | j �| j�I d H }| jr<| j|kr<|| _| jdd�I d H |S )NF)Zinuse_connections)r? �discover_masterr& r0 r@ r6 )r r@ r r r r1 x s
z)SentinelConnectionPool.get_master_address)�returnc C s� | j �| j�I dH }|rn| jdu r8t�dt|�d �| _tt|��D ](}| jd t|� | _|| j }|V qDz| �� I dH V W n t y� Y n0 t
d| j����dS )zRound-robin slave balancerNr � zNo slave found for ) r? �discover_slavesr& rA �randomZrandint�len�ranger1 r r )r �slaves�_r3 r r r r2 � s
z$SentinelConnectionPool.rotate_slaves)
r r r �__doc__r r* rD r rF r1 r r2 r8 r r r" r r: O s
r: c @ s� e Zd ZdZddd�Zdd� Zdd � Zeee d
�d d �Z
ed
�dd�Z e e
eeeef d�dd�Zeeeeef d�dd�Zeefeee ee d�dd�Zeefeee ee d�dd�ZdS ) �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.
r Nc sD |d u rdd� |� � D �}|� _� fdd�|D �� _|� _|� _d S )Nc S s i | ]\}}|� d �r||�qS )Zsocket_)�
startswith)�.0�k�vr r r �
<dictcomp>� s z%Sentinel.__init__.<locals>.<dictcomp>c s& g | ] \}}t f ||d �� j���qS ))r' r( )r �sentinel_kwargs)rT �hostnamer( rC r r �
<listcomp>� s �z%Sentinel.__init__.<locals>.<listcomp>)�itemsrX � sentinels�min_other_sentinelsr> )r r\ r] rX r> r rC r r � s �
�zSentinel.__init__c � st t ��dd��}d��� v r&��d� |rR� �fdd�| jD �}tj|� I dH n t�| j�j � i ���I dH dS )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.
�onceFc s" g | ]}t �|j� i �����qS r )�asyncioZTask�execute_command)rT �sentinel��argsr! r r rZ � s �z,Sentinel.execute_command.<locals>.<listcomp>NT)
r7 r; �keysr r\ r_ ZgatherrL �choicer` )r rc r! r^ Ztasksr rb r r` � s
� zSentinel.execute_commandc C sN g }| j D ](}|�|jjd � d|jjd � �� q
| jj� dd�|�� d�S )Nr' �:r( z <sentinels=[�,z]>)r\ �appendr r> r# r �join)r Zsentinel_addressesra r r r r* � s
��zSentinel.__repr__)�stater& rI c C s2 |d r|d s|d r dS |d | j k r.dS dS )Nr0 �is_sdown�is_odownFznum-other-sentinelsT)r] )r rj r& r r r �check_master_state� s
zSentinel.check_master_state)r& c
� s� t | j�D ]�\}}z|�� I dH }W n ttfy> Y q
Y n0 |�|�}|r
| �||�r
|| jd | jd< | j|< |d |d f S q
td|����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.
Nr �ipr( zNo master found for )� enumerater\ Zsentinel_mastersr r r; rm r )r r& Z sentinel_nora Zmastersrj r r r rH � s
�zSentinel.discover_master)rO rI c C s: g }|D ],}|d s|d r q|� |d |d f� q|S )z1Remove slaves that are in an ODOWN or SDOWN staterl rk rn r( )rh )r rO Z slaves_aliver3 r r r �
filter_slaves s zSentinel.filter_slaves)r& rI c � sX | j D ]L}z|�|�I dH }W n tttfy: Y qY n0 | �|�}|r| S qg S )z;Returns a list of alive slaves for service ``service_name``N)r\ Zsentinel_slavesr r r rp )r r& ra rO r r r rK s
zSentinel.discover_slaves)r& � redis_class�connection_pool_classc K s4 d|d< t | j�}|�|� |||| fi |��d�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.
Tr0 �r ��dictr> �update�r r& rq rr r! r> r r r �
master_for s
���zSentinel.master_forc K s4 d|d< t | j�}|�|� |||| fi |��d�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.
Fr0 rs rt rw r r r � slave_forB s
���zSentinel.slave_for)r N)r r r rQ r r` r* ru �strr7 rm rH r r r r r rp rK r r: r rx ry r r r r rR � s8 �
� � �� )��rR ) r_ rL r< �typingr r r r r r Zredis.asyncio.clientr Zredis.asyncio.connectionr r
r r Zredis.commandsr
Zredis.exceptionsr r r r Z redis.utilsr r r r r9 r: rR r r r r �<module> s 1E