Python 3 asyncio basic producer / consumer example

asyncio_producer_consumer.py

import asyncio
import random

q = asyncio.Queue()


async def producer(num):
    while True:
        await q.put(num + random.random())
        await asyncio.sleep(random.random())

async def consumer(num):
    while True:
        value = await q.get()
        print('Consumed', num, value)

loop = asyncio.get_event_loop()

for i in range(6):
    loop.create_task(producer(i))

for i in range(3):
    loop.create_task(consumer(i))

loop.run_forever()

Question:
Let's say there is a really heavy async def function I want to put into a queue. E.g.

async def bigcalculation():
  ...

How would I put this into the queue?
Is await q.put(await bigcalculation()) correct or would that cause it to immediately do the calculation at the producer?

@NNTin
await puts the coroutine into the only thread of the event loop. So it will block the producer, as well as all other producers.

You can use asyncio.run_in_executor to move the big calculation to other threads, e.g.

def bigcalculation():
  ...

await q.put(await asyncio.run_in_executor(None, bigcalculation))