feat: ⚡️ Removal of the use of asyncio and queue

Author: remimd

injection/_core/common/asynchronous.py

@@ -1,17 +1,7 @@
-import asyncio
 from abc import abstractmethod
-from collections.abc import Awaitable, Callable, Coroutine, Generator
+from collections.abc import Awaitable, Callable, Generator
 from dataclasses import dataclass
-from typing import Any, Protocol, runtime_checkable
-
-
-def run_sync[T](coroutine: Coroutine[Any, Any, T]) -> T:
-    loop = asyncio.get_event_loop()
-
-    try:
-        return loop.run_until_complete(coroutine)
-    finally:
-        coroutine.close()
+from typing import Any, NoReturn, Protocol, runtime_checkable
 
 
 @dataclass(repr=False, eq=False, frozen=True, slots=True)
@@ -37,13 +27,13 @@ def call(self, /, *args: P.args, **kwargs: P.kwargs) -> T:
 
 @dataclass(repr=False, eq=False, frozen=True, slots=True)
 class AsyncCaller[**P, T](Caller[P, T]):
-    callable: Callable[P, Coroutine[Any, Any, T]]
+    callable: Callable[P, Awaitable[T]]
 
     async def acall(self, /, *args: P.args, **kwargs: P.kwargs) -> T:
         return await self.callable(*args, **kwargs)
 
-    def call(self, /, *args: P.args, **kwargs: P.kwargs) -> T:
-        return run_sync(self.callable(*args, **kwargs))
+    def call(self, /, *args: P.args, **kwargs: P.kwargs) -> NoReturn:
+        raise RuntimeError("Can't call async callable synchronously.")
 
 
 @dataclass(repr=False, eq=False, frozen=True, slots=True)

injection/_core/common/lazy.py

@@ -1,56 +1,55 @@
-from collections.abc import Callable, Iterator, Mapping
-from types import MappingProxyType
+from collections.abc import AsyncIterator, Awaitable, Callable, Iterator
+from functools import partial
 
-from injection._core.common.invertible import Invertible
+from injection._core.common.asynchronous import SimpleAwaitable
+from injection._core.common.invertible import Invertible, SimpleInvertible
 
 
-class Lazy[T](Invertible[T]):
-    __slots__ = ("__iterator", "__is_set")
+def lazy[T](factory: Callable[..., T]) -> Invertible[T]:
+    def cache() -> Iterator[T]:
+        nonlocal factory
+        value = factory()
+        del factory
 
-    __iterator: Iterator[T]
-    __is_set: bool
+        while True:
+            yield value
 
-    def __init__(self, factory: Callable[..., T]) -> None:
-        self.__setup_cache(factory)
-
-    def __invert__(self) -> T:
-        return next(self.__iterator)
-
-    @property
-    def is_set(self) -> bool:
-        return self.__is_set
+    getter = partial(next, cache())
+    return SimpleInvertible(getter)
 
-    def __setup_cache(self, factory: Callable[..., T]) -> None:
-        def infinite_yield() -> Iterator[T]:
-            nonlocal factory
-            cached = factory()
-            self.__is_set = True
-            del factory
 
-            while True:
-                yield cached
+def alazy[T](factory: Callable[..., Awaitable[T]]) -> Awaitable[T]:
+    async def cache() -> AsyncIterator[T]:
+        nonlocal factory
+        value = await factory()
+        del factory
 
-        self.__iterator = infinite_yield()
-        self.__is_set = False
+        while True:
+            yield value
 
+    getter = partial(anext, cache())
+    return SimpleAwaitable(getter)
 
-class LazyMapping[K, V](Mapping[K, V]):
-    __slots__ = ("__lazy",)
 
-    __lazy: Lazy[Mapping[K, V]]
+class Lazy[T](Invertible[T]):
+    __slots__ = ("__invertible", "__is_set")
 
-    def __init__(self, iterator: Iterator[tuple[K, V]]) -> None:
-        self.__lazy = Lazy(lambda: MappingProxyType(dict(iterator)))
+    __invertible: Invertible[T]
+    __is_set: bool
 
-    def __getitem__(self, key: K, /) -> V:
-        return (~self.__lazy)[key]
+    def __init__(self, factory: Callable[..., T]) -> None:
+        @lazy
+        def invertible() -> T:
+            value = factory()
+            self.__is_set = True
+            return value
 
-    def __iter__(self) -> Iterator[K]:
-        yield from ~self.__lazy
+        self.__invertible = invertible
+        self.__is_set = False
 
-    def __len__(self) -> int:
-        return len(~self.__lazy)
+    def __invert__(self) -> T:
+        return ~self.__invertible
 
     @property
     def is_set(self) -> bool:
-        return self.__lazy.is_set
+        return self.__is_set

injection/_core/injectables.py

