Added populate_existing and skip_locked for with_for_update.

mobiusml · Oct 22, 2024 · 836c21c · 836c21c
1 parent 1e4f384
commit 836c21c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/aana/storage/repository/task.py b/aana/storage/repository/task.py
@@ -92,7 +92,8 @@ def fetch_unprocessed_tasks(self, limit: int | None = None) -> list[TaskEntity]:
             )
             .order_by(desc(TaskEntity.priority), TaskEntity.created_at)
             .limit(limit)
-            .with_for_update()
+            .populate_existing()
+            .with_for_update(skip_locked=True)
             .all()
         )
         for task in tasks:
@@ -214,7 +215,8 @@ def update_expired_tasks(
                     TaskEntity.updated_at <= cutoff_time,
                 ),
             )
-            .with_for_update()
+            .populate_existing()
+            .with_for_update(skip_locked=True)
             .all()
         )
         for task in tasks: