Select for update with of=VolumeModel

r4victor · r4victor · commit 2324aedaa6f1 · 2025-07-24T16:17:08.000+05:00
diff --git a/src/dstack/_internal/server/background/tasks/process_submitted_jobs.py b/src/dstack/_internal/server/background/tasks/process_submitted_jobs.py
@@ -359,7 +359,7 @@ async def _process_submitted_job(session: AsyncSession, job_model: JobModel):
         .where(VolumeModel.id.in_(volumes_ids))
         .options(joinedload(VolumeModel.user).load_only(UserModel.name))
         .order_by(VolumeModel.id)  # take locks in order
-        .with_for_update(key_share=True)
+        .with_for_update(key_share=True, of=VolumeModel)
     )
     async with get_locker(get_db().dialect_name).lock_ctx(VolumeModel.__tablename__, volumes_ids):
         if len(volume_models) > 0:

Original file line number	Diff line number	Diff line change
`@@ -359,7 +359,7 @@ async def _process_submitted_job(session: AsyncSession, job_model: JobModel):`
`359`	`359`	`.where(VolumeModel.id.in_(volumes_ids))`
`360`	`360`	`.options(joinedload(VolumeModel.user).load_only(UserModel.name))`
`361`	`361`	`.order_by(VolumeModel.id) # take locks in order`
`362`		`- .with_for_update(key_share=True)`
	`362`	`+ .with_for_update(key_share=True, of=VolumeModel)`
`363`	`363`	`)`
`364`	`364`	`async with get_locker(get_db().dialect_name).lock_ctx(VolumeModel.__tablename__, volumes_ids):`
`365`	`365`	`if len(volume_models) > 0:`