Skip to content

Commit b5cb23d

Browse files
authored
refactor: Registration of injectable buses
1 parent 4cae09f commit b5cb23d

File tree

2 files changed

+337
-264
lines changed

2 files changed

+337
-264
lines changed

cq/_core/message.py

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,12 @@
3434
)
3535

3636

37+
@injection.injectable(
38+
ignore_type_hint=True,
39+
inject=False,
40+
on=CommandBus,
41+
mode="fallback",
42+
)
3743
def new_command_bus(*, threadsafe: bool | None = None) -> Bus[Command, Any]:
3844
bus = SimpleBus(command_handler.manager)
3945
transaction_scope_middleware = InjectionScopeMiddleware(
@@ -45,27 +51,21 @@ def new_command_bus(*, threadsafe: bool | None = None) -> Bus[Command, Any]:
4551
return bus
4652

4753

54+
@injection.injectable(
55+
ignore_type_hint=True,
56+
inject=False,
57+
on=EventBus,
58+
mode="fallback",
59+
)
4860
def new_event_bus() -> Bus[Event, None]:
4961
return TaskBus(event_handler.manager)
5062

5163

64+
@injection.injectable(
65+
ignore_type_hint=True,
66+
inject=False,
67+
on=QueryBus,
68+
mode="fallback",
69+
)
5270
def new_query_bus() -> Bus[Query, Any]:
5371
return SimpleBus(query_handler.manager)
54-
55-
56-
@injection.injectable(inject=False, mode="fallback")
57-
def _() -> CommandBus: # type: ignore[type-arg]
58-
return new_command_bus()
59-
60-
61-
@injection.injectable(inject=False, mode="fallback")
62-
def _() -> EventBus:
63-
return new_event_bus()
64-
65-
66-
@injection.injectable(inject=False, mode="fallback")
67-
def _() -> QueryBus: # type: ignore[type-arg]
68-
return new_query_bus()
69-
70-
71-
del _

0 commit comments

Comments
 (0)