In Python 3.12 or above, use asyncio eager task factory

- This is a performance improvement when calling async functions that do not use await, as they are run immediately and never scheduled in the loop
This commit is contained in:
Nuno Campos
2025-03-27 11:00:37 -07:00
parent bfd271e00e
commit 1c5a354a7d
+6 -1
View File
@@ -10,6 +10,7 @@ T = TypeVar("T")
AnyFuture = Union[asyncio.Future, concurrent.futures.Future]
CONTEXT_NOT_SUPPORTED = sys.version_info < (3, 11)
EAGER_NOT_SUPPORTED = sys.version_info < (3, 12)
def _get_loop(fut: asyncio.Future) -> asyncio.AbstractEventLoop:
@@ -159,8 +160,12 @@ def _ensure_future(
try:
if CONTEXT_NOT_SUPPORTED:
return loop.create_task(coro_or_future, name=name)
else:
elif EAGER_NOT_SUPPORTED:
return loop.create_task(coro_or_future, name=name, context=context)
else:
return asyncio.eager_task_factory(
loop, coro_or_future, name=name, context=context
)
except RuntimeError:
if not called_wrap_awaitable:
coro_or_future.close()