Your IP : 216.73.216.213
B
�as- � @ s� d dl Z d dlZd dlmZ d dlmZmZ d dlmZm Z m
Z
m Z d dl m
Z
G dd� de�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� d�ZdS )� N)�Redis)�ConnectionPool�
Connection)�ConnectionError�
ResponseError�
ReadOnlyError� TimeoutError)� str_if_bytesc @ s e Zd ZdS )�MasterNotFoundErrorN)�__name__�
__module__� __qualname__� r r �?/opt/alt/python37/lib/python3.7/site-packages/redis/sentinel.pyr
s r
c @ s e Zd ZdS )�SlaveNotFoundErrorN)r r r
r r r r r s r c sD e Zd Z� fdd�Zdd� Z� fdd�Zdd� Z� fd d
�Z� ZS ) �SentinelManagedConnectionc s |� d�| _t� jf |� d S )N�connection_pool)�popr �super�__init__)�self�kwargs)� __class__r r r s z"SentinelManagedConnection.__init__c C s< | j }dt| �j|jf }| jr8d| j| jf }|| }|S )Nz%s<service=%s%%s>z,host=%s,port=%s)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 Zaddress)r r r �
connect_to s
z$SentinelManagedConnection.connect_toc C sb | j r
d S | jjr$| �| j�� � n:x4| j�� D ]&}y
| �|�S tk
rT w0Y q0X q0W t�d S )N)Z_sockr � is_masterr" �get_master_address�
rotate_slavesr r )r �slaver r r r ( s
z!SentinelManagedConnection.connectc s> y
t � �� S tk
r8 | jjr2| �� td��� Y nX d S )Nz"The previous master is now a slave)r r! r r r# �
disconnectr )r )r r r r! 5 s
z'SentinelManagedConnection.read_response) r r r
r r r" r r! �
__classcell__r r )r r r s
r 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 sZ |� dt�|d< |�dd�| _|�dd�| _t� jf |� t�| �| j d< || _
|| _ d S )NZconnection_classr# Tr Fr ) �getr r r# r r r �weakref�proxy�connection_kwargsr �sentinel_manager)r r r. r )r r r r L s z SentinelConnectionPool.__init__c C s dt | �j| j| jrdpdf S )Nz%s<service=%s(%s)Zmasterr& )r r r r# )r r r r r V s z SentinelConnectionPool.__repr__c s t � �� d | _d | _d S )N)r �reset�master_address�slave_rr_counter)r )r r r r/ ] s
z SentinelConnectionPool.resetc s8 | j p | j o | j|j|jfk}tt| �}|o6|�|�S )N)r# r0 r r r r) �owns_connection)r Z
connection�check�parent)r r r r2 b s
z&SentinelConnectionPool.owns_connectionc C s4 | j �| j�}| jr0| j|kr0|| _| jdd� |S )NF)Zinuse_connections)r. �discover_masterr r# r0 r' )r r0 r r r r$ i s
z)SentinelConnectionPool.get_master_addressc c s� | j �| j�}|rl| jdkr2t�dt|�d �| _x8tt|��D ](}| jd t|� | _|| j }|V q@W y| �� V W n t k
r� Y nX t
d| j ��dS )zRound-robin slave balancerNr � zNo slave found for %r) r. �discover_slavesr r1 �randomZrandint�len�ranger$ r
r )r �slaves�_r& r r r r% t s
z$SentinelConnectionPool.rotate_slaves) r r r
�__doc__r r r/ r2 r$ r% r( r r )r r r) D s
r) c @ s^ e Zd ZdZddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z e
e fdd�Z e
e fdd�Z
dS )�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 krdd� |� � 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� j��qS r )r �sentinel_kwargs)r@ Zhostnamer )r r r �
<listcomp>� s z%Sentinel.__init__.<locals>.<listcomp>)�itemsrD � sentinels�min_other_sentinelsr- )r rG rH rD r- r )r r r � s
zSentinel.__init__c C sN g }x0| j D ]&}|�d|jjd |jjd f � q W dt| �jd�|�f S )Nz%s:%sr r z%s<sentinels=[%s]>�,)rG �appendr r- r r �join)r Zsentinel_addresses�sentinelr 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)rH )r �stater r r r �check_master_state� s
zSentinel.check_master_statec
C s� x�t | j�D ]v\}}y |�� }W n ttfk
r: w Y nX |�|�}|r | �||�r || jd | jd< | j|< |d |d fS q W td|f ��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 %rN)� enumeraterG Zsentinel_mastersr r r* rP r
)r r Z sentinel_norL ZmastersrO r r r r5 � s
zSentinel.discover_masterc C s> g }x4|D ],}|d s
|d r q
|� |d |d f� q
W |S )z1Remove slaves that are in an ODOWN or SDOWN staterN rM rQ r )rJ )r r; Z slaves_aliver& r r r �
filter_slaves� s
zSentinel.filter_slavesc C sR xL| j D ]B}y|�|�}W n tttfk
r6 wY nX | �|�}|r|S qW g S )z;Returns a list of alive slaves for service ``service_name``)rG Zsentinel_slavesr r r rS )r r rL r; r r r r7 � s
zSentinel.discover_slavesc K s0 d|d< t | j�}|�|� |||| f|�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 � redis_class�connection_pool_classr r- r r r �
master_for� s
zSentinel.master_forc K s0 d|d< t | j�}|�|� |||| f|�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# )r )rT r- rU )r r rV rW r r- r r r � slave_for s
zSentinel.slave_for)r N)r r r
r= r r rP r5 rS r7 r r) rX rY r r r r r> � s
r> )r8 r+ Z redis.clientr Zredis.connectionr r Zredis.exceptionsr r r r Z redis.utilsr r
r r r) r> r r r r �<module> s 1C