@@ -12,7 +12,7 @@
     runtime_checkable,
 )
 
-from injection._core.common.asynchronous import Caller, run_sync
+from injection._core.common.asynchronous import Caller
 from injection._core.scope import Scope, get_active_scopes, get_scope
 from injection.exceptions import InjectionError
 
@@ -138,8 +138,8 @@ async def abuild(self, scope: Scope) -> T:
         cm = await self.factory.acall()
         return await scope.aenter(cm)
 
-    def build(self, scope: Scope) -> T:
-        return run_sync(self.abuild(scope))
+    def build(self, scope: Scope) -> NoReturn:
+        raise RuntimeError("Can't use async context manager synchronously.")
 
 
 class CMScopedInjectable[T](ScopedInjectable[ContextManager[T], T]):

injection/_core/module.py

@@ -1,8 +1,7 @@
 from __future__ import annotations
 
-import asyncio
 from abc import ABC, abstractmethod
-from collections import OrderedDict
+from collections import OrderedDict, deque
 from collections.abc import (
     AsyncIterator,
     Awaitable,
@@ -26,7 +25,6 @@
 )
 from inspect import signature as inspect_signature
 from logging import Logger, getLogger
-from queue import Empty, Queue
 from types import MethodType
 from typing import (
     Any,
@@ -50,7 +48,7 @@
 from injection._core.common.event import Event, EventChannel, EventListener
 from injection._core.common.invertible import Invertible, SimpleInvertible
 from injection._core.common.key import new_short_key
-from injection._core.common.lazy import Lazy, LazyMapping
+from injection._core.common.lazy import Lazy, alazy, lazy
 from injection._core.common.type import (
     InputType,
     TypeInfo,
@@ -296,6 +294,9 @@ def unlock(self) -> Self:
 
     async def all_ready(self) -> None:
         for injectable in self.__injectables:
+            if injectable.is_locked:
+                continue
+
             await injectable.aget_instance()
 
     def add_listener(self, listener: EventListener) -> Self:
@@ -511,7 +512,7 @@ def constant[T](
         mode: Mode | ModeStr = Mode.get_default(),
     ) -> Any:
         def decorator(wp: type[T]) -> type[T]:
-            lazy_instance = Lazy(wp)
+            lazy_instance = lazy(wp)
             self.injectable(
                 lambda: ~lazy_instance,
                 ignore_type_hint=True,
@@ -568,7 +569,7 @@ def make_injected_function[**P, T](
     def make_injected_function(self, wrapped, /):  # type: ignore[no-untyped-def]
         metadata = InjectMetadata(wrapped)
 
-        @metadata.on_setup
+        @metadata.task
         def listen() -> None:
             metadata.update(self)
             self.add_listener(metadata)
@@ -646,12 +647,10 @@ def aget_lazy_instance[T](
 
     def aget_lazy_instance(self, cls, default=None, *, cache=False):  # type: ignore[no-untyped-def]
         if cache:
-            coroutine = self.aget_instance(cls, default)
-            return asyncio.ensure_future(coroutine)
+            return alazy(lambda: self.aget_instance(cls, default))
 
         function = self.make_injected_function(lambda instance=default: instance)
-        metadata = function.__inject_metadata__
-        metadata.set_owner(cls)
+        metadata = function.__inject_metadata__.set_owner(cls)
         return SimpleAwaitable(metadata.acall)
 
     @overload
@@ -674,11 +673,10 @@ def get_lazy_instance[T](
 
     def get_lazy_instance(self, cls, default=None, *, cache=False):  # type: ignore[no-untyped-def]
         if cache:
-            return Lazy(lambda: self.get_instance(cls, default))
+            return lazy(lambda: self.get_instance(cls, default))
 
         function = self.make_injected_function(lambda instance=default: instance)
-        metadata = function.__inject_metadata__
-        metadata.set_owner(cls)
+        metadata = function.__inject_metadata__.set_owner(cls)
         return SimpleInvertible(metadata.call)
 
     def update[T](self, updater: Updater[T]) -> Self:
@@ -832,10 +830,7 @@ def mod(name: str | None = None, /) -> Module:
 
 @dataclass(repr=False, frozen=True, slots=True)
 class Dependencies:
-    mapping: Mapping[str, Injectable[Any]]
-
-    def __bool__(self) -> bool:
-        return bool(self.mapping)
+    lazy_mapping: Lazy[Mapping[str, Injectable[Any]]]
 
     def __iter__(self) -> Iterator[tuple[str, Any]]:
         for name, injectable in self.mapping.items():
@@ -849,10 +844,11 @@ async def __aiter__(self) -> AsyncIterator[tuple[str, Any]]:
 
     @property
     def are_resolved(self) -> bool:
-        if isinstance(self.mapping, LazyMapping) and not self.mapping.is_set:
-            return False
+        return self.lazy_mapping.is_set
 
-        return bool(self)
+    @property
+    def mapping(self) -> Mapping[str, Injectable[Any]]:
+        return ~self.lazy_mapping
 
     async def aget_arguments(self) -> dict[str, Any]:
         return {key: value async for key, value in self}
@@ -861,12 +857,13 @@ def get_arguments(self) -> dict[str, Any]:
         return dict(self)
 
     @classmethod
-    def from_mapping(cls, mapping: Mapping[str, Injectable[Any]]) -> Self:
-        return cls(mapping)
+    def from_iterable(cls, iterable: Iterable[tuple[str, Injectable[Any]]]) -> Self:
+        lazy_mapping = Lazy(lambda: dict(iterable))
+        return cls(lazy_mapping)
 
     @classmethod
     def empty(cls) -> Self:
-        return cls.from_mapping({})
+        return cls.from_iterable(())
 
     @classmethod
     def resolve(
@@ -875,8 +872,8 @@ def resolve(
         module: Module,
         owner: type | None = None,
     ) -> Self:
-        dependencies = LazyMapping(cls.__resolver(signature, module, owner))
-        return cls.from_mapping(dependencies)
+        iterable = cls.__resolver(signature, module, owner)
+        return cls.from_iterable(iterable)
 
     @classmethod
     def __resolver(
@@ -917,21 +914,21 @@ class InjectMetadata[**P, T](Caller[P, T], EventListener):
     __slots__ = (
         "__dependencies",
         "__owner",
-        "__setup_queue",
         "__signature",
+        "__tasks",
         "__wrapped",
     )
 
     __dependencies: Dependencies
     __owner: type | None
-    __setup_queue: Queue[Callable[..., Any]] | None
     __signature: Signature
+    __tasks: deque[Callable[..., Any]]
     __wrapped: Callable[P, T]
 
     def __init__(self, wrapped: Callable[P, T], /) -> None:
         self.__dependencies = Dependencies.empty()
         self.__owner = None
-        self.__setup_queue = Queue()
+        self.__tasks = deque()
         self.__wrapped = wrapped
 
     @property
@@ -964,12 +961,12 @@ def bind(
         return self.__bind(args, kwargs, additional_arguments)
 
     async def acall(self, /, *args: P.args, **kwargs: P.kwargs) -> T:
-        self.__setup()
+        self.__run_tasks()
         arguments = await self.abind(args, kwargs)
         return self.wrapped(*arguments.args, **arguments.kwargs)
 
     def call(self, /, *args: P.args, **kwargs: P.kwargs) -> T:
-        self.__setup()
+        self.__run_tasks()
         arguments = self.bind(args, kwargs)
         return self.wrapped(*arguments.args, **arguments.kwargs)
 
@@ -989,14 +986,9 @@ def update(self, module: Module) -> Self:
         self.__dependencies = Dependencies.resolve(self.signature, module, self.__owner)
         return self
 
-    def on_setup[**_P, _T](self, wrapped: Callable[_P, _T] | None = None, /) -> Any:
+    def task[**_P, _T](self, wrapped: Callable[_P, _T] | None = None, /) -> Any:
         def decorator(wp: Callable[_P, _T]) -> Callable[_P, _T]:
-            queue = self.__setup_queue
-
-            if queue is None:
-                raise RuntimeError(f"`{self}` is already up.")
-
-            queue.put_nowait(wp)
+            self.__tasks.append(wp)
             return wp
 
         return decorator(wrapped) if wrapped else decorator
@@ -1027,24 +1019,10 @@ def __bind(
         bound.arguments = bound.arguments | additional_arguments | bound.arguments
         return Arguments(bound.args, bound.kwargs)
 
-    def __close_setup_queue(self) -> None:
-        self.__setup_queue = None
-
-    def __setup(self) -> None:
-        if (queue := self.__setup_queue) is None:
-            return
-
-        while True:
-            try:
-                task = queue.get_nowait()
-            except Empty:
-                break
-
+    def __run_tasks(self) -> None:
+        while tasks := self.__tasks:
+            task = tasks.popleft()
             task()
-            queue.task_done()
-
-        queue.join()
-        self.__close_setup_queue()
 
 
 class InjectedFunction[**P, T](ABC):

injection/_core/scope.py

@@ -201,9 +201,7 @@ def __exit__(
         return self.delegate.__exit__(exc_type, exc_value, traceback)
 
     async def aenter[T](self, context_manager: AsyncContextManager[T]) -> NoReturn:
-        raise ScopeError(
-            "Synchronous scope doesn't support asynchronous context manager."
-        )
+        raise ScopeError("Synchronous scope doesn't support async context manager.")
 
     def enter[T](self, context_manager: ContextManager[T]) -> T:
         return self.delegate.enter_context(context_manager)