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* t |
t � s! t d � � |
} n t } t | t � r� t | � \ } }
|
j d | � } |
j d | � } |
j d | � } |
j d | � } d |
k r� |
d s� |
d r� | s� t d |
d | f � � | p� |
d } | | | | | d | d | d | d
| d | d | d
| �} y | j d d � Id HWnA t k
r�} 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.
z-pool_class does not meet the AbcPool contractr
r r
�timeoutr z Conflicting ssl options are setr r r r r r � override_minFN) �
issubclassr �AssertionError�ConnectionsPool�
isinstance�strr �
setdefault�
_fill_free� Exception�close� wait_closed)�addressr
r r r
r r r r r r r �cls�options�pool�ex� r' �>/opt/alt/python35/lib/python3.5/site-packages/aioredis/pool.py� create_pool
s:
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� t | t � r | d k s6 t d | t | � f � � | d k sN t d � � t | t � ri | d k s� t d | t | � f � � | | k s� t d | | f � � | d k r� t j � } | | _ | | _ | | _ | | _ | | _
| | _ | | _ | | _
| | _ t j d | � | _ t � | _ d | _ t j d t d | � d | � | _ t j d | � | _ d | _ d | _ |
| _ d S) Nr zminsize must be int >= 0z"Arbitrary pool size is disallowed.zmaxsize must be int > 0zInvalid pool min/max sizes�maxlen�lockr ) r �intr �type�asyncioZget_event_loop�_address�_db� _password�_ssl� _encoding�
_parser_class�_minsize�_create_connection_timeout�_loop� collections�deque�_pool�set�_used�
_acquiring� Conditionr �_cond�Event� _close_state�
_close_waiter� _pubsub_conn�_connection_cls) �selfr" r
r r
r r r r r r r r' r' r( �__init__C s4 $ 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)rE r' r' r( �__repr__c s zConnectionsPool.__repr__c C s | j S)zMinimum pool size.)r5 )rE r' r' r( r h s zConnectionsPool.minsizec C s
| j j S)zMaximum pool size.)r: r* )rE r' r' r( r m s zConnectionsPool.maxsizec C s | j t | j � | j S)zCurrent pool size.)rJ �lenr< r= )rE r' r' r( �sizer s zConnectionsPool.sizec C s
t | j � S)z#Current number of free connections.)rL r: )rE r' r' r( rJ w s zConnectionsPool.freesizec C s | j S)N)r/ )rE 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)r? � _do_clear)rE 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 )r: �popleftr �appendr! r. �gatherr7 )rE �waiters�connr' r' r( rN � s
zConnectionsPool._do_clearc � s� | j j � Id H| j Id H�� | j s: t | j � � g } x9 | j r{ | j j � } | j � | j | j � � qC Wx. | j
D]# } | j � | j | j � � q� Wt j | d | j
� Id Ht j d t | � � Wd QRXd S)Nr zClosed %d connection(s))rA �waitr? r= r r: rP r rQ r! r< r. rR r7 r �debugrL )rE rS rT 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)rA �is_setr. �
ensure_futurerW r7 rB r; )rE r' r' r( r � s zConnectionsPool.closec C s
| j j � S)zTrue if pool is closed.)rA rX )rE r' r' r( �closed� s zConnectionsPool.closedc � sI | j j � Id H| j d k s' t � t j | j d | j �Id Hd S)z Wait until pool gets closed.Nr )rA rU rB r r. Zshieldr7 )rE r' r' r( r! � s zConnectionsPool.wait_closedc C s
| j p d S)z Currently selected db index.r )r0 )rE r' r' r( r
� s zConnectionsPool.dbc C s | j S)zCurrent set codec or None.)r3 )rE 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)rE �command�args�kwrT r" �fut�coror' r' r( r\ � 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)r[ �execute_pubsub�_wait_execute_pubsub)rE r_ ZchannelsrT r" r' r' r( rd � 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 rC rZ r" �rangerJ r: �rotate� in_pubsub�remover< �addr/ )rE r_ r` Z is_pubsub�irT r' r' r( r[ � s&
z ConnectionsPool.get_connectionc G s | S)zjHook to check result or catch exception (like MovedError).
This method can be coroutine.
r' )rE rb �datar' r' r( r] � s z ConnectionsPool._check_resultc � sH | j | | � Id H} z | j | | | � Id HSWd | j | � Xd S)z'Acquire connection and execute command.N)�acquirer\ �release)rE r" r_ r` ra rT r' r' r( r^ s z ConnectionsPool._wait_executec � s� | j r t d � � | j d k sE | j j sE t d | j f � � | j Id H�s | j ri t d � � | j d k s� | j j r� | j | � Id H} | | _ | j } | j | | | � Id HSWd QRXd S)NzPool is closedz Expected no or closed connection)rZ r rC r r? �_create_new_connectionrd )rE r" r_ r` ra rT r' r' r( re 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)r? rh rJ r: �selectr0 )rE r
�resrm r' r' r( rr s %zConnectionsPool.selectc
� sV | | _ | j Id H�8 x0 t | j � D] } | j | j | � Id Hq( WWd QRXd S)N)r1 r? rh rJ r: �auth)rE r rm r' r' r( rt & s zConnectionsPool.authc C s$ | j r | j j r | j j Sd S)Nr )rC rZ rj )rE r' r' r( rj , s
zConnectionsPool.in_pubsubc C s- | j r | j j r | j j St j i � S)N)rC rZ �pubsub_channels�types�MappingProxyType)rE r' r' r( ru 2 s
z ConnectionsPool.pubsub_channelsc C s- | j r | j j r | j j St j i � S)N)rC rZ �pubsub_patternsrv rw )rE r' r' r( rx 8 s
z ConnectionsPool.pubsub_patternsc
� s� | j r t d � � | j Id H�� | j r9 t d � � x� | j d d � Id H| j r� | j j � } | j s t | � � | | j k s� t | | j f � � | 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) rZ r r? r rJ r: rP r r< rl rU )rE r_ r` rT r' r' r( ro > s $zConnectionsPool.acquirec C s# | | j k s! t d | f � � | j j | � | j s| j r` t j d | � | j � n� | j r� 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)Invalid connection, maybe from other poolz,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)r< r rk rZ Zin_transactionr Zwarningr rj �_waitersr
r rJ r: rQ r. rY �_wakeupr7 )rE rT r' r' r( rp 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 )rh rJ r: rZ rP ri )rE rm rT 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) r{ rM r r= rq r/ r\ r: rQ rJ r )rE r rT 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 r0 r1 r2 r3 r4 r6 rD r7 )rE r" r' r' r( rq � s z&ConnectionsPool._create_new_connectionc � sB | j Id H� | j j � Wd QRX| d k r> | j � Id Hd S)N)r? �notifyr! )rE Z closing_connr' r' r( rz � s zConnectionsPool._wakeupc C s t d � � d S)Nz6'await' should be used as a context manager expression)� RuntimeError)rE r' r' r( � __enter__� s zConnectionsPool.__enter__c G s d S)Nr' )rE r` r' r' r( �__exit__� s zConnectionsPool.__exit__c c s$ | j � j � Ed H} t | | � S)N)ro � __await__�_ConnectionContextManager)rE rT r' r' r( r� � 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)rE r' r' r( �get� s zConnectionsPool.get))rI �
__module__� __qualname__�__doc__rF rK �propertyr r rM rJ r" rO rN rW r rZ r! r
r
r\ rd r[ r] r^ re rr rt rj ru rx ro rp r{ r rq rz r~ r r� r� 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) r� r: �_connc C s | | _ | | _ d S)N)r: r� )rE r% rT r' r' r( rF � s z"_ConnectionContextManager.__init__c C s | j S)N)r� )rE r' r' r( r~ � s z#_ConnectionContextManager.__enter__c C s1 z | j j | j � Wd d | _ d | _ Xd S)N)r: rp r� )rE �exc_type� exc_value�tbr' r' r( r � s z"_ConnectionContextManager.__exit__N)r: r� )rI r� r� � __slots__rF r~ r r' r' r' r( r� � s r� c @ s: e Z d Z d
Z d d � Z d d � Z d d � Z d S) r� r: r� c C s | | _ d | _ d S)N)r: r� )rE r% r' r' r( rF � s z'_AsyncConnectionContextManager.__init__c � s$ | j j � Id H} | | _ | j S)N)r: ro r� )rE rT r' r' r( �
__aenter__� s z)_AsyncConnectionContextManager.__aenter__c � s1 z | j j | j � Wd d | _ d | _ Xd S)N)r: rp r� )rE r� r� r� r' r' r( � __aexit__� s z(_AsyncConnectionContextManager.__aexit__N)r: r� )rI r� r� r� rF r� r� r' r' r' r( r� � s r� )r. r8 rv Z
connectionr r �logr �utilr �errorsr �abcr Zlocksr r) r r� r� r' r' r' r( �<module> s 0� �