fix: unsubscribe from sub queue on exit

ApeWorX · Nov 13, 2023 · 338717f · 338717f
1 parent c6b632b
commit 338717f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/silverback/runner.py b/silverback/runner.py
@@ -99,6 +99,8 @@ async def _block_task(self, block_handler: AsyncTaskiqDecoratedTask):
             result = await block_task.wait_result()
             self._handle_result(result)
 
+        await self.subscriptions.unsubscribe(sub_id)
+
     async def _event_task(
         self, contract_event: ContractEvent, event_handler: AsyncTaskiqDecoratedTask
     ):
@@ -124,6 +126,8 @@ async def _event_task(
             result = await event_task.wait_result()
             self._handle_result(result)
 
+        await self.subscriptions.unsubscribe(sub_id)
+
     async def run(self):
         async with Web3SubscriptionsManager(self.ws_uri) as subscriptions:
             self.subscriptions = subscriptions