messagebus has uow, manages queue [handle_has_uow_and_queue]

hjwp · hjwp · commit 3812f166aea4 · 2024-03-01T15:05:12.000Z
diff --git a/src/allocation/service_layer/messagebus.py b/src/allocation/service_layer/messagebus.py
@@ -4,15 +4,19 @@
 from . import handlers
 
 if TYPE_CHECKING:
-    from allocation.service_layer import unit_of_work
+    from . import unit_of_work
 
 
 def handle(
     event: events.Event,
     uow: unit_of_work.AbstractUnitOfWork,
 ):
-    for handler in HANDLERS[type(event)]:
-        handler(event, uow=uow)
+    queue = [event]
+    while queue:
+        event = queue.pop(0)
+        for handler in HANDLERS[type(event)]:
+            handler(event, uow=uow)
+            queue.extend(uow.collect_new_events())
 
 
 HANDLERS = {
