VSCode Jupyter单元异步任务无法跨单元启停问题求助

核心思路 #

Jupyter的IPython环境内置了持续运行的asyncio事件循环，直接执行阻塞式异步协程会占用当前单元线程，导致后续单元无法运行。解决方案是把异步任务作为后台任务提交到事件循环，通过全局变量保存任务对象和控制标志，实现跨单元的启停控制。

启动异步任务单元 #

import asyncio

# 全局变量：控制任务启停的标志、保存后台任务对象
stop_task = False
background_task = None

async def persistent_task():
    global stop_task
    counter = 0
    while not stop_task:
        print(f"后台任务运行中，计数: {counter}")
        counter += 1
        await asyncio.sleep(1)  # 模拟异步耗时操作
    print("后台任务已优雅停止")

async def launch_task():
    global stop_task, background_task
    # 重置控制标志
    stop_task = False
    # 将任务提交到当前事件循环（后台运行）
    background_task = asyncio.create_task(persistent_task())
    print("后台任务已启动")

# 启动任务（Jupyter支持直接await协程）
await launch_task()

运行此单元后，任务会在后台持续输出，当前单元立即执行完成，下方单元可正常操作。

停止异步任务单元 #

提供两种停止方式，按需选择：

方式1：优雅停止（等待当前循环结束） #

# 设置停止标志，让任务自行退出
stop_task = True

# 可选：等待任务完全结束后输出确认
if background_task and not background_task.done():
    await background_task
    print("任务已完成优雅停止")

方式2：强制停止（立即中断任务） #

if background_task and not background_task.done():
    # 取消任务
    background_task.cancel()
    try:
        # 捕获任务被取消时抛出的异常
        await background_task
    except asyncio.CancelledError:
        print("任务已被强制取消")

关键说明 #

• 全局变量stop_task和background_task必须在启动单元中定义，确保两个单元共享同一变量上下文。
• 不要手动创建新的asyncio事件循环（如asyncio.new_event_loop()），否则会和Jupyter内置循环冲突，导致任务无法后台运行。
• 若启动多个任务，需维护任务列表，避免任务对象被覆盖。

内容的提问来源于stack exchange，提问作者DL-Newbie