Set a default timezone on mix ecto.create for PG

Author: josevalim

integration_test/pg/storage_test.exs

@@ -78,6 +78,22 @@ defmodule Ecto.Integration.StorageTest do
     drop_database()
   end
 
+  test "storage up sets timezone to UTC by default" do
+    assert Postgres.storage_up(params()) == :ok
+    {output, _} = run_psql("SHOW timezone;", [params()[:database]])
+    assert output =~ "Etc/UTC"
+  after
+    drop_database()
+  end
+
+  test "storage up with custom timestamp timezone" do
+    assert Postgres.storage_up([timestamp: "America/New_York"] ++ params()) == :ok
+    {output, _} = run_psql("SHOW timezone;", [params()[:database]])
+    assert output =~ "America/New_York"
+  after
+    drop_database()
+  end
+
   test "storage down (twice in a row)" do
     create_database()
     assert Postgres.storage_down(params()) == :ok

lib/ecto/adapters/postgres.ex

@@ -228,7 +228,7 @@ defmodule Ecto.Adapters.Postgres do
 
         case run_query(create_command, opts) do
           {:ok, _} ->
-            :ok
+            storage_up_timestamp(database, opts)
 
           {:error, %{postgres: %{code: :duplicate_database}}} ->
             {:error, :already_up}
@@ -239,6 +239,21 @@ defmodule Ecto.Adapters.Postgres do
     end
   end
 
+  defp storage_up_timestamp(database, opts) do
+    timestamp = Keyword.get(opts, :timestamp, "Etc/UTC")
+
+    if timestamp do
+      alter_command = ~s(ALTER DATABASE "#{database}" SET timezone TO '#{timestamp}')
+
+      case run_query(alter_command, opts) do
+        {:ok, _} -> :ok
+        {:error, error} -> {:error, Exception.message(error)}
+      end
+    else
+      :ok
+    end
+  end
+
   defp concat_if(content, nil, _), do: content
   defp concat_if(content, false, _), do: content
   defp concat_if(content, value, fun), do: content <> " " <> fun.(value)