异步并发任务设置超时和融断机制
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
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· winform 绘制太阳,地球,月球 运作规律