add code_change/4 to add the missing fields in the state

archethic-foundation · Dec 19, 2023 · fe9ae2a · fe9ae2a
1 parent 2fbba9e
commit fe9ae2a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lib/archethic/p2p/client/connection.ex b/lib/archethic/p2p/client/connection.ex
@@ -624,6 +624,16 @@ defmodule Archethic.P2P.Client.Connection do
     :ets.delete(@table_name, node_public_key)
   end
 
+  def code_change("1.4.2", state, data, _extra) do
+    {:ok, state,
+     data
+     |> Map.merge(%{
+       reconnect_attempts: 0,
+       heartbeats_sent: 0,
+       heartbeats_received: 0
+     })}
+  end
+
   def code_change(_old_vsn, state, data, _extra), do: {:ok, state, data}
 
   defp backoff(attempts) do