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

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

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