Your IP : 216.73.216.213
�D�Z�5 � @ s� d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l
m Z d d d g Z e
� Z Gd
d � d e � Z Gd d � d � Z Gd
d � d � Z Gd d � d e � Z Gd d � d � Z d S)� N� )�
AbcChannel)� _converters)�ChannelClosedError)�logger�Channel� EndOfStream�Receiverc @ s� e Z d Z d Z d d d � Z d d � Z e d d � � Z e d d
� � Z e d d � � Z d
d d d d d � Z
d d d � Z d
d d d d d � Z d d � Z
d d � Z d d d � Z d S) r z Wrapper around asyncio.Queue.Nc C s8 t d | � | _ t t | � | � | _ | | _ d S)N�loop)�
ClosableQueue�_queuer �type�_name� _is_pattern)�self�name�
is_patternr
� r �@/opt/alt/python35/lib/python3.5/site-packages/aioredis/pubsub.py�__init__ s zChannel.__init__c C s+ d j | j j | j | j | j j � � S)Nz'<{} name:{!r}, is_pattern:{}, qsize:{}>)�format� __class__�__name__r r r �qsize)r r r r �__repr__ s zChannel.__repr__c C s | j S)z Encoded channel name/pattern.)r )r r r r r # s z Channel.namec C s | j S)z0Set to True if channel is subscribed to pattern.)r )r r r r r ( s zChannel.is_patternc C s | j j S)a Returns True until there are messages in channel or
connection is subscribed to it.
Can be used with ``while``:
>>> ch = conn.pubsub_channels['chan:1']
>>> while ch.is_active:
... msg = await ch.get() # may stuck for a long time
)r � exhausted)r r r r � is_active- s zChannel.is_active�encoding�decoderc � s� | j j r t � � | j j � Id H} | t k r9 d S| j rN | \ } } | d k ri | j | � } | d k r� | | � } | j r� | | f S| S)z�Coroutine that waits for and returns a message.
:raises aioredis.ChannelClosedError: If channel is unsubscribed
and has no messages.
N)r r r �getr r �decode)r r r �msgZ dest_channelr r r r ; s
z Channel.getzutf-8c � s | j d | d t j � Id HS)z Shortcut to get JSON messages.r r N)r �json�loads)r r r r r �get_jsonV s zChannel.get_jsonc C s" t | d d d � d | d | �S)z�Same as get method but its native coroutine.
Usage example:
>>> async for msg in ch.iter():
... print(msg)
r c S s | j S)N)r )�chr r r �<lambda>c s z Channel.iter.<locals>.<lambda>r r )� _IterHelper)r r r r r r �iterZ s z Channel.iterc � s9 | j s
d S| j j � s d S| j j � Id H| j S)z�Waits for message to become available in channel
or channel is closed (unsubscribed).
Possible usage:
>>> while (await ch.wait_message()):
... msg = await ch.get()
FTN)r r �empty�wait)r r r r � wait_messageg s zChannel.wait_messagec C s | j j | � d S)N)r �put)r �datar r r �
put_nowaity s zChannel.put_nowaitc C s | j j s | j j � d S)z~Marks channel as inactive.
Internal method, will be called from connection
on `unsubscribe` command.
N)r �closed�close)r �excr r r r0 | s z
Channel.close)r �
__module__� __qualname__�__doc__r r �propertyr r r r r$ r( r+ r. r0 r r r r r s
c @ s: e Z d Z d Z d d � Z d d � Z d d
� Z d S)
r' �_ch�
_is_active�_args�_kwc O s( | | _ | | _ | | _ | | _ d S)N)r6 r7 r8 r9 )r r% r �args�kwr r r r � s z_IterHelper.__init__c C s | S)Nr )r r r r � __aiter__� s z_IterHelper.__aiter__c � sN | j | j � s t � | j j | j | j � Id H} | d k rJ t � | S)N)r7 r6 �StopAsyncIterationr r8 r9 )r r! r r r � __anext__� s z_IterHelper.__anext__N)r6 r7 r8 r9 )r r2 r3 � __slots__r r< r> r r r r r' � s r' c @ s� e Z d Z d Z d d d d � Z d d � Z d d � Z d d
� Z e d d � � Z e d
d � � Z
d d d d d d � Z d d � Z e d d � � Z
d d � Z d d d d d d � Z d d d � Z d d � Z d d d � Z d S)!r a� Multi-producers, single-consumer Pub/Sub queue.
Can be used in cases where a single consumer task
must read messages from several different channels
(where pattern subscriptions may not work well
or channels can be added/removed dynamically).
Example use case:
>>> from aioredis.pubsub import Receiver
>>> from aioredis.abc import AbcChannel
>>> mpsc = Receiver(loop=loop)
>>> async def reader(mpsc):
... async for channel, msg in mpsc.iter():
... assert isinstance(channel, AbcChannel)
... print("Got {!r} in channel {!r}".format(msg, channel))
>>> asyncio.ensure_future(reader(mpsc))
>>> await redis.subscribe(mpsc.channel('channel:1'),
... mpsc.channel('channel:3'))
... mpsc.channel('channel:5'))
>>> await redis.psubscribe(mpsc.pattern('hello'))
>>> # publishing 'Hello world' into 'hello-channel'
>>> # will print this message:
Got b'Hello world' in channel b'hello-channel'
>>> # when all is done:
>>> await redis.unsubscribe('channel:1', 'channel:3', 'channel:5')
>>> await redis.punsubscribe('hello')
>>> mpsc.stop()
>>> # any message received after stop() will be ignored.
Nc C sU | d k r t j � } | d k r- | j } t d | � | _ i | _ | | _ d S)Nr
)�asyncioZget_event_loop�
check_stopr r �_refs� _on_close)r r
Zon_closer r r r � s zReceiver.__init__c C s( d j | j t | j � | j j � � S)Nz-<Receiver is_active:{}, senders:{}, qsize:{}>)r r �lenrB r r )r r r r r � s zReceiver.__repr__c C sh t t | � | � } | d f | j k rW t | | d d �} | | j | d f <| S| j | d f S)zvCreate a channel.
Returns ``_Sender`` object implementing
:class:`~aioredis.abc.AbcChannel`.
Fr )r r
rB �_Sender)r r Zenc_namer% r r r �channel� s zReceiver.channelc C sd t t | � | � } | d f | j k rS t | | d d �} | | j | d f <| j | d f S)z~Create a pattern channel.
Returns ``_Sender`` object implementing
:class:`~aioredis.abc.AbcChannel`.
Tr )r r
rB rE )r �patternZ enc_patternr% r r r rG � s zReceiver.patternc C s# t j d d � | j j � D� � S)zRead-only channels dict.c S s% i | ] } | j s | | j � q Sr )r r )�.0r% r r r �
<dictcomp>� s z%Receiver.channels.<locals>.<dictcomp>)�types�MappingProxyTyperB �values)r r r r �channels� s zReceiver.channelsc C s# t j d d � | j j � D� � S)zRead-only patterns dict.c S s% i | ] } | j r | | j � q Sr )r r )rH r% r r r rI � s z%Receiver.patterns.<locals>.<dictcomp>)rJ rK rB rL )r r r r �patterns� s zReceiver.patternsr r c � s� | j j r t � � | j j � Id H} | t k r9 d S| \ } } | j rZ | \ } } | d k ru | j | � } | d k r� | | � } | j r� | | | f f S| | f S)a� Wait for and return pub/sub message from one of channels.
Return value is either:
* tuple of two elements: channel & message;
* tuple of three elements: pattern channel, (target channel & message);
* or None in case Receiver is not active or has just been stopped.
:raises aioredis.ChannelClosedError: If listener is stopped
and all messages have been received.
N)r r r r r r r )r r r �objr% r! Zdest_chr r r r � s z Receiver.getc � s< | j j � s d S| j j r# d S| j j � Id H| j S)z Blocks until new message appear.TFN)r r) r/ r* r )r r r r r+ s zReceiver.wait_messagec C s0 | j j r d St d d � | j j � D� � S)z5Returns True if listener has any active subscription.Fc s s | ] } | j Vq d S)N)r )rH r% r r r � <genexpr>+ s z%Receiver.is_active.<locals>.<genexpr>)r r �anyrB rL )r r r r r & s zReceiver.is_activec C s | j j � d S)z�Stop receiving messages.
All new messages after this call will be ignored,
so you must call unsubscribe before stopping this listener.
N)r r0 )r r r r �stop- s z
Receiver.stopc C s" t | d d d � d | d | �S)z�Returns async iterator.
Usage example:
>>> async for ch, msg in mpsc.iter():
... print(ch, msg)
r c S s | j j S)N)r r )�rr r r r& B s z Receiver.iter.<locals>.<lambda>r r )r' )r r r r r r r( 9 s z
Receiver.iterc C s | j s | j � d S)ZTBDN)rB rR )r rF r1 r r r rA F s zReceiver.check_stopc C s= | j j r# t j d | | � d S| j j | | f � d S)Nz9Pub/Sub listener message after stop: sender: %r, data: %r)r r/ r Zwarningr, )r r- �senderr r r � _put_nowaitQ s
zReceiver._put_nowaitc C s3 | j j | j | j f � | j | d | �d S)Nr1 )rB �popr r rC )r rT r1 r r r �_closeY s zReceiver._close)r r2 r3 r4 r r rF rG r5 rM rN r r+ r rR r( rA rU rW r r r r r � s
)
c @ s� e Z d Z d Z d d � Z d d � Z e d d � � Z e d d � � Z e d
d � � Z d d
d d
d d � Z
d d � Z d
d d � Z d
S)rE zEWrite-Only Channel.
Does not allow direct ``.get()`` calls.
c C s8 | | _ t t | � | � | _ | | _ d | _ d S)NF)� _receiverr r
r r �_closed)r Zreceiverr r r r r r d s z_Sender.__init__c C s% d j | j j | j | j | j � S)Nz,<{} name:{!r}, is_pattern:{}, receiver:{!r}>)r r r r r rX )r r r r r j s z_Sender.__repr__c C s | j S)z Encoded channel name or pattern.)r )r r r r r o s z _Sender.namec C s | j S)z0Set to True if channel is subscribed to pattern.)r )r r r r r t s z_Sender.is_patternc C s | j S)N)rY )r r r r r y s z_Sender.is_activer Nr c � s t d � � d S)Nz.MPSC channel does not allow direct get() calls)� RuntimeError)r r r r r r r } s z _Sender.getc C s | j j | d | �d S)NrT )rX rU )r r- r r r r. � s z_Sender.put_nowaitc C s0 | j r
d Sd | _ | j j | d | �d S)NTr1 )rY rX rW )r r1 r r r r0 � s z
_Sender.close)
r r2 r3 r4 r r r5 r r r r r. r0 r r r r rE ^ s rE c @ s� e Z d Z d d d d � Z d d � Z d d � Z d d
� Z d d � Z e d
d � � Z e d d � � Z
d d � Z d d � Z d d � Z
d S)r r
Nc C s1 t j � | _ t j d | � | _ d | _ d S)Nr
F)� collections�dequer r@ ZEvent�_eventrY )r r
r r r r � s zClosableQueue.__init__c � s/ x( | j p | j s* | j j � Id Hq Wd S)NT)r rY r] r* )r r r r r* � s zClosableQueue.waitc � sO | j � Id H| j r& | j r& t S| j j � } | j sK | j j � | S)N)r* r rY r �popleftr] �clear)r �itemr r r r � s
zClosableQueue.getc C s. | j r
d S| j j | � | j j � d S)N)rY r �appendr] �set)r r` r r r r, � s zClosableQueue.putc C s d | _ | j j � d S)z,Mark queue as closed and notify all waiters.TN)rY r] rb )r r r r r0 � s zClosableQueue.closec C s | j S)N)rY )r r r r r/ � s zClosableQueue.closedc C s | j o | j S)N)rY r )r r r r r � s zClosableQueue.exhaustedc C s | j S)N)r )r r r r r) � s zClosableQueue.emptyc C s
t | j � S)N)rD r )r r r r r � s zClosableQueue.qsizec C s. | j r d n d } d j | t | j � � S)Nr/ �openz<Queue {} size:{}>)rY r rD r )r r/ r r r r � s zClosableQueue.__repr__)r r2 r3 r r* r r, r0 r5 r/ r r) r r r r r r r � s r )r@ r" rJ r[ �abcr �utilr �errorsr �logr �__all__�objectr r r' r rE r r r r r �<module> s p�/