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
|
#!/usr/bin/env python
"""
Rabbitmq.com pub/sub example
https://www.rabbitmq.com/tutorials/tutorial-five-python.html
"""
import asyncio
import aioamqp
import random
import sys
async def callback(channel, body, envelope, properties):
print("consumer {} received {} ({})".format(envelope.consumer_tag, body, envelope.delivery_tag))
async def receive_log():
try:
transport, protocol = await aioamqp.connect('localhost', 5672)
except aioamqp.AmqpClosedConnection:
print("closed connections")
return
channel = await protocol.channel()
exchange_name = 'topic_logs'
await channel.exchange(exchange_name, 'topic')
result = await channel.queue(queue_name='', durable=False, auto_delete=True)
queue_name = result['queue']
binding_keys = sys.argv[1:]
if not binding_keys:
print("Usage: %s [binding_key]..." % (sys.argv[0],))
sys.exit(1)
for binding_key in binding_keys:
await channel.queue_bind(
exchange_name='topic_logs',
queue_name=queue_name,
routing_key=binding_key
)
print(' [*] Waiting for logs. To exit press CTRL+C')
await channel.basic_consume(callback, queue_name=queue_name)
event_loop = asyncio.get_event_loop()
event_loop.run_until_complete(receive_log())
event_loop.run_forever()
|
