1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
import asyncio
import aioredis
STOPWORD = 'STOP'
async def pubsub():
pool = await aioredis.create_pool(
'redis://localhost',
minsize=5, maxsize=10)
async def reader(channel):
while (await channel.wait_message()):
msg = await channel.get(encoding='utf-8')
# ... process message ...
print("message in {}: {}".format(channel.name, msg))
if msg == STOPWORD:
return
with await pool as conn:
await conn.execute_pubsub('subscribe', 'channel:1')
channel = conn.pubsub_channels['channel:1']
await reader(channel) # wait for reader to complete
await conn.execute_pubsub('unsubscribe', 'channel:1')
# Explicit connection usage
conn = await pool.acquire()
try:
await conn.execute_pubsub('subscribe', 'channel:1')
channel = conn.pubsub_channels['channel:1']
await reader(channel) # wait for reader to complete
await conn.execute_pubsub('unsubscribe', 'channel:1')
finally:
pool.release(conn)
pool.close()
await pool.wait_closed() # closing all open connections
def main():
loop = asyncio.get_event_loop()
tsk = asyncio.ensure_future(pubsub(), loop=loop)
async def publish():
pub = await aioredis.create_redis(
'redis://localhost')
while not tsk.done():
# wait for clients to subscribe
while True:
subs = await pub.pubsub_numsub('channel:1')
if subs[b'channel:1'] == 1:
break
await asyncio.sleep(0, loop=loop)
# publish some messages
for msg in ['one', 'two', 'three']:
await pub.publish('channel:1', msg)
# send stop word
await pub.publish('channel:1', STOPWORD)
pub.close()
await pub.wait_closed()
loop.run_until_complete(asyncio.gather(publish(), tsk, loop=loop))
if __name__ == '__main__':
import os
if 'redis_version:2.6' not in os.environ.get('REDIS_VERSION', ''):
main()
|
