Suppress query logs in mix ecto.load when quiet flag is given. (#503)

Author: greg-rychlewski

lib/ecto/adapters/sql.ex

@@ -566,11 +566,11 @@ defmodule Ecto.Adapters.SQL do
   Returns `true` if the `table` exists in the `repo`, otherwise `false`.
   The table is checked against the current database/schema in the connection.
   """
-  @spec table_exists?(Ecto.Repo.t, table :: String.t) :: boolean
-  def table_exists?(repo, table) when is_atom(repo) do
+  @spec table_exists?(Ecto.Repo.t, table :: String.t, opts :: Keyword.t) :: boolean
+  def table_exists?(repo, table, opts \\ []) when is_atom(repo) do
     %{sql: sql} = adapter_meta = Ecto.Adapter.lookup_meta(repo)
     {query, params} = sql.table_exists_query(table)
-    query!(adapter_meta, query, params, []).num_rows != 0
+    query!(adapter_meta, query, params, opts).num_rows != 0
   end
 
   # Returns a formatted table for a given query `result`.

lib/mix/tasks/ecto.load.ex

@@ -56,9 +56,10 @@ defmodule Mix.Tasks.Ecto.Load do
   """
 
   @impl true
-  def run(args, table_exists? \\ &Ecto.Adapters.SQL.table_exists?/2) do
+  def run(args, table_exists? \\ &Ecto.Adapters.SQL.table_exists?/3) do
     {opts, _} = OptionParser.parse!(args, strict: @switches, aliases: @aliases)
     opts = Keyword.merge(@default_opts, opts)
+    opts = if opts[:quiet], do: Keyword.put(opts, :log, false), else: opts
 
     Enum.each(parse_repo(args), fn repo ->
       ensure_repo(repo, args)
@@ -70,7 +71,7 @@ defmodule Mix.Tasks.Ecto.Load do
       )
 
       {migration_repo, source} = Ecto.Migration.SchemaMigration.get_repo_and_source(repo, repo.config())
-      {:ok, loaded?, _} = Ecto.Migrator.with_repo(migration_repo, &table_exists?.(&1, source))
+      {:ok, loaded?, _} = Ecto.Migrator.with_repo(migration_repo, table_exists_closure(table_exists?, source, opts))
 
       for repo <- Enum.uniq([repo, migration_repo]) do
         cond do
@@ -87,6 +88,14 @@ defmodule Mix.Tasks.Ecto.Load do
     end)
   end
 
+  defp table_exists_closure(fun, source, opts) when is_function(fun, 3) do
+    &fun.(&1, source, opts)
+  end
+
+  defp table_exists_closure(fun, source, _opts) when is_function(fun, 2) do
+    &fun.(&1, source)
+  end
+
   defp skip_safety_warnings? do
     Mix.Project.config()[:start_permanent] != true
   end