[PEx] Improve cleanup of disk tasks on exit

Author: aman-goel

Src/PRuntimes/PExplicitRuntime/src/main/java/pexplicit/PExplicit.java

@@ -82,8 +82,6 @@ public static void main(String[] args) {
             }
         } finally {
             // log end-of-run metrics
-            StatWriter.log("result", PExplicitGlobal.getResult());
-            StatWriter.log("status", String.format("%s", PExplicitGlobal.getStatus()));
             StatWriter.log("exit-code", String.format("%d", exit_code));
 
             // exit

Src/PRuntimes/PExplicitRuntime/src/main/java/pexplicit/RuntimeExecutor.java

@@ -75,6 +75,9 @@ private static void printStats() {
                 PExplicitGlobal.setStatus(STATUS.VERIFIED_UPTO_MAX_STEPS);
             }
         }
+
+        StatWriter.log("result", PExplicitGlobal.getResult());
+        StatWriter.log("status", String.format("%s", PExplicitGlobal.getStatus()));
     }
 
     private static void preprocess() {
@@ -136,9 +139,9 @@ private static void process(boolean resume) throws Exception {
             future.cancel(true);
             executor.shutdownNow();
             scheduler.updateResult();
-            SearchTask.Cleanup();
             printStats();
             PExplicitLogger.logEndOfRun(scheduler, Duration.between(TimeMonitor.getStart(), Instant.now()).getSeconds());
+            SearchTask.Cleanup();
         }
     }
 

Src/PRuntimes/PExplicitRuntime/src/main/java/pexplicit/runtime/logger/PExplicitLogger.java

@@ -13,7 +13,6 @@
 import pexplicit.runtime.machine.PMonitor;
 import pexplicit.runtime.machine.State;
 import pexplicit.runtime.machine.events.PContinuation;
-import pexplicit.runtime.scheduler.Schedule;
 import pexplicit.runtime.scheduler.choice.ScheduleSearchUnit;
 import pexplicit.runtime.scheduler.choice.SearchUnit;
 import pexplicit.runtime.scheduler.explicit.ExplicitSearchScheduler;

Src/PRuntimes/PExplicitRuntime/src/main/java/pexplicit/runtime/scheduler/explicit/strategy/SearchStrategy.java

@@ -110,7 +110,7 @@ public int getNumPendingDataChoices() {
         return numUnexplored;
     }
 
-    public abstract void addNewTask(SearchTask task);
-
     abstract SearchTask popNextTask();
+
+    public abstract void addNewTask(SearchTask task);
 }

Src/PRuntimes/PExplicitRuntime/src/main/java/pexplicit/runtime/scheduler/explicit/strategy/SearchTask.java

@@ -1,7 +1,6 @@
 package pexplicit.runtime.scheduler.explicit.strategy;
 
 import lombok.Getter;
-import org.apache.commons.io.FileUtils;
 import pexplicit.runtime.PExplicitGlobal;
 import pexplicit.runtime.logger.PExplicitLogger;
 import pexplicit.runtime.machine.PMachineId;
@@ -54,11 +53,13 @@ public static void Initialize() {
 
     public static void Cleanup() {
         String taskPath = PExplicitGlobal.getConfig().getOutputFolder() + "/tasks/";
-        try {
-            FileUtils.deleteDirectory(new File(taskPath));
-        } catch (IOException e) {
-            throw new RuntimeException("Failed to cleanup tasks at " + taskPath, e);
+        File taskDir = new File(taskPath);
+        String[] entries = taskDir.list();
+        for (String s : entries) {
+            File currentFile = new File(taskDir.getPath(), s);
+            currentFile.delete();
         }
+        taskDir.delete();
     }
 
     public boolean isInitialTask() {