staterecovery: disable synch and maining services again on afterExter…

…nalServicePostMainLoop callback
Consensys · Jan 28, 2025 · 11e3b15 · 11e3b15
1 parent 4753f40
commit 11e3b15
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/...covery/besu-plugin/src/main/kotlin/linea/staterecovery/plugin/LineaStateRecoveryPlugin.kt b/...covery/besu-plugin/src/main/kotlin/linea/staterecovery/plugin/LineaStateRecoveryPlugin.kt
@@ -113,6 +113,9 @@ open class LineaStateRecoveryPlugin : BesuPlugin {
   }
 
   override fun afterExternalServicePostMainLoop() {
+    // we need to recall this again because Sync and Mining services
+    // bay have been started after the plugin start
+    this.recoveryModeManager.enableRecoveryModeIfNecessary()
   }
 
   override fun stop() {

diff --git a/state-recovery/besu-plugin/src/main/kotlin/linea/staterecovery/plugin/RecoveryModeManager.kt b/state-recovery/besu-plugin/src/main/kotlin/linea/staterecovery/plugin/RecoveryModeManager.kt
@@ -27,6 +27,11 @@ class RecoveryModeManager(
 
   init {
     log.info("RecoveryModeManager initializing: headBlockNumber={}", headBlockNumber)
+    enableRecoveryModeIfNecessary()
+  }
+
+  @Synchronized
+  fun enableRecoveryModeIfNecessary() {
     if (hasReachedTargetBlock()) {
       log.info(
         "enabling recovery mode immediately at blockNumber={} recoveryTargetBlockNumber={}",
@@ -92,9 +97,6 @@ class RecoveryModeManager(
 
   /** Switches the node to recovery mode.  */
   private fun switchToRecoveryMode() {
-    check(!recoveryModeTriggered.get()) {
-      "cannot enable already enabled recovery mode"
-    }
     log.warn("Stopping synchronization service")
     synchronizationService.stop()
 
@@ -105,7 +107,7 @@ class RecoveryModeManager(
     miningService.stop()
 
     log.info(
-      "Switched to state recovery mode at block={}",
+      "switched to state recovery mode at block={}",
       headBlockNumber
     )
     recoveryModeTriggered.set(true)