From b656236af723779b15b692678ff9cb2330e412ca Mon Sep 17 00:00:00 2001
From: wendrul <53628737+wendrul@users.noreply.github.com>
Date: Tue, 14 May 2024 22:14:12 +0200
Subject: [PATCH] Use userDB instead of DB (#3723)

---
 backend/windmill-api/src/jobs.rs | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/backend/windmill-api/src/jobs.rs b/backend/windmill-api/src/jobs.rs
index 1cd39b4091..9f91f27a30 100644
--- a/backend/windmill-api/src/jobs.rs
+++ b/backend/windmill-api/src/jobs.rs
@@ -1031,7 +1031,8 @@ struct ListableQueuedJob {
 }
 
 async fn list_queue_jobs(
-    Extension(db): Extension<DB>,
+    authed: ApiAuthed,
+    Extension(user_db): Extension<UserDB>,
     Path(w_id): Path<String>,
     Query(lq): Query<ListQueueQuery>,
 ) -> error::JsonResult<Vec<ListableQueuedJob>> {
@@ -1061,9 +1062,11 @@ async fn list_queue_jobs(
         ],
     )
     .sql()?;
+    let mut tx = user_db.begin(&authed).await?;
     let jobs = sqlx::query_as::<_, ListableQueuedJob>(&sql)
-        .fetch_all(&db)
+        .fetch_all(&mut *tx)
         .await?;
+    tx.commit().await?;
     Ok(Json(jobs))
 }
 
@@ -3954,7 +3957,7 @@ pub struct ListCompletedQuery {
 
 async fn list_completed_jobs(
     authed: ApiAuthed,
-    Extension(db): Extension<DB>,
+    Extension(user_db): Extension<UserDB>,
     Path(w_id): Path<String>,
     Query(pagination): Query<Pagination>,
     Query(lq): Query<ListCompletedQuery>,
@@ -4002,9 +4005,11 @@ async fn list_completed_jobs(
         ],
     )
     .sql()?;
+    let mut tx = user_db.begin(&authed).await?;
     let jobs = sqlx::query_as::<_, ListableCompletedJob>(&sql)
-        .fetch_all(&db)
+        .fetch_all(&mut *tx)
         .await?;
+    tx.commit().await?;
     Ok(Json(jobs))
 }