From c6eb979fce611a2a0e43a95985cd770e6c8a54dd Mon Sep 17 00:00:00 2001
From: Dustin Farris <dustin.farris@pepsico.com>
Date: Tue, 24 Jan 2023 08:39:39 -1000
Subject: [PATCH] feat(ecto-adapter): implement `query_many/4`

This enables a multi-statement query, e.g. "select 1; select 2".
---
 lib/snowflex/connection.ex      |  3 +++
 lib/snowflex/ecto/connection.ex | 21 +++++++++++++++++++--
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/lib/snowflex/connection.ex b/lib/snowflex/connection.ex
index 784e0f5..398ec42 100644
--- a/lib/snowflex/connection.ex
+++ b/lib/snowflex/connection.ex
@@ -223,6 +223,9 @@ defmodule Snowflex.Connection do
     %Result{num_rows: count, success: true, statement: query.statement}
   end
 
+  defp parse_result(results, query) when is_list(results),
+    do: Enum.map(results, fn result -> parse_result(result, query) end)
+
   defp parse_result(result, _query), do: result
 
   defp parse_result(columns, rows, query) do
diff --git a/lib/snowflex/ecto/connection.ex b/lib/snowflex/ecto/connection.ex
index 26416e7..c6db7d2 100644
--- a/lib/snowflex/ecto/connection.ex
+++ b/lib/snowflex/ecto/connection.ex
@@ -20,9 +20,26 @@ defmodule Snowflex.EctoAdapter.Connection do
     Snowflex.query(conn, sql, params, opts)
   end
 
+  def query_many!(conn, sql, params, opts) do
+    case query_many(conn, sql, params, opts) do
+      {:ok, result} -> result
+      {:error, error} -> raise error
+    end
+  end
+
   @impl true
-  def query_many(_conn, _sql, _params, _opts) do
-    raise "query_many is not supported by Snowflake"
+  @doc """
+  Run a multi-statement query.
+
+  The `MULTI_STATEMENT_COUNT` Snowflake parameter must be set to 0 to allow
+  multi-statement queries.
+
+  https://docs.snowflake.com/en/sql-reference/parameters.html#multi-statement-count
+  """
+  def query_many(conn, sql, params, opts) do
+    with {:ok, result} <- query(conn, sql, params, opts) do
+      {:ok, List.wrap(result)}
+    end
   end
 
   @impl true