fix(worker): add shutdown_event to worker handler per TaskIQ v0.11.9

ApeWorX · Nov 24, 2024 · 1eb8268 · 1eb8268
1 parent ee68a4b
commit 1eb8268
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -68,7 +68,7 @@
         "eth-pydantic-types",  # Use same version as eth-ape
         "packaging",  # Use same version as eth-ape
         "pydantic_settings",  # Use same version as eth-ape
-        "taskiq[metrics]>=0.11.3,<0.12",
+        "taskiq[metrics]>=0.11.9,<0.12",
         "tomlkit>=0.12,<1",  # For reading/writing global platform profile
         "fief-client[cli]>=0.19,<1",  # for platform auth/cluster login
     ],

diff --git a/silverback/worker.py b/silverback/worker.py
@@ -7,6 +7,7 @@
 
 
 async def run_worker(broker: AsyncBroker, worker_count=2, shutdown_timeout=90):
+    shutdown_event = asyncio.Event()
     try:
         tasks = []
         with ThreadPoolExecutor(max_workers=worker_count) as pool:
@@ -19,7 +20,7 @@ async def run_worker(broker: AsyncBroker, worker_count=2, shutdown_timeout=90):
                     max_prefetch=0,
                 )
                 broker.is_worker_process = True
-                tasks.append(receiver.listen())
+                tasks.append(receiver.listen(shutdown_event))
 
             await asyncio.gather(*tasks)
     finally: