From 8ec7b9069ef2071552764ed38d71f13af417caea Mon Sep 17 00:00:00 2001
From: Soaibuzzaman <soaib.safi@gmail.com>
Date: Fri, 26 Jul 2024 14:11:40 +0200
Subject: [PATCH] Updated time out for alloy

---
 .../alloy/api/AlloyInstanceController.java    | 51 +++++++++++--------
 1 file changed, 30 insertions(+), 21 deletions(-)

diff --git a/alloy-api/src/main/java/de/buw/fmp/alloy/api/AlloyInstanceController.java b/alloy-api/src/main/java/de/buw/fmp/alloy/api/AlloyInstanceController.java
index 6db15b0..4f6d452 100644
--- a/alloy-api/src/main/java/de/buw/fmp/alloy/api/AlloyInstanceController.java
+++ b/alloy-api/src/main/java/de/buw/fmp/alloy/api/AlloyInstanceController.java
@@ -21,12 +21,7 @@
 
 import java.util.LinkedHashMap;
 import java.util.Map;
-import java.util.concurrent.Callable;
 import java.util.concurrent.ExecutionException;
-import java.util.concurrent.Executors;
-import java.util.concurrent.Future;
-import java.util.concurrent.ScheduledExecutorService;
-import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 
 import org.json.JSONObject;
@@ -77,8 +72,13 @@ public String getInstance(@RequestBody InstanceRequest instanceRequest, @PathVar
         A4Solution instance;
         SafeList<Sig> sigs = module.getAllSigs();
         try {
-            final Command finalRunCommand = runCommand;
-            instance = runTimed(() -> TranslateAlloyToKodkod.execute_command(A4Reporter.NOP, sigs, finalRunCommand, options), TIME_OUT);            
+            Command finalRunCommand = runCommand;
+            instance = runTimed(new InstanceRunner() {
+                @Override
+                public A4Solution runInstance() {
+                    return TranslateAlloyToKodkod.execute_command(A4Reporter.NOP, sigs, finalRunCommand, options);
+                }
+            }, TIME_OUT);
         } catch (Exception e) {
             // return error message as JSON with http status code 400
             JSONObject obj = new JSONObject();
@@ -148,7 +148,12 @@ public String getNextInstance(@RequestBody String specId) throws IOException {
 
         try {
             final A4Solution finalInstance = instance;
-            instance = runTimed(() -> finalInstance.next(), TIME_OUT);
+            instance = runTimed(new InstanceRunner() {
+                @Override
+                public A4Solution runInstance() {
+                    return finalInstance.next();
+                }
+            }, TIME_OUT);
         } catch (Exception e) {
             JSONObject obj = new JSONObject();
             String message = e.getMessage();
@@ -190,22 +195,26 @@ public void removeOlsInstances() {
         instances.entrySet().removeIf(entry -> currentTime - entry.getValue().getLastAccessed() > 3600000);
     }
 
-    public A4Solution runTimed(Callable<A4Solution> callable, int seconds) throws Exception {        
-        ScheduledExecutorService executor = Executors.newSingleThreadScheduledExecutor();
-        Future<A4Solution> task = executor.submit(callable);
-        A4Solution instance = null;
-        try {
-            instance = task.get(seconds, TimeUnit.SECONDS);
-        } catch(InterruptedException | TimeoutException e) {
-            task.cancel(true);
-            throw e;
-        } catch (ExecutionException e) {
-            throw e;
+    public A4Solution runTimed(InstanceRunner r, int seconds) throws InterruptedException, ExecutionException, TimeoutException {
+        Thread t = new Thread(r);
+        t.start();
+        Thread.sleep(seconds * 1000);
+        t.stop();
+        if (r.instance == null) {
+            throw new TimeoutException("Analysis timed out after " + seconds + " seconds.");
         }
+        return r.instance;
+    }
 
-        executor.shutdown();
+    private abstract class InstanceRunner implements Runnable {
+        private A4Solution instance;
 
-        return instance;
+        public abstract A4Solution runInstance();
+
+        @Override
+        public void run() {
+            instance = runInstance();        
+        }
     }
     
 }