From 874ffae11631e46d921919e451b4233529eee937 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?V=C3=ADctor=20Olaya?= <volayaf@gmail.com>
Date: Tue, 3 Sep 2024 12:53:35 +0200
Subject: [PATCH] feat(sf): added warehouse option for SF (#524)

---
 clouds/snowflake/.env.template        | 1 +
 clouds/snowflake/common/run-query.js  | 3 ++-
 clouds/snowflake/common/run-script.js | 3 ++-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/clouds/snowflake/.env.template b/clouds/snowflake/.env.template
index 0bfa4262b..967be5aa0 100644
--- a/clouds/snowflake/.env.template
+++ b/clouds/snowflake/.env.template
@@ -13,3 +13,4 @@ SF_USER=<user>
 SF_PASSWORD=<password>
 SF_ROLE=<role>  # optional
 SF_SHARE=<share>  # optional
+SF_WAREHOUSE=<warehouse>  # optional
diff --git a/clouds/snowflake/common/run-query.js b/clouds/snowflake/common/run-query.js
index e5274e112..f51b92a62 100755
--- a/clouds/snowflake/common/run-query.js
+++ b/clouds/snowflake/common/run-query.js
@@ -8,7 +8,8 @@ const connection = snowflake.createConnection({
     account: process.env.SF_ACCOUNT,
     username: process.env.SF_USER,
     password: process.env.SF_PASSWORD,
-    role: process.env.SF_ROLE
+    role: process.env.SF_ROLE,
+    warehouse: process.env.SF_WAREHOUSE,
 });
 
 connection.connect((err) => {
diff --git a/clouds/snowflake/common/run-script.js b/clouds/snowflake/common/run-script.js
index 902808742..d3203d4ec 100755
--- a/clouds/snowflake/common/run-script.js
+++ b/clouds/snowflake/common/run-script.js
@@ -18,7 +18,8 @@ const connection = snowflake.createConnection({
     account: process.env.SF_ACCOUNT,
     username: process.env.SF_USER,
     password: process.env.SF_PASSWORD,
-    role: process.env.SF_ROLE
+    role: process.env.SF_ROLE,
+    warehouse: process.env.SF_WAREHOUSE,
 });
 
 connection.connect((err) => {