RepairWorker code_change

bchamagne · Feb 16, 2024 · 8ac7cbb · 8ac7cbb
1 parent 82e340e
commit 8ac7cbb
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/lib/archethic/self_repair/repair_worker.ex b/lib/archethic/self_repair/repair_worker.ex
@@ -7,7 +7,7 @@ defmodule Archethic.SelfRepair.RepairWorker do
   alias Archethic.SelfRepair.NotifierSupervisor
 
   use GenServer, restart: :transient
-  @vsn 1
+  @vsn 2
 
   require Logger
 
@@ -102,6 +102,15 @@ defmodule Archethic.SelfRepair.RepairWorker do
 
   def handle_info(_, data), do: {:noreply, data}
 
+  # add the genesis_address to the state
+  def code_change(1, state, _extra) do
+    [genesis_address] = Registry.keys(RepairRegistry, self())
+    state = Map.put(state, :genesis_address, genesis_address)
+    {:ok, state}
+  end
+
+  def code_change(_version, state, _extra), do: {:ok, state}
+
   defp start_repair(
          data = %{
            storage_addresses: [],