monitor handled tasks

Author: caoli5288

src/main/java/com/mengcraft/script/HandledTask.java

@@ -1,9 +1,11 @@
 package com.mengcraft.script;
 
+import com.mengcraft.script.util.Utils;
 import lombok.AccessLevel;
 import lombok.EqualsAndHashCode;
 import lombok.RequiredArgsConstructor;
 
+import javax.script.Bindings;
 import java.util.UUID;
 
 /**
@@ -16,7 +18,7 @@ public class HandledTask implements Runnable {
     private final UUID uniqueId = UUID.randomUUID();
 
     private final ScriptPlugin plugin;
-    private final Runnable run;
+    private final Bindings invokableObj;
     private final int period;
 
     private Runnable complete;
@@ -32,9 +34,14 @@ public void setId(int id) {
 
     @Override
     public void run() {
+        long clock = System.currentTimeMillis();
         try {
-            run.run();
+            Utils.invoke(invokableObj);
         } finally {
+            clock = System.currentTimeMillis() - clock;
+            if (clock >= 5) {
+                plugin.getLogger().warning(String.format("Tasks consume too much time. (%s millis)\n%s", clock, invokableObj.toString()));
+            }
             if (period < 1) complete();
         }
     }

src/main/java/com/mengcraft/script/ScriptBootstrap.java

@@ -8,6 +8,7 @@
 import com.mengcraft.script.plugin.ScriptingLoader;
 import com.mengcraft.script.util.BossBarWrapper;
 import com.mengcraft.script.util.Named;
+import com.mengcraft.script.util.Utils;
 import lombok.RequiredArgsConstructor;
 import lombok.SneakyThrows;
 import lombok.experimental.var;
@@ -85,6 +86,7 @@ public static ScriptEngine jsEngine() {
     public void onLoad() {
         plugin = this;
         jsEngine = new ScriptEngineManager(getClassLoader()).getEngineByName("nashorn");
+        Utils.setup(jsEngine);
     }
 
     @Override

src/main/java/com/mengcraft/script/ScriptPlugin.java

@@ -141,33 +141,33 @@ public boolean depend(List<String> depend, Runnable runnable) {
         return call.call();
     }
 
-    public HandledTask runTask(Runnable runnable) {
-        return runTask(runnable, 0, -1, false);
+    public HandledTask runTask(Bindings invokable) {
+        return runTask(invokable, 0, -1, false);
     }
 
-    public HandledTask runTask(@NonNull Runnable runnable, int delay, int period, boolean b) {
+    public HandledTask runTask(@NonNull Bindings invokable, int delay, int period, boolean b) {
         Preconditions.checkState(isHandled(), "unloaded");
-        HandledTask task = new HandledTask(this, runnable, period);
+        HandledTask task = new HandledTask(this, invokable, period);
         tasks.add(task);
         task.setId(b ? Bukkit.getScheduler().runTaskTimerAsynchronously(main, task, delay, period).getTaskId()
                 : Bukkit.getScheduler().runTaskTimer(main, task, delay, period).getTaskId());
         return task;
     }
 
-    public HandledTask runTask(Runnable runnable, int delay, int period) {
-        return runTask(runnable, delay, period, false);
+    public HandledTask runTask(Bindings invokable, int delay, int period) {
+        return runTask(invokable, delay, period, false);
     }
 
-    public HandledTask runTask(Runnable runnable, int delay, boolean b) {
-        return runTask(runnable, delay, -1, b);
+    public HandledTask runTask(Bindings invokable, int delay, boolean b) {
+        return runTask(invokable, delay, -1, b);
     }
 
-    public HandledTask runTask(Runnable runnable, int delay) {
-        return runTask(runnable, delay, -1, false);
+    public HandledTask runTask(Bindings invokable, int delay) {
+        return runTask(invokable, delay, -1, false);
     }
 
-    public HandledTask runTask(Runnable runnable, boolean b) {
-        return runTask(runnable, 0, -1, b);
+    public HandledTask runTask(Bindings invokable, boolean b) {
+        return runTask(invokable, 0, -1, b);
     }
 
     boolean remove(HandledCommand i) {

src/main/java/com/mengcraft/script/util/Utils.java

@@ -2,21 +2,23 @@
 
 import com.mengcraft.script.ScriptBootstrap;
 import lombok.SneakyThrows;
-import lombok.experimental.var;
 import org.bukkit.entity.Player;
 import org.bukkit.event.EventPriority;
 import org.yaml.snakeyaml.Yaml;
 
 import javax.script.Bindings;
 import javax.script.Invocable;
+import javax.script.ScriptEngine;
 import java.math.BigDecimal;
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.function.Consumer;
 import java.util.function.Function;
 
 public class Utils {
 
     private static final Yaml YAML = new Yaml();
     private static Function<Object, Bindings> java_from_invoker;
+    private static Consumer<Bindings> function_invoker;
 
     public static Yaml getYaml() {
         return YAML;
@@ -40,13 +42,22 @@ public static void sendBossBar(Player p, BossBarWrapper bar, int tick) {
         });
     }
 
+    @SneakyThrows
+    public static void setup(ScriptEngine js) {
+        Bindings bindings = js.createBindings();
+        js.eval("function apply(a){return Java.from(a);}", bindings);
+        java_from_invoker = ((Invocable) js).getInterface(bindings, Function.class);
+        bindings = js.createBindings();
+        js.eval("function accept(a){a();}", bindings);
+        function_invoker = ((Invocable) js).getInterface(bindings, Consumer.class);
+    }
+
+    public static void invoke(Bindings bindings) {
+        function_invoker.accept(bindings);
+    }
+
     @SneakyThrows
     public static <T> Bindings fromJava(T obj) {
-        if (java_from_invoker == null) {
-            var js = ScriptBootstrap.jsEngine();
-            var function = js.eval("(function a(a){return Java.from(a);})");
-            java_from_invoker = ((Invocable) js).getInterface(function, Function.class);
-        }
         return java_from_invoker.apply(obj);
     }