starred/windmill
1
0
Fork 0
mirror of https://github.com/windmill-labs/windmill.git synced 2026-08-22 00:01:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

re-route agent workers job if needed prior to applying concurrency limits

Browse Source
This commit is contained in:
Ruben Fiszel
2025-10-17 17:15:05 +00:00
parent ceada62b5d
commit 4136a42b48
1 changed files with 24 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/windmill-queue/src/jobs.rs
+24 -23
View File
@@ -2330,6 +2330,30 @@ pub async fn pull(
(job, false)
};
if let Some(job) = job.as_ref() {
if job.is_flow() || job.is_dependency() {
let per_workspace = per_workspace_tag(&job.workspace_id).await;
let base_tag = if job.is_flow() {
"flow".to_string()
} else {
"dependency".to_string()
};
let tag = if per_workspace {
format!("{}-{}", base_tag, job.workspace_id)
} else {
base_tag
};
sqlx::query!(
"UPDATE v2_job_queue SET tag = $1, running = false WHERE id = $2",
tag,
job.id
)
.execute(db)
.await?;
continue;
}
}
let pulled_job_result = match job {
#[cfg(feature = "private")]
Some(job)
@@ -2356,29 +2380,6 @@ pub async fn pull(
Ok::<_, Error>(pulled_job_result)
}?;
if let Some(job) = njob.job.as_ref() {
if job.is_flow() || job.is_dependency() {
let per_workspace = per_workspace_tag(&job.workspace_id).await;
let base_tag = if job.is_flow() {
"flow".to_string()
} else {
"dependency".to_string()
};
let tag = if per_workspace {
format!("{}-{}", base_tag, job.workspace_id)
} else {
base_tag
};
sqlx::query!(
"UPDATE v2_job_queue SET tag = $1, running = false WHERE id = $2",
tag,
job.id
)
.execute(db)
.await?;
continue;
}
}
return Ok(njob);
};