Курс Python → Обмен данными с asyncio.Queue

Asyncio.Queue предоставляет удобный способ обмена данными между асинхронными задачами в Python. Он позволяет одной задаче помещать данные в очередь, а другой задаче забирать их для обработки. Это особенно полезно в случаях, когда необходимо организовать взаимодействие между разными частями программы, работающими параллельно.

Для использования asyncio.Queue необходимо создать объект класса asyncio.Queue(). Затем асинхронные задачи могут использовать методы put() для добавления элементов в очередь и get() для извлечения элементов из очереди. Кроме того, можно использовать методы task_done() для уведомления о завершении обработки элемента и join() для ожидания завершения обработки всех элементов очереди.


import asyncio

async def producer(queue):
    for i in range(5):
        await queue.put(i)
        print(f'Producer put: {i}')
    await queue.put(None)

async def consumer(queue):
    while True:
        data = await queue.get()
        if data is None:
            break
        print(f'Consumer got: {data}')
        queue.task_done()

async def main():
    queue = asyncio.Queue()
    producer_task = asyncio.create_task(producer(queue))
    consumer_task = asyncio.create_task(consumer(queue))
    await asyncio.gather(producer_task, consumer_task)
    await queue.join()

asyncio.run(main())

В приведенном примере мы создаем две асинхронные задачи — producer, который помещает числа от 0 до 4 в очередь, и consumer, который извлекает и печатает эти числа. Когда producer завершает свою работу, он помещает значение None в очередь, чтобы consumer мог завершить свою работу. Мы используем asyncio.gather() для запуска обеих задач параллельно и await queue.join() для ожидания завершения обработки всех элементов очереди.

Твои коллеги будут рады, поделись в