RATIS-2111 Reinitialize should load the latest snapshot (#1111)

apache · Jun 16, 2024 · de2e777 · de2e777
1 parent e7e0c79
commit de2e777
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/...s-examples/src/main/java/org/apache/ratis/examples/arithmetic/ArithmeticStateMachine.java b/...s-examples/src/main/java/org/apache/ratis/examples/arithmetic/ArithmeticStateMachine.java
@@ -81,7 +81,7 @@ public void initialize(RaftServer server, RaftGroupId groupId,
   @Override
   public void reinitialize() throws IOException {
     close();
-    loadSnapshot(storage.getLatestSnapshot());
+    loadSnapshot(storage.loadLatestSnapshot());
   }
 
   @Override

diff --git a/...-examples/src/main/java/org/apache/ratis/examples/counter/server/CounterStateMachine.java b/...-examples/src/main/java/org/apache/ratis/examples/counter/server/CounterStateMachine.java
@@ -138,7 +138,7 @@ public void initialize(RaftServer server, RaftGroupId groupId, RaftStorage raftS
    */
   @Override
   public void reinitialize() throws IOException {
-    load(storage.getLatestSnapshot());
+    load(storage.loadLatestSnapshot());
   }
 
   /**

diff --git a/ratis-server/src/main/java/org/apache/ratis/statemachine/impl/SimpleStateMachineStorage.java b/ratis-server/src/main/java/org/apache/ratis/statemachine/impl/SimpleStateMachineStorage.java
@@ -217,6 +217,10 @@ public SingleFileSnapshotInfo getLatestSnapshot() {
     if (s != null) {
       return s;
     }
+    return loadLatestSnapshot();
+  }
+
+  public SingleFileSnapshotInfo loadLatestSnapshot() {
     final File dir = stateMachineDir;
     if (dir == null) {
       return null;