如何检测当前代码是否运行在 uvloop 加速的事件循环中

最直接的方式是用 isinstance(loop, uvloop.Loop),但需确保 uvloop 已导入且事件循环已创建;若无法保证导入,可安全检查 loop.__class__.__name__ == “Loop” 且 __module__.startswith(“uvloop”);最佳实践是在 uvloop.install() 后设全局标记。

如何检测当前代码是否运行在 uvloop 加速的事件循环中

检查 asyncio.get_event_loop() 是否为 uvloop 实例

最直接的方式是判断当前事件循环对象的类型。uvloop 的事件循环类是 uvloop.Loop,它继承自 asyncio.AbstractEventLoop,但不是标准库asyncio.SelectorEventLoopasyncio.ProactorEventLoop

注意:必须在事件循环已启动或至少已创建后调用,否则 asyncio.get_event_loop() 可能返回未初始化的占位对象(如 Python 3.12+ 中的 asyncio._RunningLoop),导致 isinstance 判定失败。

推荐写法:

import asyncio
import uvloop

def is_running_on_uvloop(): try: loop = asyncio.get_event_loop() return isinstance(loop, uvloop.Loop) except RuntimeError:

未在事件循环中调用(例如主线程且 loop 尚未创建)

    return False

通过 loop.__class__.__name__ 做字符串匹配

当无法确保 uvloop 已导入(比如只在某些环境安装了 uvloop),直接 import uvloop 可能引发 ModuleNotFoundError。此时可绕过类型检查,改用类名识别——只要循环实例存在,其类名就明确暴露实现细节。

这个方式更轻量、无依赖,但需注意类名可能随 uvloop 版本微调(不过至今稳定为 "Loop")。

安全写法:

AdsGo AI

AdsGo AI

全自动 AI 广告专家,助您在数分钟内完成广告搭建、优化及扩量

下载

def is_running_on_uvloop():
    try:
        loop = asyncio.get_event_loop()
        return loop.__class__.__name__ == "Loop" and loop.__class__.__module__.startswith("uvloop")
    except RuntimeError:
        return False
  • 同时校验 __name____module__ 可避免误判其他同名类
  • 不建议只看 __name__,因为用户代码里也可能定义叫 Loop 的类

uvloop.install() 后主动设标识位

如果你能控制事件循环初始化流程(例如在应用入口显式调用 uvloop.install()),最可靠的做法是自己埋一个标记,而非事后检测。这规避了所有运行时反射和类型导入问题。

典型场景:FastAPI/Uvicorn 启动脚本、自研服务初始化逻辑。

示例:

import asyncio
import uvloop

在 install() 之后立即设置全局标记

uvloop.install() __using_uvloop__ = True

后续任意位置可用

if __using_uvloop__: print("Using uvloop acceleration")

  • 比运行时检测更快,零开销
  • 适用于需要提前分支逻辑的场景(如调整日志级别、启用特定监控钩子)
  • 缺点是无法覆盖“外部框架自动启用 uvloop”的情况(如 Uvicorn 用 --loop uvloop

为什么不能只靠 sys.modulesimportlib.util.find_spec

检测 "uvloop" 是否在 sys.modules 里,或能否 import,只能说明模块**可被加载**,不代表当前事件循环就是 uvloop。常见反例:

  • 项目依赖了 uvloop,但启动时没调用 uvloop.install()
  • 使用了 asyncio.set_event_loop_policy(uvloop.EventLoopPolicy()),但尚未创建 loop
  • Uvicorn 进程中 uvloop 被 import,但 worker 进程实际跑的是默认 loop(取决于启动参数)

所以,真正有意义的判定必须落在**当前活跃的 loop 实例**上,而不是模块存在性。

实际调试时如果发现 isinstance(loop, uvloop.Loop) 返回 False,优先检查是否漏了 uvloop.install() 或是否在子线程里调用了该函数(uvloop 不支持跨线程复用 loop 实例)。

https://www.php.cn/faq/2028089.html

发表回复

Your email address will not be published. Required fields are marked *