import asyncio

import aioredis

​

async def handle_msg(msg):

    print('Got Message:', msg)

    i = int(msg['sleep_for'])

    print('Sleep for {}'.format(i))

    await asyncio.sleep(i)

    print('End sleep')

​

async def reader(ch):

    while (await ch.wait_message()):

        msg = await ch.get_json()

        asyncio.ensure_future(handle_msg(msg))

​

async def main():

    sub = await aioredis.create_redis(('localhost', 6379))

    res = await sub.subscribe('chan:1')

    ch1 = res[0]

    tsk = await reader(ch1)

​

​

loop = asyncio.get_event_loop()

loop.run_until_complete(main())

loop.close() 

​

​

import asyncio

​

import async_timeout

​

import aioredis

​

STOPWORD = "STOP"

​

​

async def reader(channel: aioredis.client.PubSub):

    while True:

        try:

            async with async_timeout.timeout(1):

                message = await channel.get_message(ignore_subscribe_messages=True)

                if message is not None:

                    print(f"(Reader) Message Received: {message}")

                    if message["data"].decode() == STOPWORD:

                        print("(Reader) STOP")

                        break

                await asyncio.sleep(0.01)

        except asyncio.TimeoutError:

            pass

​

​

async def main():

    redis = aioredis.from_url("redis://localhost")

    pubsub = redis.pubsub()

    await pubsub.subscribe("channel:1", "channel:2")

​

    future = asyncio.create_task(reader(pubsub))

​

    await redis.publish("channel:1", "Hello")

    await redis.publish("channel:2", "World")

    await redis.publish("channel:1", STOPWORD)

​

    await future

​

​

if __name__ == "__main__":

    asyncio.run(main())