Merge pull request #255 from lzaoral/worker-do-not-accept-new-tasks-w…

…hen-locked worker: do not open additional tasks if the `TaskManager` is locked
release-engineering · Mar 25, 2024 · 37ea28d · 37ea28d
2 parents d245e0c + bc1095f
commit 37ea28d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/kobo/worker/taskmanager.py b/kobo/worker/taskmanager.py
@@ -313,6 +313,10 @@ def get_next_task(self):
 
         # process tasks that could be transitioned to the OPEN state
         for task_info in tasks_to_open:
+            # do not take additional tasks
+            if self.locked:
+                return
+
             self.take_task(task_info)
 
     def take_task(self, task_info):