fix: enhance task cancellation logic to prevent errors on closed tasks

Thymis-io · Jan 20, 2025 · d97ae5d · d97ae5d
1 parent 2689a1f
commit d97ae5d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/controller/thymis_controller/task/executor.py b/controller/thymis_controller/task/executor.py
@@ -97,7 +97,11 @@ def cancel_task(self, task_id: uuid.UUID):
     def cancel_all_tasks(self):
         future_tasks = list(self.futures.keys())
         for task_id in future_tasks:
-            if task_id in self.futures and self.futures[task_id][0].running():
+            if (
+                task_id in self.futures
+                and self.futures[task_id][0].running()
+                and not self.futures[task_id][1].closed
+            ):
                 self.cancel_task(task_id)
 
     def listen_child_messages(self, conn: Connection, task_id: uuid.UUID):