feat: use tick thread to retrieve current time

killme2008 · Jun 8, 2024 · 63990cf · 63990cf
1 parent a147bc6
commit 63990cf
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 4 deletions.
diff --git a/src/main/java/com/googlecode/aviator/runtime/RuntimeUtils.java b/src/main/java/com/googlecode/aviator/runtime/RuntimeUtils.java
@@ -3,7 +3,6 @@
 import java.io.IOException;
 import java.math.MathContext;
 import java.util.Map;
-import java.util.concurrent.TimeUnit;
 import com.googlecode.aviator.AviatorEvaluator;
 import com.googlecode.aviator.AviatorEvaluatorInstance;
 import com.googlecode.aviator.Options;
@@ -31,7 +30,8 @@
  */
 public final class RuntimeUtils {
 
-  private static final int CHECKPOINTS = 3000;
+  private static final int CHECKPOINTS =
+      Integer.parseInt(System.getProperty("aviator.execution.timeout.checkpoints", "2000"));
 
   private RuntimeUtils() {
 

diff --git a/src/main/java/com/googlecode/aviator/utils/Utils.java b/src/main/java/com/googlecode/aviator/utils/Utils.java
@@ -8,6 +8,10 @@
 import java.security.MessageDigest;
 import java.util.Map;
 import java.util.Properties;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.ThreadFactory;
+import java.util.concurrent.TimeUnit;
 import com.googlecode.aviator.runtime.RuntimeUtils;
 import com.googlecode.aviator.runtime.type.AviatorBigInt;
 import com.googlecode.aviator.runtime.type.AviatorDecimal;
@@ -22,15 +26,43 @@
  *
  */
 public class Utils {
-  private static final String CURRENT_VERSION = "5.0.0";
+  private static final String CURRENT_VERSION = "5.4.2";
 
+  private static final int TICK_INTERVAL_MS =
+      Integer.parseInt(System.getProperty("aviator.tick.interval_ms", "1"));
 
   private Utils() {
 
   }
 
+  private static class StaticHolder {
+    static ScheduledExecutorService tickService =
+        Executors.newSingleThreadScheduledExecutor(new ThreadFactory() {
+
+          @Override
+          public Thread newThread(Runnable r) {
+            Thread thread = new Thread(r);
+            thread.setDaemon(true);
+            thread.setName("aviatorscript-tick-thread");
+            return thread;
+          }
+        });
+
+    static volatile long nowNs = System.nanoTime();
+
+    static {
+      tickService.scheduleAtFixedRate(new Runnable() {
+
+        @Override
+        public void run() {
+          nowNs = System.nanoTime();
+        }
+      }, 0, TICK_INTERVAL_MS, TimeUnit.MILLISECONDS);
+    }
+  }
+
   public static long currentTimeNanos() {
-    return System.nanoTime();
+    return StaticHolder.nowNs;
   }
 
   private static final ThreadLocal<MessageDigest> MESSAGE_DIGEST_LOCAL =