feat: ✨ Add threadsafe parameter new_command_bus function

Author: remimd

cq/_core/message.py

@@ -1,4 +1,4 @@
-from typing import Any
+from typing import Any, Final
 
 import injection
 
@@ -22,23 +22,24 @@
 AnyCommandBus = CommandBus[Any]
 
 
-command_handler: HandlerDecorator[Command, Any] = HandlerDecorator(
+command_handler: Final[HandlerDecorator[Command, Any]] = HandlerDecorator(
     SingleHandlerManager(),
 )
-event_handler: HandlerDecorator[Event, None] = HandlerDecorator(
+event_handler: Final[HandlerDecorator[Event, None]] = HandlerDecorator(
     MultipleHandlerManager(),
 )
-query_handler: HandlerDecorator[Query, Any] = HandlerDecorator(
+query_handler: Final[HandlerDecorator[Query, Any]] = HandlerDecorator(
     SingleHandlerManager(),
 )
 
 
 @injection.injectable(inject=False, mode="fallback")
-def new_command_bus() -> CommandBus:  # type: ignore[type-arg]
+def new_command_bus(*, threadsafe: bool | None = None) -> CommandBus:  # type: ignore[type-arg]
     bus = SimpleBus(command_handler.manager)
     transaction_scope_middleware = InjectionScopeMiddleware(
         CQScope.TRANSACTION,
         exist_ok=True,
+        threadsafe=threadsafe,
     )
     bus.add_middlewares(transaction_scope_middleware)
     return bus

cq/middlewares/scope.py

@@ -21,11 +21,13 @@ class InjectionScopeMiddleware:
 
     async def __call__(self, *args: Any, **kwargs: Any) -> MiddlewareResult[Any]:
         async with AsyncExitStack() as stack:
-            context_manager = adefine_scope(self.scope_name, threadsafe=self.threadsafe)
+            cm = adefine_scope(self.scope_name, threadsafe=self.threadsafe)
             try:
-                await stack.enter_async_context(context_manager)
+                await stack.enter_async_context(cm)
+
             except ScopeAlreadyDefinedError:
                 if not self.exist_ok:
                     raise
-            del context_manager
+
+            del cm
             yield