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 �pool�sZ host_infor r r �__repr__ s z"SentinelManagedConnection.__repr__c sD |\| _ | _t� �� | jjr@| �d� t| �� �dkr@t d��d S )N�PING�PONGz PING failed)
r r r �connectr �check_connection� send_commandr �
read_responser )r �address)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)�_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 )N�connection_classr) Tr# Fr ) �getr r r) r# r r �weakref�proxy�connection_kwargsr �sentinel_manager)r r r5 r )r r r r L s z SentinelConnectionPool.__init__c C s dt | �j| j| jrdpdf S )Nz%s<service=%s(%s)�masterr, )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 r7 ] s
z SentinelConnectionPool.resetc s8 | j p | j o | j|j|jfk}tt| �}|o6|�|�S )N)r) r8 r r r r/ �owns_connection)r �
connection�check�parent)r r r r: b s
z&SentinelConnectionPool.owns_connectionc C s4 | j �| j�}| jr0| j|kr0|| _| jdd� |S )NF)�inuse_connections)r5 �discover_masterr r) r8 r- )r r8 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) r5 �discover_slavesr r9 �random�randint�len�ranger* r
r )r �slaves�_r, r r r r+ t s
z$SentinelConnectionPool.rotate_slaves) r r r
�__doc__r r r7 r: 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)rK �hostnamer )r r r �
<listcomp>� s z%Sentinel.__init__.<locals>.<listcomp>)�itemsrO � sentinels�min_other_sentinelsr4 )r rS rT rO r4 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]>�,)rS �appendr r4 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)rT )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)� enumeraterS �sentinel_mastersr r r1 r\ r
)r r Z sentinel_norX Zmastersr[ r r r r? � 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 staterZ rY r] r )rV )r rF 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``)rS �sentinel_slavesr r r r` )r r rX rF r r r rA � 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 )�dictr4 �update)r r � redis_class�connection_pool_classr r4 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 )rb r4 rc )r r rd re r r4 r r r � slave_for s
zSentinel.slave_for)r N)r r r
rH r r r\ r? r` rA r r/ rf rg r r r r rI � s
rI )rB r2 � redis.clientr �redis.connectionr r �redis.exceptionsr r r r � redis.utilsr r
r r r/ rI r r r r �<module> s 1C