import asyncio
import signal

from aio_message_handler.consumer import Consumer


async def main():
    consumer = Consumer("amqp://guest:guest@127.0.0.1/")

    @consumer.message_handler(
        exchange="myexchange",
        binding_key="key", 
        prefetch_count=5)
    async def handler(msg):
        print('received:', msg.body)
        await msg.ack()
    
    await consumer.start()


if __name__ == "__main__":
    try:
        loop = asyncio.get_event_loop()
        loop.create_task(main())
        loop.run_forever()
    finally:
        loop.close()

import asyncio
import signal

from aio_message_handler.consumer import Consumer


async def main():
    consumer = Consumer("amqp://guest:guest@127.0.0.1/")

    async def shutdown(signal, loop):
        print(f"Received exit signal {signal.name}...")
        await consumer.stop()
        loop.stop()

    loop = asyncio.get_event_loop()
    # May want to catch other signals too
    signals = (signal.SIGHUP, signal.SIGTERM, signal.SIGINT)
    for s in signals:
        loop.add_signal_handler(s, lambda s=s: asyncio.create_task(shutdown(s, loop)))

    @consumer.message_handler(
        exchange="myexchange",
        binding_key="key", 
        prefetch_count=5)
    async def handler(msg):
        print('received:', msg.body)
        await msg.ack()
    
    await consumer.start()


if __name__ == "__main__":
    try:
        loop = asyncio.get_event_loop()
        loop.create_task(main())
        loop.run_forever()
    finally:
        loop.close()