fix: add missing silverback_shutdown task exec in runner

ApeWorX · Nov 18, 2023 · 8d2427c · 8d2427c
1 parent ec43642
commit 8d2427c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/silverback/runner.py b/silverback/runner.py
@@ -150,6 +150,17 @@ async def run(self):
 
         await asyncio.gather(*tasks)
 
+        # Execute Silverback shutdown task before shutting down the broker
+        if shutdown_handler := self.app.get_shutdown_handler():
+            task = await shutdown_handler.kiq()
+            result = await task.wait_result()
+            await self._handle_result(
+                "silverback_shutdown",
+                self.last_block_seen,
+                -1,
+                result,
+            )
+
         await self.app.broker.shutdown()