如何使用asyncio的slow_callback_duration定位阻塞协程

核心结论 #

你并没有错误使用slow_callback_duration参数，只是误解了该参数的实际作用。

原理说明 #

asyncio的事件循环是单线程调度模型，每次只会从任务队列中取出一个任务/回调执行，slow_callback_duration统计的是事件循环单次调度一个任务的总执行时长，只会记录当前被调度的顶层任务信息，不会深入追踪协程内部的调用栈层级。

你示例中触发4秒超时警告时，事件循环正在调度main对应的根任务（Task-1）执行，执行路径是timeshare → two_level_sleeper → selfish_sleeper，全程没有await让出控制权，因此4秒的同步阻塞全被算在本次Task-1的调度时长中，警告自然只会指向顶层的Task-1，不会提示内部的selfish_sleeper信息。

无第三方工具的定位方案 #

如果不能使用yappi等分析工具，可以用以下两种方法快速定位阻塞位置：

1. 自定义慢回调钩子打印调用栈
   在main函数开启debug配置后添加如下代码，触发慢回调时会自动打印当前任务的完整调用栈：

import traceback

def handle_slow_callback(task, duration):
    print(f"\n检测到慢任务，总耗时 {duration:.3f} 秒，调用栈如下：")
    for frame in task.get_stack():
        traceback.print_stack(frame)

# 绑定慢回调处理逻辑，Python3.7+ 支持
event_loop = asyncio.get_event_loop()
event_loop.slow_callback_duration = 0.001
event_loop.set_debug(True)
# 可配合环境变量 export PYTHONASYNCIODEBUG=1 获得更全的调试信息

2. 低侵入排查方案
   在怀疑存在同步阻塞的协程入口、出口处添加打印当前时间的日志，通过相邻日志的时间差即可快速定位到阻塞的具体代码段。

内容的提问来源于stack exchange，提问作者amos