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� | d k s$ t | � s$ t | � � | j j r9 t � � | j j � Id H} | t k r] d S| j rr | \ } } | d k r� | 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) �callable�AssertionErrorr 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 r2 | s z
Channel.close)r �
__module__� __qualname__�__doc__r r �propertyr r r r! r& r* r- r0 r2 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)r8 r9 r: r; )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)r9 r8 �StopAsyncIterationr! r: r; )r r# r r r � __anext__� s z_IterHelper.__anext__N)r8 r9 r: r; )r r4 r5 � __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 s | d k s* t | � s* t d | f � � | d k rB t j � } | d k rW | j } t d | � | _ i | _ | | _ d S)Nz!on_close must be None or callabler
) r r �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 �lenrD 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
rD �_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
rD rG )r �patternZ enc_patternr' r r r rI � 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�MappingProxyTyperD �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 )rJ r' r r r rK � s z%Receiver.patterns.<locals>.<dictcomp>)rL rM rD rN )r r r r �patterns� s zReceiver.patternsr r c � s� | d k s$ t | � s$ t | � � | j j r9 t � � | j j � Id H} | t k r] d S| \ } } | j r~ | \ } } | d k r� | 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 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+ r1 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 )rJ r' r r r � <genexpr>+ s z%Receiver.is_active.<locals>.<genexpr>)r r �anyrD rN )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 r2 )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)rD rT )r rH r3 r r r rC 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 r1 r Zwarningr. )r r/ �senderr r r � _put_nowaitQ s
zReceiver._put_nowaitc C s3 | j j | j | j f � | j | d | �d S)Nr3 )rD �popr r rE )r rV r3 r r r �_closeY s zReceiver._close)r r4 r5 r6 r r rH rI r7 rO rP r! r- r rT r* rC rW rY 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)rG 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 rZ )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)r[ )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)NrV )rZ rW )r r/ r r r r0 � s z_Sender.put_nowaitc C s0 | j r
d Sd | _ | j j | d | �d S)NTr3 )r[ rZ rY )r r3 r r r r2 � s z
_Sender.close)
r r4 r5 r6 r r r7 r r r r! r0 r2 r r r r rG ^ s rG 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 rB �Event�_eventr[ )r r
r r r r � s zClosableQueue.__init__c � s/ x( | j p | j s* | j j � Id Hq Wd S)NT)r r[ r` r, )r r r r r, � s zClosableQueue.waitc � s| | j � Id H| j s< | j s< t d | j | j f � � | j rS | j rS t S| j j � } | j sx | j j � | S)NzUnexpected queue state)r, r r[ r r �popleftr` �clear)r �itemr r r r! � s
zClosableQueue.getc C s. | j r
d S| j j | � | j j � d S)N)r[ r �appendr` �set)r rc r r r r. � s zClosableQueue.putc C s d | _ | j j � d S)z,Mark queue as closed and notify all waiters.TN)r[ r` re )r r r r r2 � s zClosableQueue.closec C s | j S)N)r[ )r r r r r1 � s zClosableQueue.closedc C s | j o | j S)N)r[ 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)rF r )r r r r r � s zClosableQueue.qsizec C s. | j r d n d } d j | t | j � � S)Nr1 �openz<Queue {} size:{}>)r[ r rF r )r r1 r r r r � s zClosableQueue.__repr__)r r4 r5 r r, r! r. r2 r7 r1 r r+ r r r r r r r � s r )rB r$ rL r] �abcr �utilr �errorsr �logr �__all__�objectr r r) r rG r r r r r �<module> s p�/