File: //lib/python3.6/site-packages/redis/__pycache__/sentinel.cpython-36.opt-1.pyc
3
be�^�- � @ s� d dl Z d dlZd dlmZ d dlmZmZ d dlmZm Z m
Z
m Z d dl m
Z
mZmZ G dd� de�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZdS )� N)�Redis)�ConnectionPool�
Connection)�ConnectionError�
ResponseError�
ReadOnlyError� TimeoutError)� iteritems� nativestr�xrangec @ s e Zd ZdS )�MasterNotFoundErrorN)�__name__�
__module__� __qualname__� r r � /usr/lib/python3.6/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" |j d�| _tt| �jf |� d S )N�connection_pool)�popr �superr �__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 sH |\| _ | _tt| �j� | jjrD| jd� t| j � �dkrDt
d��d S )NZPINGZPONGz PING failed) r 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| jj� � n:x4| jj� D ]&}y
| j|�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 sB yt t| �j� S tk
r< | jjr6| j� td��� Y nX d S )Nz"The previous master is now a slave)r 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 s^ |j dt�|d< |jdd�| _|jdd�| _tt| �jf |� tj | �| j
d< || _ || _ d S )NZconnection_classr% Tr" Fr )
�getr r r% r" r r+ r �weakref�proxy�connection_kwargsr �sentinel_manager)r r r0 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 t| �j� d | _d | _d S )N)r r+ �reset�master_address�slave_rr_counter)r )r r r r1 ] s z SentinelConnectionPool.resetc s8 | j p | j o | j|j|jfk}tt| �}|o6|j|�S )N)r% r2 r r r r+ �owns_connection)r Z
connectionZcheck�parent)r r r r4 b s
z&SentinelConnectionPool.owns_connectionc C s4 | j j| j�}| jr0| j|kr0|| _| jdd� |S )NF)Zinuse_connections)r0 �discover_masterr r% r2 r) )r r2 r r r r& i s
z)SentinelConnectionPool.get_master_addressc c s� | j j| j�}|rl| jdkr2tjdt|�d �| _x8tt|��D ](}| jd t|� | _|| j }|V q@W y| j� V W n t k
r� Y nX t
d| j ��dS )zRound-robin slave balancerNr � zNo slave found for %r) r0 �discover_slavesr r3 �randomZrandint�lenr r& r r )r �slaves�_r( r r r r' t s
z$SentinelConnectionPool.rotate_slaves) r
r r �__doc__r r r1 r4 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� t |�D �}|� _� fdd�|D �� _|� _|� _d S )Nc S s i | ]\}}|j 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>)r rD � sentinels�min_other_sentinelsr/ )r rF rG rD r/ r )r r r � s
zSentinel.__init__c C sN g }x0| j D ]&}|jd|jjd |jjd f � q W dt| �jdj|�f S )Nz%s:%sr r z%s<sentinels=[%s]>�,)rF �appendr r/ r r
�join)r Zsentinel_addresses�sentinelr r r r � s
zSentinel.__repr__c C s4 |d s|d s|d r dS |d | j k r0dS dS )Nr% �is_sdown�is_odownFznum-other-sentinelsT)rG )r �stater r r r �check_master_state� s
zSentinel.check_master_statec
C s� x�t | j�D ]v\}}y |j� }W n ttfk
r: w Y nX |j|�}|r | j||�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)� enumeraterF Zsentinel_mastersr r r, rO r )r r Z sentinel_norK ZmastersrN r r r r6 � s
zSentinel.discover_masterc C s> g }x4|D ],}|d s
|d r q
|j |d |d f� q
W |S )z1Remove slaves that are in an ODOWN or SDOWN staterM rL rP r )rI )r r; Z slaves_aliver( r r r �
filter_slaves� s
zSentinel.filter_slavesc C sR xL| j D ]B}y|j|�}W n tttfk
r6 wY nX | j|�}|r|S qW g S )z;Returns a list of alive slaves for service ``service_name``)rF Zsentinel_slavesr r r rR )r r rK r; r r r r8 � s
zSentinel.discover_slavesc K s0 d|d< t | j�}|j|� |||| f|�d�S )aF
Returns a redis client instance for the ``service_name`` master.
A SentinelConnectionPool class is used to retrive 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 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.
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�}|j|� |||| f|�d�S )a�
Returns redis client instance for the ``service_name`` slave(s).
A SentinelConnectionPool class is used to retrive the slave's
address before establishing a new connection.
By default clients will be a 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 )rS r/ rT )r r rU rV r r/ r r r � slave_for s
zSentinel.slave_for)r N)r
r r r= r r rO r6 rR r8 r r+ rW rX r r r r r> � s
r> )r9 r- Z redis.clientr Zredis.connectionr r Zredis.exceptionsr r r r Z
redis._compatr r
r r r r r+ �objectr> r r r r �<module> s 1C