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
|
"""
RPC server, aioamqp implementation of RPC examples from RabbitMQ tutorial
"""
import asyncio
import aioamqp
def fib(n):
if n == 0:
return 0
elif n == 1:
return 1
else:
return fib(n-1) + fib(n-2)
async def on_request(channel, body, envelope, properties):
n = int(body)
print(" [.] fib(%s)" % n)
response = fib(n)
await channel.basic_publish(
payload=str(response),
exchange_name='',
routing_key=properties.reply_to,
properties={
'correlation_id': properties.correlation_id,
},
)
await channel.basic_client_ack(delivery_tag=envelope.delivery_tag)
async def rpc_server():
transport, protocol = await aioamqp.connect()
channel = await protocol.channel()
await channel.queue_declare(queue_name='rpc_queue')
await channel.basic_qos(prefetch_count=1, prefetch_size=0, connection_global=False)
await channel.basic_consume(on_request, queue_name='rpc_queue')
print(" [x] Awaiting RPC requests")
event_loop = asyncio.get_event_loop()
event_loop.run_until_complete(rpc_server())
event_loop.run_forever()
|