Your IP : 216.73.216.213
�D�Z�A � @ s d d l Z d d l Z d d l Z d d l m Z m Z d d l m Z d d l m Z d d l
m Z d d l m
Z
d d l m Z d d d
d d d d d d
d d d d d d d d d d d d d d d � Z Gd d � d e
� Z Gd d � d � Z Gd d � d � Z d S) � N� )�create_connection�_PUBSUB_COMMANDS)�logger)� parse_url)�PoolClosedError)�AbcPool)�Lock�db�password�ssl�encoding�minsize�maxsize�
�parser�loop�create_connection_timeout�pool_cls�connection_clsc � s: |
r |
} n t } t | t � r� t | � \ } }
|
j d | � } |
j d | � } |
j d | � } |
j d | � } d |
k r� | p� |
d } | | | | | d | d | d | d | d | d
| d | �} y | j d d
� Id HWnA t k
r5} z! | j � | j � Id H� WYd d } ~ Xn X| S)a{ Creates Redis Pool.
By default it creates pool of Redis instances, but it is
also possible to create pool of plain connections by passing
``lambda conn: conn`` as commands_factory.
*commands_factory* parameter is deprecated since v0.2.9
All arguments are the same as for create_connection.
Returns RedisPool instance or a pool_cls if it is given.
r
r r
�timeoutr r r r r r r � override_minFN) �ConnectionsPool�
isinstance�strr �
setdefault�
_fill_free� Exception�close� wait_closed)�addressr
r r r
r r r r r r r �cls�options�poolZex� r$ �>/opt/alt/python35/lib/python3.5/site-packages/aioredis/pool.py� create_pool
s2
r& c @ s; e Z d Z d Z d d d d d d d d d d d d d d d �Z d
d � Z e d d
� � Z e d d � � Z e d d � � Z e d d � � Z
e d d � � Z d d � Z d d � Z
d d � Z d d � Z e d d � � Z d d! � Z e d" d# � � Z e d$ d% � � Z d& d' � Z d( d) � Z f d* d+ � Z d, d- � Z d. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z e d6 d7 � � Z e d8 d9 � � Z e d: d; � � Z d f d< d= � Z d> d? � Z d@ dA � Z! dB dC � Z" dD dE � Z# d dF dG � Z$ dH dI � Z% dJ dK � Z& dL dM � Z' dN dO � Z( d S)Pr zRedis connections pool.Nr r r r r c C s� | d k r t j � } | | _ | | _ | | _ | | _ | | _ | | _ | | _ | | _ | | _
t j d | � | _
t � | _ d | _ t j d t d | � d | � | _ t j d | � | _ d | _ d | _ |
| _ d S)N�maxlenr �lockr )�asyncioZget_event_loop�_address�_db� _password�_ssl� _encoding�
_parser_class�_minsize�_create_connection_timeout�_loop� collections�deque�_pool�set�_used�
_acquiringZ Conditionr �_condZEvent� _close_state�
_close_waiter� _pubsub_conn�_connection_cls) �selfr r
r r
r r r r r r r r$ r$ r% �__init__C s& $ zConnectionsPool.__init__c C s+ d j | j j | j | j | j | j � S)Nz#<{} [db:{}, size:[{}:{}], free:{}]>)�format� __class__�__name__r
r r �freesize)r> r$ r$ r% �__repr__c s zConnectionsPool.__repr__c C s | j S)zMinimum pool size.)r0 )r> r$ r$ r% r h s zConnectionsPool.minsizec C s
| j j S)zMaximum pool size.)r5 r' )r> r$ r$ r% r m s zConnectionsPool.maxsizec C s | j t | j � | j S)zCurrent pool size.)rC �lenr7 r8 )r> r$ r$ r% �sizer s zConnectionsPool.sizec C s
t | j � S)z#Current number of free connections.)rE r5 )r> r$ r$ r% rC w s zConnectionsPool.freesizec C s | j S)N)r* )r> r$ r$ r% r | s zConnectionsPool.addressc � s) | j Id H� | j � Id HWd QRXd S)zPClear pool connections.
Close and remove all free connections.
N)r9 � _do_clear)r> r$ r$ r% �clear� s zConnectionsPool.clearc � sa g } x9 | j rA | j j � } | j � | j | j � � q Wt j | d | j � Id Hd S)Nr )r5 �popleftr �appendr r) �gatherr2 )r> �waiters�connr$ r$ r% rG � s
zConnectionsPool._do_clearc � s� | j j � Id H| j Id H�� g } x9 | j rb | j j � } | j � | j | j � � q* Wx. | j D]# } | j � | j | j � � qm Wt j
| d | j � Id Ht j
d t | � � Wd QRXd S)Nr zClosed %d connection(s))r: �waitr9 r5 rI r rJ r r7 r) rK r2 r �debugrE )r> rL rM r$ r$ r% � _do_close� s
zConnectionsPool._do_closec C sA | j j � s= t j | j � d | j �| _ | j j � d S)zLClose all free and in-progress connections and mark pool as closed.
r N)r: �is_setr) �
ensure_futurerP r2 r; r6 )r> r$ r$ r% r � s zConnectionsPool.closec C s
| j j � S)zTrue if pool is closed.)r: rQ )r> r$ r$ r% �closed� s zConnectionsPool.closedc � s4 | j j � Id Ht j | j d | j �Id Hd S)z Wait until pool gets closed.Nr )r: rN r) Zshieldr; r2 )r> r$ r$ r% r � s zConnectionsPool.wait_closedc C s
| j p d S)z Currently selected db index.r )r+ )r> r$ r$ r% r
� s zConnectionsPool.dbc C s | j S)zCurrent set codec or None.)r. )r> r$ r$ r% r
� s zConnectionsPool.encodingc O s� | j | | � \ } } | d k rO | j | | | � } | j | | | | � S| j | | | | � } | j | | | | � Sd S)a+ Executes redis command in a free connection and returns
future waiting for result.
Picks connection from free pool and send command through
that connection.
If no connection is found, returns coroutine waiting for
free connection to execute command.
N)�get_connection�execute�
_check_result�
_wait_execute)r> �command�args�kwrM r �fut�coror$ r$ r% rU � s zConnectionsPool.executec G sK | j | � \ } } | d k r1 | j | | � S| j | | | i � Sd S)a� Executes Redis (p)subscribe/(p)unsubscribe commands.
ConnectionsPool picks separate connection for pub/sub
and uses it until explicitly closed or disconnected
(unsubscribing from all channels/patterns will leave connection
locked for pub/sub use).
There is no auto-reconnect for this PUB/SUB connection.
Returns asyncio.gather coroutine waiting for all channels/patterns
to receive answers.
N)rT �execute_pubsub�_wait_execute_pubsub)r> rX ZchannelsrM r r$ r$ r% r] � s
z ConnectionsPool.execute_pubsubc C s� | j � j � } | t k } | rU | j rU | j j sL | j | j j f Sd | _ x� t | j � D]t } | j d } | j j d � | j r� qe | j
r� qe | r� | | _ | j j | � | j j
| � | | j f SWd | j f S)zDGet free connection from pool.
Returns connection.
Nr r )�upper�stripr r< rS r �rangerC r5 �rotate� in_pubsub�remover7 �addr* )r> rX rY Z is_pubsub�irM r$ r$ r% rT � s&
z ConnectionsPool.get_connectionc G s | S)zjHook to check result or catch exception (like MovedError).
This method can be coroutine.
r$ )r> r[ �datar$ r$ r% rV � s z ConnectionsPool._check_resultc � sH | j | | � Id H} z | j | | | � Id HSWd | j | � Xd S)z'Acquire connection and execute command.N)�acquirerU �release)r> r rX rY rZ rM r$ r$ r% rW s z ConnectionsPool._wait_executec � s� | j r t d � � | j Id H�s | j r9 t d � � | j d k sT | j j rq | j | � Id H} | | _ | j } | j | | | � Id HSWd QRXd S)NzPool is closed)rS r r9 r< �_create_new_connectionr] )r> r rX rY rZ rM r$ r$ r% r^ s z$ConnectionsPool._wait_execute_pubsubc
� sd d } | j Id H�I xA t | j � D]' } | oI | j | j | � Id H} q% W| | _ Wd QRX| S)z~Changes db index for all free connections.
All previously acquired connections will be closed when released.
TN)r9 ra rC r5 �selectr+ )r> r
�resrf r$ r$ r% rk s %zConnectionsPool.selectc
� sV | | _ | j Id H�8 x0 t | j � D] } | j | j | � Id Hq( WWd QRXd S)N)r, r9 ra rC r5 �auth)r> r rf r$ r$ r% rm & s zConnectionsPool.authc C s$ | j r | j j r | j j Sd S)Nr )r< rS rc )r> r$ r$ r% rc , s
zConnectionsPool.in_pubsubc C s- | j r | j j r | j j St j i � S)N)r< rS �pubsub_channels�types�MappingProxyType)r> r$ r$ r% rn 2 s
z ConnectionsPool.pubsub_channelsc C s- | j r | j j r | j j St j i � S)N)r< rS �pubsub_patternsro rp )r> r$ r$ r% rq 8 s
z ConnectionsPool.pubsub_patternsc
� s� | j r t d � � | j Id H�t | j r9 t d � � xW | j d d � Id H| j r} | j j � } | j j | � | S| j j � Id Hq< WWd QRXd S)zYAcquires a connection from free pool.
Creates new connection if needed.
zPool is closedNr T)
rS r r9 r rC r5 rI r7 re rN )r> rX rY rM r$ r$ r% rh > s zConnectionsPool.acquirec C s | j j | � | j s� | j r? t j d | � | j � n� | j re t j d | � | j � n} | j r� t j d | � | j � nW | j | j k r� | j
r� | j | j
k r� | j j
| � q� | j � n
| j � t j | j � d | j �d S)z�Returns used connection back into pool.
When returned connection has db index that differs from one in pool
the connection will be closed and dropped.
When queue of free connections is full the connection will be dropped.
z,Connection %r is in transaction, closing it.z/Connection %r is in subscribe mode, closing it.z/Connection %r has pending commands, closing it.r N)r7 rd rS Zin_transactionr Zwarningr rc Z_waitersr
r rC r5 rJ r) rR �_wakeupr2 )r> rM r$ r$ r% ri S s(
zConnectionsPool.releasec C sT xM t | j � D]< } | j d } | j r< | j j � q | j j d � q Wd S)Nr r )ra rC r5 rS rI rb )r> rf rM r$ r$ r% � _drop_closedu s
z ConnectionsPool._drop_closedc � s | j � x | j | j k r� | j d 7_ z= | j | j � Id H} | j d � Id H| j j | � Wd | j d 8_ | j � Xq
W| j r� d S| rxw | j r| j | j
k r| j d 7_ z+ | j | j � Id H} | j j | � Wd | j d 8_ | j � Xq� Wd S)Nr Zping) rs rF r r8 rj r* rU r5 rJ rC r )r> r rM r$ r$ r% r } s&
zConnectionsPool._fill_freec C sR t | d | j d | j d | j d | j d | j d | j d | j d | j �S) Nr
r r r
r r r r ) r r+ r, r- r. r/ r1 r= r2 )r> r r$ r$ r% rj � s z&ConnectionsPool._create_new_connectionc � sB | j Id H� | j j � Wd QRX| d k r> | j � Id Hd S)N)r9 Znotifyr )r> Z closing_connr$ r$ r% rr � s zConnectionsPool._wakeupc C s t d � � d S)Nz6'await' should be used as a context manager expression)� RuntimeError)r> r$ r$ r% � __enter__� s zConnectionsPool.__enter__c G s d S)Nr$ )r> rY r$ r$ r% �__exit__� s zConnectionsPool.__exit__c c s$ | j � j � Ed H} t | | � S)N)rh � __await__�_ConnectionContextManager)r> rM r$ r$ r% rw � s zConnectionsPool.__await__c C s
t | � S)z�Return async context manager for working with connection.
async with pool.get() as conn:
await conn.execute('get', 'my-key')
)� _AsyncConnectionContextManager)r> r$ r$ r% �get� s zConnectionsPool.get))rB �
__module__� __qualname__�__doc__r? rD �propertyr r rF rC r rH rG rP r rS r r
r
rU r] rT rV rW r^ rk rm rc rn rq rh ri rs r rj rr ru rv rw rz r$ r$ r$ r% r @ sR
" r c @ s: e Z d Z d
Z d d � Z d d � Z d d � Z d S) rx r5 �_connc C s | | _ | | _ d S)N)r5 r )r> r# rM r$ r$ r% r? � s z"_ConnectionContextManager.__init__c C s | j S)N)r )r> r$ r$ r% ru � s z#_ConnectionContextManager.__enter__c C s1 z | j j | j � Wd d | _ d | _ Xd S)N)r5 ri r )r> �exc_type� exc_value�tbr$ r$ r% rv � s z"_ConnectionContextManager.__exit__N)r5 r )rB r{ r| � __slots__r? ru rv r$ r$ r$ r% rx � s rx c @ s: e Z d Z d
Z d d � Z d d � Z d d � Z d S) ry r5 r c C s | | _ d | _ d S)N)r5 r )r> r# r$ r$ r% r? � s z'_AsyncConnectionContextManager.__init__c � s$ | j j � Id H} | | _ | j S)N)r5 rh r )r> rM r$ r$ r% �
__aenter__� s z)_AsyncConnectionContextManager.__aenter__c � s1 z | j j | j � Wd d | _ d | _ Xd S)N)r5 ri r )r> r� r� r� r$ r$ r% � __aexit__� s z(_AsyncConnectionContextManager.__aexit__N)r5 r )rB r{ r| r� r? r� r� r$ r$ r$ r% ry � s ry )r) r3 ro Z
connectionr r �logr �utilr �errorsr �abcr Zlocksr r& r rx ry r$ r$ r$ r% �<module> s 0� �