From 3bd461ffa1fcbe19c9007b1b15970be21cff08ef Mon Sep 17 00:00:00 2001
From: HugoCasa <hugo@casademont.ch>
Date: Wed, 14 Aug 2024 14:27:43 +0200
Subject: [PATCH] fix: use job timeout for snowflake timeout (#4240)

---
 backend/windmill-worker/src/snowflake_executor.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/backend/windmill-worker/src/snowflake_executor.rs b/backend/windmill-worker/src/snowflake_executor.rs
index ad0c77c0fb..aac2210356 100644
--- a/backend/windmill-worker/src/snowflake_executor.rs
+++ b/backend/windmill-worker/src/snowflake_executor.rs
@@ -17,7 +17,7 @@ use windmill_queue::{CanceledBy, HTTP_CLIENT};
 
 use serde::{Deserialize, Serialize};
 
-use crate::common::run_future_with_polling_update_job_poller;
+use crate::common::{resolve_job_timeout, run_future_with_polling_update_job_poller};
 use crate::{common::build_args_values, AuthedClientBackgroundTask};
 
 #[derive(Serialize)]
@@ -304,7 +304,11 @@ pub async fn do_snowflake(
             json!(database.database.unwrap().to_uppercase()),
         );
     }
-    body.insert("timeout".to_string(), json!(10)); // in seconds
+    let timeout = resolve_job_timeout(&db, &job.workspace_id, job.id, job.timeout)
+        .await
+        .0
+        .as_secs();
+    body.insert("timeout".to_string(), json!(timeout));
 
     let queries = parse_sql_blocks(query);