Select for update with of=InstanceModel

r4victor · r4victor · commit 52be0772e326 · 2025-07-24T16:08:23.000+05:00
diff --git a/src/dstack/_internal/server/background/tasks/process_instances.py b/src/dstack/_internal/server/background/tasks/process_instances.py
@@ -160,7 +160,7 @@ async def _process_next_instance():
                 .options(joinedload(InstanceModel.project).load_only(ProjectModel.ssh_private_key))
                 .order_by(InstanceModel.last_processed_at.asc())
                 .limit(1)
-                .with_for_update(skip_locked=True, key_share=True)
+                .with_for_update(skip_locked=True, key_share=True, of=InstanceModel)
             )
             instance = res.scalar()
             if instance is None:

Original file line number	Diff line number	Diff line change
`@@ -160,7 +160,7 @@ async def _process_next_instance():`
`160`	`160`	`.options(joinedload(InstanceModel.project).load_only(ProjectModel.ssh_private_key))`
`161`	`161`	`.order_by(InstanceModel.last_processed_at.asc())`
`162`	`162`	`.limit(1)`
`163`		`- .with_for_update(skip_locked=True, key_share=True)`
	`163`	`+ .with_for_update(skip_locked=True, key_share=True, of=InstanceModel)`
`164`	`164`	`)`
`165`	`165`	`instance = res.scalar()`
`166`	`166`	`if instance is None:`