Done

anecdotes-ai · Sep 26, 2021 · 0dd474b · 0dd474b
1 parent 346b92f
commit 0dd474b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/action.yml b/action.yml
@@ -25,6 +25,9 @@ inputs:
   queries:
     description: 'List of SQL queries, seperated by ;'
     required: true
+  sync:
+    description: 'Whether to run the queries sync (default is async)'
+    required: false
 
 outputs:
   queries_results:

diff --git a/main.py b/main.py
@@ -44,7 +44,7 @@ def main():
                 print("### Running query ###")
                 print(f"[!] Query id - {query_result.query_id}")
                 print(f"[!] Running query ### - {query}")
-                json_results[query_result.query_id] = query_result.fetch_all()
+                json_results[query_result.query_id] = query_result.fetch_results_sync()
 
     utils.set_github_action_output('queries_results', json.dumps(json_results))
 

diff --git a/snowflake_connector.py b/snowflake_connector.py
@@ -64,4 +64,8 @@ async def fetch_results(self):
         while self.is_query_running():
             await asyncio.sleep(0.1)
 
-        return self._fetch_results()
+        return self._fetch_results()
+
+    def fetch_results_sync(self):
+        self.cursor.get_results_from_sfqid(self.query_id)
+        return self.cursor.fetchall()