python里使用Condition对象来唤醒指定数量的协程

最新推荐文章于 2025-07-04 10:36:30 发布

caimouse

最新推荐文章于 2025-07-04 10:36:30 发布

阅读量1.9k

点赞数

CC 4.0 BY-SA版权

分类专栏： milang(小语）文章标签： asyncio python tensorflow

本文链接：https://blog.youkuaiyun.com/caimouse/article/details/77916801

milang(小语）专栏收录该内容

389 篇文章

订阅专栏

本文介绍Python的asyncio库中Condition对象的用法，通过示例展示如何利用Condition对象指定数量的协程被唤醒，以及其与事件Event的区别。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在asyncio库里，定义Condition对象，它的行为与事件Event有点像，区别是事件是通知所有对象，Condition对象可以指定一定数量的协程被通知，它是通过函数notify()来实现的，如果参数里放2，就是通知两个协程，例子如下：

import asyncio


async def consumer(condition, n):
    with await condition:
        print('consumer {} is waiting'.format(n))
        await condition.wait()
        print('consumer {} triggered'.format(n))
    print('ending consumer {}'.format(n))


async def manipulate_condition(condition):
    print('starting manipulate_condition')

    # pause to let consumers start
    await asyncio.sleep(0.1)

    for i in range(1, 3):
        with await condition:
            print('notifying {} consumers'.format(i))
            condition.notify(n=i)
        await asyncio.sleep(0.1)

    with await condition:
        print('notifying remaining consumers')
        condition.notify_all()

    print('ending manipulate_condition')


async def main(loop):
    # Create a condition
    condition = asyncio.Condition()

    # Set up tasks watching the condition
    consumers = [
        consumer(condition, i)
        for i in range(5)
    ]

    # Schedule a task to manipulate the condition variable
    loop.create_task(manipulate_condition(condition))

    # Wait for the consumers to be done
    await asyncio.wait(consumers)


event_loop = asyncio.get_event_loop()
try:
    result = event_loop.run_until_complete(main(event_loop))
finally:
    event_loop.close()

结果输出如下：

starting manipulate_condition
consumer 2 is waiting
consumer 3 is waiting
consumer 4 is waiting
consumer 1 is waiting
consumer 0 is waiting
notifying 1 consumers
consumer 2 triggered
ending consumer 2
notifying 2 consumers
consumer 3 triggered
ending consumer 3
consumer 4 triggered
ending consumer 4
notifying remaining consumers
ending manipulate_condition
consumer 1 triggered
ending consumer 1
consumer 0 triggered
ending consumer 0