将旧代码迁移至TaskGroup遇到的功能缺失问题咨询

1. 批量取消组内所有任务 #

TaskGroup 确实没有提供直接的 cancel_all() 方法，但可以通过两种合规方式实现需求，避免自行维护任务列表或外层捕获 CancelledError 的麻烦：

• 遍历组内任务主动取消：虽然 TaskGroup 的 _tasks 是私有属性，但在生产环境中可安全访问（Python 核心团队对此类场景的使用持默许态度）。当收到停止命令时，遍历该集合并对未完成任务调用 cancel()：

  import asyncio
  
  async def main():
      stop_trigger = asyncio.Event()
      
      # 模拟监听API停止命令的任务
      async def watch_stop_signal():
          await asyncio.sleep(15)  # 假设15秒后收到停止指令
          stop_trigger.set()
      
      async with asyncio.TaskGroup() as tg:
          tg.create_task(watch_stop_signal())
          tg.create_task(run_server_job())
          tg.create_task(background_monitor())
          
          await stop_trigger.wait()
          # 取消所有未完成任务
          for task in tg._tasks:
              if not task.done():
                  task.cancel()
  

• 基于全局信号的主动取消：在所有业务任务中嵌入对全局取消信号（如 asyncio.Event）的检查，信号触发时主动抛出 CancelledError，TaskGroup 会自动处理剩余任务的取消流程，无需依赖私有属性：

  async def run_server_job(stop_trigger: asyncio.Event):
      while not stop_trigger.is_set():
          # 执行核心业务逻辑
          await process_requests()
          await asyncio.sleep(0.5)
      # 触发任务取消
      raise asyncio.CancelledError()
  

2. 单个任务失败不影响整个组 #

TaskGroup 的默认行为是：若任意任务抛出未捕获的非 CancelledError 异常，会立即取消所有组内任务并退出上下文。要避免这种连锁反应，只需在任务内部捕获所有业务异常，并实现重试逻辑，不让异常泄露到 TaskGroup 层面：

async def scheduled_api_task(stop_trigger: asyncio.Event):
    while not stop_trigger.is_set():
        try:
            # 调用外部API
            response = await fetch_external_data()
            await process_response(response)
            await asyncio.sleep(60)  # 每分钟调用一次
        except ConnectionError as e:
            # 处理临时网络故障，记录日志后重试
            print(f"API连接失败: {str(e)}，3秒后重试")
            await asyncio.sleep(3)
        except Exception as e:
            # 处理其他未知异常，避免任务崩溃
            print(f"API任务异常: {str(e)}，10秒后重试")
            await asyncio.sleep(10)
        # 响应停止信号，及时退出
        if stop_trigger.is_set():
            break

这样，单个任务的临时失败只会在内部处理，不会触发整个 TaskGroup 的取消逻辑。同时，任务通过监听 stop_trigger，在收到停止命令时能正常退出，配合 TaskGroup 的上下文管理完成资源清理。

结论 #

TaskGroup 完全适配长期运行的服务器场景，你没有遗漏核心用法，只需结合信号监听、任务内部异常处理即可满足业务需求。

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