summaryrefslogtreecommitdiff
path: root/venv/lib/python3.11/site-packages/litestar/concurrency.py
diff options
context:
space:
mode:
Diffstat (limited to 'venv/lib/python3.11/site-packages/litestar/concurrency.py')
-rw-r--r--venv/lib/python3.11/site-packages/litestar/concurrency.py109
1 files changed, 0 insertions, 109 deletions
diff --git a/venv/lib/python3.11/site-packages/litestar/concurrency.py b/venv/lib/python3.11/site-packages/litestar/concurrency.py
deleted file mode 100644
index 90eadbf..0000000
--- a/venv/lib/python3.11/site-packages/litestar/concurrency.py
+++ /dev/null
@@ -1,109 +0,0 @@
-from __future__ import annotations
-
-import asyncio
-import contextvars
-from functools import partial
-from typing import TYPE_CHECKING, Callable, TypeVar
-
-import sniffio
-from typing_extensions import ParamSpec
-
-if TYPE_CHECKING:
- from concurrent.futures import ThreadPoolExecutor
-
- import trio
-
-
-T = TypeVar("T")
-P = ParamSpec("P")
-
-
-__all__ = (
- "sync_to_thread",
- "set_asyncio_executor",
- "get_asyncio_executor",
- "set_trio_capacity_limiter",
- "get_trio_capacity_limiter",
-)
-
-
-class _State:
- EXECUTOR: ThreadPoolExecutor | None = None
- LIMITER: trio.CapacityLimiter | None = None
-
-
-async def _run_sync_asyncio(fn: Callable[P, T], *args: P.args, **kwargs: P.kwargs) -> T:
- ctx = contextvars.copy_context()
- bound_fn = partial(ctx.run, fn, *args, **kwargs)
- return await asyncio.get_running_loop().run_in_executor(get_asyncio_executor(), bound_fn) # pyright: ignore
-
-
-async def _run_sync_trio(fn: Callable[P, T], *args: P.args, **kwargs: P.kwargs) -> T:
- import trio
-
- return await trio.to_thread.run_sync(partial(fn, *args, **kwargs), limiter=get_trio_capacity_limiter())
-
-
-async def sync_to_thread(fn: Callable[P, T], *args: P.args, **kwargs: P.kwargs) -> T:
- """Run the synchronous callable ``fn`` asynchronously in a worker thread.
-
- When called from asyncio, uses :meth:`asyncio.loop.run_in_executor` to
- run the callable. No executor is specified by default so the current loop's executor
- is used. A specific executor can be set using
- :func:`~litestar.concurrency.set_asyncio_executor`. This does not affect the loop's
- default executor.
-
- When called from trio, uses :func:`trio.to_thread.run_sync` to run the callable. No
- capacity limiter is specified by default, but one can be set using
- :func:`~litestar.concurrency.set_trio_capacity_limiter`. This does not affect trio's
- default capacity limiter.
- """
- if (library := sniffio.current_async_library()) == "asyncio":
- return await _run_sync_asyncio(fn, *args, **kwargs)
-
- if library == "trio":
- return await _run_sync_trio(fn, *args, **kwargs)
-
- raise RuntimeError("Unsupported async library or not in async context")
-
-
-def set_asyncio_executor(executor: ThreadPoolExecutor | None) -> None:
- """Set the executor in which synchronous callables will be run within an asyncio
- context
- """
- try:
- sniffio.current_async_library()
- except sniffio.AsyncLibraryNotFoundError:
- pass
- else:
- raise RuntimeError("Cannot set executor from running loop")
-
- _State.EXECUTOR = executor
-
-
-def get_asyncio_executor() -> ThreadPoolExecutor | None:
- """Get the executor in which synchronous callables will be run within an asyncio
- context
- """
- return _State.EXECUTOR
-
-
-def set_trio_capacity_limiter(limiter: trio.CapacityLimiter | None) -> None:
- """Set the capacity limiter used when running synchronous callable within a trio
- context
- """
- try:
- sniffio.current_async_library()
- except sniffio.AsyncLibraryNotFoundError:
- pass
- else:
- raise RuntimeError("Cannot set limiter while in async context")
-
- _State.LIMITER = limiter
-
-
-def get_trio_capacity_limiter() -> trio.CapacityLimiter | None:
- """Get the capacity limiter used when running synchronous callable within a trio
- context
- """
- return _State.LIMITER