Fixed bad logical bug from Acorn's MainState.load rollback

[simantics/platform.git] / bundles / org.simantics.acorn / src / org / simantics / acorn / MainState.java

diff --git a/bundles/org.simantics.acorn/src/org/simantics/acorn/MainState.java b/bundles/org.simantics.acorn/src/org/simantics/acorn/MainState.java
index 63fc5d4937b06cd0aff10b571506e66830741535..ebc4079d51c64ccf1c2b40e372b89c3602e399cf 100644 (file)
--- a/bundles/org.simantics.acorn/src/org/simantics/acorn/MainState.java
+++ b/bundles/org.simantics.acorn/src/org/simantics/acorn/MainState.java
@@ -15,7 +15,6 @@ import java.util.function.Predicate;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
-import org.simantics.acorn.exception.InvalidHeadStateException;
 import org.simantics.databoard.Bindings;
 import org.simantics.databoard.binding.mutable.MutableVariant;
 import org.simantics.databoard.util.binary.BinaryMemory;
@@ -49,10 +48,10 @@ public class MainState implements Serializable {
                     Bindings.getBindingUnchecked(MainState.class));
             int latestRevision = state.headDir - 1;
             try {
-                HeadState.validateHeadStateIntegrity(directory.resolve(latestRevision + "/" + HeadState.HEAD_STATE));
-                archiveRevisionDirectories(directory, latestRevision, rollbackCallback);
-                return state;
-            } catch (InvalidHeadStateException e) {
+                if (HeadState.validateHeadStateIntegrity(directory.resolve(latestRevision + "/" + HeadState.HEAD_STATE))) {
+                    archiveRevisionDirectories(directory, latestRevision, rollbackCallback);
+                    return state;
+                }
                 LOGGER.warn("Failed to start database from revision " + latestRevision + " stored in " + mainState + ". " + HeadState.HEAD_STATE + " is invalid.");
                 return rollback(directory, rollbackCallback);
             } catch (FileNotFoundException e) {
@@ -61,7 +60,7 @@ public class MainState implements Serializable {
             }
         } catch (FileNotFoundException e) {
             // The database may also be totally empty at this point
-            if (!listRevisionDirs(directory, true, MainState::isInteger).isEmpty())
+            if (listRevisionDirs(directory, true, MainState::isInteger).isEmpty())
                 return new MainState(0);
 
             LOGGER.warn("Unclean exit detected, " + mainState + " not found. Initiating automatic rollback.");
@@ -77,7 +76,7 @@ public class MainState implements Serializable {
     private static MainState rollback(Path directory, Runnable rollbackCallback) throws IOException {
         LOGGER.warn("Database rollback initiated for " + directory);
         rollbackCallback.run();
-        Path latest = findNewHeadStateDir(directory, rollbackCallback);
+        Path latest = findNewHeadStateDir(directory);
         int latestRevision = latest != null ? safeParseInt(-1, latest) : -1;
         // +1 because we want to return the next head version to use,
         // not the latest existing version.
@@ -126,16 +125,10 @@ public class MainState implements Serializable {
      * @return
      * @throws IOException
      */
-    private static Path findNewHeadStateDir(Path directory, Runnable rollbackCallback) throws IOException {
-        for (Path last : listRevisionDirs(directory, true, MainState::isInteger)) {
-            try {
-                HeadState.validateHeadStateIntegrity(last.resolve(HeadState.HEAD_STATE));
+    private static Path findNewHeadStateDir(Path directory) throws IOException {
+        for (Path last : listRevisionDirs(directory, true, MainState::isInteger))
+            if (HeadState.validateHeadStateIntegrity(last.resolve(HeadState.HEAD_STATE)))
                 return last;
-            } catch (IOException | InvalidHeadStateException e) {
-                // Cleanup is done in {@link cleanRevisionDirectories} method
-                rollbackCallback.run();
-            }
-        }
         return null;
     }