Support Reth DB migration (#2551)

Author: yorickdowne

ethd

@@ -1374,6 +1374,34 @@ repair-reth() {
 }
 
 
+migrate-reth() {
+  local yn
+
+# Check for Reth
+  __value="${COMPOSE_FILE}"
+  if [[ ! "${__value}" =~ reth\.yml ]]; then
+    echo "You do not appear to be using Reth. Aborting"
+    return 0
+  fi
+
+  if [[ "${__non_interactive}" -eq 0 ]]; then
+    echo
+    while true; do
+      read -rp "WARNING - About to migrate the Reth database to v2. This can take up to 24 hours for an archive node. Do you wish to continue? (Y/n) " yn
+      case "${yn}" in
+        [Nn]o|[Nn]) echo "Skipping Reth database migration"; return 0;;
+        *) break;;
+      esac
+    done
+  fi
+
+  __docompose exec execution touch /var/lib/reth/migrate-v2
+  echo "Restarting Reth to initiate migration"
+  echo "You can monitor progress with \"./ethd logs -f execution\"."
+  restart execution
+}
+
+
 __detect_v6() {
   local v6_works
 
@@ -5980,6 +6008,8 @@ __full_help() {
   echo "    stops the Lighthouse consensus client and prunes state."
   echo "  repair-reth"
   echo "    stops the Reth execution client and repairs its database."
+  echo "  migrate-reth"
+  echo "    stops the Reth execution client and migrates its database to v2."
   echo "  resync-execution"
   echo "    removes the execution layer database and forces a resync."
   echo "  resync-consensus"
@@ -6135,7 +6165,7 @@ case "${__command}" in
     ;;
   keys|up|start|down|stop|restart|version|logs|prune-nethermind\
 |prune-besu|prune-reth|prune-history|prune-lighthouse|resync-execution|resync-consensus|attach-geth\
-|repair-reth|keyimport|space)
+|repair-reth|migrate-reth|keyimport|space)
     __verify_env_file "${__min_env_version}"
     __get_compose_file
     ${__command} "$@"

reth/docker-entrypoint.sh

@@ -175,12 +175,24 @@ fi
 if [[ -f /var/lib/reth/repair-trie ]]; then
   if [[ "${NETWORK}" =~ ^https?:// ]]; then
     echo "Can't repair database on custom network"
-    rm "var/lib/reth/repair-trie"
+    rm -f /var/lib/reth/repair-trie
   else
-    rm "var/lib/reth/repair-trie"  # Remove first in case this panics
+    rm -f /var/lib/reth/repair-trie  # Remove first in case this panics
     echo "Running Reth database trie repair. This may take up to 2 hours"
 # shellcheck disable=SC2086
-    reth db --chain "${NETWORK}" --datadir /var/lib/reth ${__static} repair-trie
+    exec reth db --chain "${NETWORK}" --datadir /var/lib/reth ${__static} repair-trie
+  fi
+fi
+
+if [[ -f /var/lib/reth/migrate-v2 ]]; then
+  if [[ "${NETWORK}" =~ ^https?:// ]]; then
+    echo "Can't migrate database on custom network"
+    rm -f /var/lib/reth/migrate-v2
+  else
+    rm -f /var/lib/reth/migrate-v2  # Remove first in case this panics
+    echo "Running Reth database migration. This may take up to 24 hours for an archive node."
+# shellcheck disable=SC2086
+    exec reth db --chain "${NETWORK}" --datadir /var/lib/reth ${__static} migrate-v2
   fi
 fi