异步并发任务设置超时和融断机制

异步并发任务设置超时和融断机制

async def concurrent_task_with_timeout(coroutines, timeout=None, max_concurrency=None):
    coroutines = [coro for coro in coroutines]
    assert all(asyncio.iscoroutine(coro) for coro in coroutines), "Required all iscoroutine for coroutines"
    results = [None] * len(coroutines)
    try:
        if max_concurrency is None:
            tasks = [asyncio.create_task(coro) for coro in coroutines]
            done, pending = await asyncio.wait(tasks, timeout=timeout)
        else:
            semaphore = asyncio.Semaphore(max_concurrency)
            gated_tasks = [gated_coro(semaphore, c) for c in coroutines]
            tasks = [asyncio.create_task(gt) for gt in gated_tasks]
            done, pending = await asyncio.wait(tasks, timeout=timeout)
        for task in done:
            index = tasks.index(task)
            try:
                results[index] = task.result()
            except Exception as e:
                log.error(f"Task {index} raised an exception: {e}")
                results[index] = None
        if pending:
            log.warning(f"Timeout! Some tasks did not complete within {timeout}s")
            for p in pending:
                p.cancel()  # 取消未完成的任务
        else:
            log.debug("All tasks completed successfully")
    except asyncio.TimeoutError:
        log.error("Timeout for all tasks")
    return results
posted @   春树&暮云  阅读(7)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· winform 绘制太阳,地球,月球 运作规律
点击右上角即可分享
微信分享提示