upd

YieldsLabs · Sep 15, 2024 · 6558039 · 6558039
1 parent 7186794
commit 6558039
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/infrastructure/event_dispatcher/event_worker.py b/infrastructure/event_dispatcher/event_worker.py
@@ -17,11 +17,9 @@ def __init__(
         self.event_handler = event_handler
         self.cancel_event = cancel_event
         self.dedup = dedup
-
         self.queue = asyncio.Queue()
-        self.tasks = asyncio.create_task(self._process_events())
 
-    async def _process_events(self):
+    async def run(self):
         async for event, args, kwargs in self._get_event_stream():
             await self.event_handler.handle_event(event, *args, **kwargs)
 

diff --git a/infrastructure/event_dispatcher/worker_pool.py b/infrastructure/event_dispatcher/worker_pool.py
@@ -29,6 +29,9 @@ def _initialize_workers(self, num_workers):
             for _ in range(num_workers)
         ]
 
+        for worker in self.workers:
+            asyncio.create_task(worker.run())
+
     async def dispatch_to_worker(self, event: Event, *args, **kwargs) -> None:
         priority_group = self.load_balancer.determine_priority_group(
             event.meta.priority