Trap exits from connect callback (#662)

elixir-ecto · Oct 14, 2023 · c8edce4 · c8edce4
1 parent cb6bdbc
commit c8edce4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/postgrex/protocol.ex b/lib/postgrex/protocol.ex
@@ -68,6 +68,9 @@ defmodule Postgrex.Protocol do
           {:ok, state}
           | {:error, Postgrex.Error.t() | %DBConnection.ConnectionError{}}
   def connect(opts) do
+    # Trap exits so that DBConnection calls `disconnect` on unexpected shutdowns
+    Process.flag(:trap_exit, true)
+
     endpoints = endpoints(opts)
 
     timeout = opts[:timeout] || @timeout