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):