File: //lib/python3.9/site-packages/redis/__pycache__/sentinel.cpython-39.pyc
a
��bR1 � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZmZm Z d dl
m Z m Z m
Z
mZ d dlmZ G dd� de �ZG d d
� d
e �ZG d d � d e�ZG d
d� dee �ZG dd� de�ZG dd� de�ZdS )� N)�Redis)�SentinelCommands)�
Connection�ConnectionPool�
SSLConnection)�ConnectionError�
ReadOnlyError�
ResponseError� TimeoutError)� str_if_bytesc @ s e Zd ZdS )�MasterNotFoundErrorN��__name__�
__module__� __qualname__� r r �2/usr/lib/python3.9/site-packages/redis/sentinel.pyr s r c @ s e Zd ZdS )�SlaveNotFoundErrorNr
r r r r r s r c sN e Zd Z� fdd�Zdd� Z� fdd�Zdd� Zd d
� Zd� 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 }t| �j� d|j� d�}| jr@d| j� d| j� �}|| }|S )N� <service=z%s>z,host=z,port=)r �typer � service_name�host�port)r Zpool�sZ host_infor r r �__repr__ s z"SentinelManagedConnection.__repr__c sD |\| _ | _t� �� | jjr@| �d� t| �� �dkr@t 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 C sd | j r
d S | jjr$| �| j�� � n<| j�� D ],}z| �|�W S tyX Y q.Y q.0 q.t�d S �N)Z_sockr � is_masterr' �get_master_address�
rotate_slavesr r )r �slaver r r �_connect_retry( s z(SentinelManagedConnection._connect_retryc C s | j �| jdd� �S )Nc S s d S r( r )�errorr r r �<lambda>6 � z3SentinelManagedConnection.connect.<locals>.<lambda>)ZretryZcall_with_retryr- �r r r r r$ 5 s z!SentinelManagedConnection.connectFc sB zt � j|d�W S ty< | jjr6| �� td��� Y n0 d S )N)�disable_decodingz"The previous master is now a slave)r r& r r r) �
disconnectr )r r2 r r r r& 8 s z'SentinelManagedConnection.read_response)F)
r r r r r# r' r- r$ r& �
__classcell__r r r r r s
r c @ s e Zd ZdS )� SentinelManagedSSLConnectionNr
r r r r r5 G s r5 c sP e Zd ZdZ� fdd�Zdd� Z� fdd�Z� fdd �Zd
d � Zd d
� 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 sn |� d|�dd�rtnt�|d< |�dd�| _|�dd�| _t� jf i |�� t� | �| j
d< || _ || _ d S )NZconnection_classZsslFr) Tr% r )
�getr r5 r r) r% r r �weakref�proxy�connection_kwargsr �sentinel_manager)r r r; r r r r r S s
��z SentinelConnectionPool.__init__c C s, | j r
dnd}t| �j� d| j� d|� d�S )NZmasterr, r �(�))r) r r r )r Zroler r r r# a s z SentinelConnectionPool.__repr__c s t � �� d | _d | _d S r( )r �reset�master_address�slave_rr_counterr1 r r r r> e s
z SentinelConnectionPool.resetc s4 | j p | j o | j|j|jfk}t� }|o2|�|�S r( )r) r? r r! r �owns_connection)r Z
connection�check�parentr r r rA j s
�z&SentinelConnectionPool.owns_connectionc C s4 | j �| j�}| jr0| j|kr0|| _| jdd� |S )NF)Zinuse_connections)r; �discover_masterr r) r? r3 )r r? r r r r* q s
z)SentinelConnectionPool.get_master_addressc c s� | j �| j�}|rh| jdu r2t�dt|�d �| _tt|��D ](}| jd t|� | _|| j }|V q>z| �� V W n t y� Y n0 t
d| j����dS )zRound-robin slave balancerNr � zNo slave found for ) r; �discover_slavesr r@ �randomZrandint�len�ranger* r r )r �slaves�_r, r r r r+ { s
z$SentinelConnectionPool.rotate_slaves) r r r �__doc__r r# r> rA r* r+ r4 r r r r r6 K s
r6 c @ sf e Zd ZdZddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
e e fdd�Z
e e fdd�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.
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 ||fi � j���qS r )r �sentinel_kwargs)rO �hostnamer! r1 r r �
<listcomp>� s �z%Sentinel.__init__.<locals>.<listcomp>)�itemsrS � sentinels�min_other_sentinelsr: )r rW rX rS r: r r1 r r � s �
�zSentinel.__init__c O sd t |�dd��}d|�� v r&|�d� |rH| jD ]}|j|i |�� q0nt�| j�j|i |�� 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.
�onceFT)�boolr7 �keysr rW �execute_commandrG �choice)r �argsr rY �sentinelr r r r\ � s
zSentinel.execute_commandc C s@ g }| j D ]}|�d�|jj�� q
t| �j� dd�|�� d�S )Nz
{host}:{port}z <sentinels=[�,z]>)rW �append�
format_mapr r: r r �join)r Zsentinel_addressesr_ r r r r# � s
�zSentinel.__repr__c C s2 |d r|d s|d r dS |d | j k r.dS dS )Nr) �is_sdown�is_odownFznum-other-sentinelsT)rX )r �stater r r r �check_master_state� s
zSentinel.check_master_statec
C s� t | j�D ]z\}}z |�� }W n ttfy8 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.
r �ipr! zNo master found for N)� enumeraterW Zsentinel_mastersr r
r7 rg r )r r Z sentinel_nor_ Zmastersrf r r r rD � s
�zSentinel.discover_masterc C s: g }|D ],}|d s|d r q|� |d |d f� q|S )z1Remove slaves that are in an ODOWN or SDOWN statere rd rh r! )ra )r rJ Z slaves_aliver, r r r �
filter_slaves� s zSentinel.filter_slavesc C sR | j D ]F}z|�|�}W n tttfy4 Y qY n0 | �|�}|r| S qg S )z;Returns a list of alive slaves for service ``service_name``)rW Zsentinel_slavesr r r
rj )r r r_ rJ r r r rF s
zSentinel.discover_slavesc 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.
Tr) �r ��dictr: �update�r r Z redis_classZconnection_pool_classr 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.
Fr) rk rl ro r r r � slave_for3 s
���zSentinel.slave_for)r N)r r r rL r r\ r# rg rD rj rF r r6 rp rq r r r r rM � s �
�
)�rM )rG r8 Z redis.clientr Zredis.commandsr Zredis.connectionr r r Zredis.exceptionsr r r r
Z redis.utilsr r r r r5 r6 rM r r r r �<module> s 4B