call disconnect on protocol when reconnecting in simple connection

elixir-ecto · Nov 12, 2024 · 9202615 · 9202615
1 parent 8a2b362
commit 9202615
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/postgrex/simple_connection.ex b/lib/postgrex/simple_connection.ex
@@ -337,7 +337,12 @@ defmodule Postgrex.SimpleConnection do
   @impl :gen_statem
   def handle_event(type, content, statem_state, state)
 
-  def handle_event(:internal, {:connect, _}, @state, %{state: {mod, mod_state}} = state) do
+  def handle_event(:internal, {:connect, :reconnect}, @state, %{protocol: protocol} = state) do
+    Protocol.disconnect(:reconnect, protocol)
+    {:keep_state, %{state | protocol: nil}, {:next_event, :internal, {:connect, :init}}}
+  end
+
+  def handle_event(:internal, {:connect, :init}, @state, %{state: {mod, mod_state}} = state) do
     opts =
       case Keyword.get(opts(mod), :configure) do
         {module, fun, args} -> apply(module, fun, [opts(mod) | args])