登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  文章 >  python教程

Python asyncio.Event 怎样让多个协程等待同一个完成信号

来源:17golang原创

时间:2026-11-05 14:59:28 395浏览 收藏

在初始化、配置加载或一次性准备工作完成后,多个协程往往只需要知道“可以继续了”,并不需要从通知对象里取走一条消息。这个场景适合用同一个 asyncio.Event:准备方调用 set(),所有正在 await event.wait() 的协程都会被唤醒;后来才开始等待的协程也会因为标志仍为真而立即通过。

要注意,Event 只表达共享状态,不携带完成结果,也不是线程同步工具。结果应放在独立的数据结构里,跨线程协作则应使用线程安全的同步原语。

把 Event 想成一个可被多个协程同时看到的布尔开关:set() 打开它,clear() 关闭它,wait() 只等待开关打开,不负责消费数据。

先把 Event 当成共享状态,而不是消息队列

asyncio.Event() 创建后内部标志默认为未设置。wait() 观察这个标志:标志为假时挂起当前协程,为真时立即返回。它没有“取走通知”的动作,所以多个等待者可以同时通过。

asyncio.Event 内部标志与 set、wait、clear、is_set 的静态关系说明图
图1:结构说明图,查看 Event 状态标志与四个操作之间的关系;它不是运行截图。

下面的示例把准备工作和等待者分开。注释只说明关键职责,正文不要把输出数字当作已经执行过的证据。

import asyncio

async def wait_until_ready(event: asyncio.Event, name: str) -> None:
    # wait 只等待共享标志,不从 Event 中取出任何业务数据
    await event.wait()
    print(f"{name} 可以继续")

async def prepare(event: asyncio.Event) -> None:
    # 这里代表配置、连接或缓存等一次性准备工作
    await asyncio.sleep(0.1)
    event.set()  # 把完成状态广播给当前和后续等待者

async def main() -> None:
    ready = asyncio.Event()
    workers = [
        asyncio.create_task(wait_until_ready(ready, "worker-A")),
        asyncio.create_task(wait_until_ready(ready, "worker-B")),
        asyncio.create_task(wait_until_ready(ready, "worker-C")),
    ]
    await prepare(ready)
    await asyncio.gather(*workers)  # 等待所有等待者完成收尾

asyncio.run(main())

这个写法的关键不在于等待者有几个,而在于它们共享同一个 ready 实例。准备方只需调用一次 set(),等待方不需要轮流领取通知。

让多个协程等待同一个完成信号

如果完成信号还伴随配置对象、连接信息或结果列表,不要把这些数据塞进 Event。发布方可以先写入独立容器,再设置事件;等待方醒来后读取容器。这样“是否完成”和“完成了什么”各自有清晰职责。

初始化协程通过共享 Event 广播给三个 worker 且结果数据独立存放的关系说明图
图2:关系说明图,查看一个共享 Event 如何连接多个等待协程,以及结果数据为何另走数据通道。
import asyncio

async def load_config(event: asyncio.Event, state: dict[str, str]) -> None:
    # 先准备业务数据,再发布“可读取”信号
    state["endpoint"] = "https://service.example"
    event.set()

async def consume_config(event: asyncio.Event, state: dict[str, str], name: str) -> str:
    # Event 只负责阻塞与唤醒,真正的数据从独立容器读取
    await event.wait()
    endpoint = state.get("endpoint")
    if endpoint is None:
        raise RuntimeError("配置已唤醒但数据尚未准备")
    return f"{name}: {endpoint}"

async def main() -> None:
    ready = asyncio.Event()
    state: dict[str, str] = {}
    tasks = [asyncio.create_task(consume_config(ready, state, name)) for name in ("worker-A", "worker-B", "worker-C")]
    await load_config(ready, state)
    results = await asyncio.gather(*tasks)
    print(results)  # 结果列表来自 gather,不是 Event 携带的消息

asyncio.run(main())

若希望每个等待者拿到不同的数据,应该考虑 asyncio.Queue;若需要在共享状态上加锁并检查条件,则更接近 asyncio.Condition。Event 的优势是信号广播简单,代价是它不保存消息内容。

用 clear() 管理下一轮生命周期

set() 会让标志保持为真,直到显式调用 clear()。因此,下一轮准备工作开始前若要让新的等待者重新阻塞,就必须先清除状态。

async def run_round(event: asyncio.Event) -> None:
    # 新一轮开始前关闭旧信号,避免新等待者直接穿过
    event.clear()
    await do_prepare_work()
    event.set()  # 当前轮次准备完成,广播给所有等待者

async def do_prepare_work() -> None:
    # 这里只代表异步准备动作,实际项目中替换成真实工作
    await asyncio.sleep(0.1)

如果事件只表示“应用已经完成初始化”,通常只调用一次 set(),不需要反复 clear()。只有把它作为可重复的阶段门闩时,才把清除和重新设置放进明确的轮次管理中。is_set() 适合做状态展示或分支判断,但它本身不是等待,也不能替代 wait()。

常见误区与原语选择

需求更合适的选择原因
多个协程等待同一个状态变真asyncio.Event一次设置可以唤醒所有等待者,后续等待者也能观察到已设置状态
等待者要按顺序领取不同数据asyncio.Queue队列保存消息,消费者之间会分摊项目
等待条件并保护共享状态asyncio.Condition等待与锁结合,醒来后重新检查条件
同一时刻只允许一个协程进入临界区asyncio.Lock提供互斥,而不是广播完成信号

另外,asyncio 同步原语不是线程安全的。不要从一个 OS 线程直接把它当作跨线程事件;线程之间应使用 threading 体系,或通过事件循环提供的线程间调度接口把工作安全地交回目标循环。

相关问题

set() 之后再调用 wait() 会怎样?

只要没有调用 clear(),标志仍为真,新的 wait() 会立即返回。

Event 能传递初始化结果吗?

不能。把结果放进独立容器、Future 或其他数据通道,Event 只负责通知“可以读取”。

为什么多个协程会一起被唤醒?

因为 set() 面向的是共享标志,官方语义是唤醒所有正在等待该事件的任务,而不是只选择一个消费者。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>