Optimization of the implementation of RelatedEvents

Author: remimd

cq/_core/related_events.py

@@ -5,6 +5,7 @@
 
 import anyio
 import injection
+from anyio.abc import TaskGroup
 
 from cq._core.message import Event, EventBus
 from cq._core.scope import CQScope
@@ -20,20 +21,23 @@ def add(self, *events: Event) -> None:
 
 
 @dataclass(repr=False, eq=False, frozen=True, slots=True)
-class SimpleRelatedEvents(RelatedEvents):
-    items: list[Event] = field(default_factory=list)
+class AnyIORelatedEvents(RelatedEvents):
+    event_bus: EventBus
+    task_group: TaskGroup
+    history: list[Event] = field(default_factory=list, init=False)
 
-    def __bool__(self) -> bool:
-        return bool(self.items)
+    def __bool__(self) -> bool:  # pragma: no cover
+        return bool(self.history)
 
     def add(self, *events: Event) -> None:
-        self.items.extend(events)
+        self.history.extend(events)
+        dispatch_method = self.event_bus.dispatch
+
+        for event in events:
+            self.task_group.start_soon(dispatch_method, event)
 
 
 @injection.scoped(CQScope.TRANSACTION, mode="fallback")
 async def related_events_recipe(event_bus: EventBus) -> AsyncIterator[RelatedEvents]:
-    yield (instance := SimpleRelatedEvents())
-
     async with anyio.create_task_group() as task_group:
-        for event in instance.items:
-            task_group.start_soon(event_bus.dispatch, event)
+        yield AnyIORelatedEvents(event_bus, task_group)