Add jvm memory monitor feature

hugegraph-server/hugegraph-api/src/main/java/org/apache/hugegraph/config/ServerOptions.java

@@ -21,6 +21,7 @@
 import static org.apache.hugegraph.config.OptionChecker.disallowEmpty;
 import static org.apache.hugegraph.config.OptionChecker.nonNegativeInt;
 import static org.apache.hugegraph.config.OptionChecker.positiveInt;
+import static org.apache.hugegraph.config.OptionChecker.rangeDouble;
 import static org.apache.hugegraph.config.OptionChecker.rangeInt;
 
 public class ServerOptions extends OptionHolder {
@@ -321,4 +322,21 @@ public static synchronized ServerOptions instance() {
                     nonNegativeInt(),
                     1000L
             );
+
+    public static final ConfigOption<Double> JVM_MEMORY_MONITOR_THRESHOLD =
+            new ConfigOption<>(
+                    "memory_monitor.threshold",
+                    "Threshold for JVM memory usage monitoring, 1 means disabling the memory " +
+                    "monitoring thread.",
+                    rangeDouble(0.0, 1.0),
+                    0.85
+            );
+
+    public static final ConfigOption<Integer> JVM_MEMORY_MONITOR_PERIOD =
+            new ConfigOption<>(
+                    "memory_monitor.period",
+                    "The period of JVM memory usage monitoring, the unit is milliseconds.",
+                    nonNegativeInt(),
+                    2000
+            );
 }

hugegraph-server/hugegraph-core/src/main/java/org/apache/hugegraph/backend/store/BackendEntryIterator.java

@@ -94,7 +94,7 @@ public Object metadata(String meta, Object... args) {
 
     public static final void checkInterrupted() {
         if (Thread.interrupted()) {
-            throw new BackendException("Interrupted, maybe it is timed out",
+            throw new BackendException("Interrupted, maybe it is timed out or uses too much memory",
                                        new InterruptedException());
         }
     }

hugegraph-server/hugegraph-dist/src/assembly/static/conf/rest-server.properties

@@ -1,7 +1,5 @@
 # bind url
-# could use '0.0.0.0' or specified (real)IP to expose external network access
 restserver.url=http://127.0.0.1:8080
-#restserver.enable_graphspaces_filter=false
 # gremlin server url, need to be consistent with host and port in gremlin-server.yaml
 #gremlinserver.url=http://127.0.0.1:8182
 
@@ -53,3 +51,7 @@ server.role=master
 
 # slow query log
 log.slow_query_threshold=1000
+
+# jvm memory usage monitor
+memory_monitor.threshold=0.8
+memory_monitor.period=1000

hugegraph-server/hugegraph-dist/src/main/java/org/apache/hugegraph/dist/HugeGraphServer.java

@@ -37,6 +37,7 @@ public class HugeGraphServer {
 
     private final RestServer restServer;
     private final GremlinServer gremlinServer;
+    private final MemoryMonitor memoryMonitor;
 
     public static void register() {
         RegisterUtil.registerBackends();
@@ -78,9 +79,15 @@ public HugeGraphServer(String gremlinServerConf, String restServerConf)
         } finally {
             System.setSecurityManager(securityManager);
         }
+
+        // Start Memory Monitor Task
+        this.memoryMonitor = new MemoryMonitor(restServerConf);
+        this.memoryMonitor.start();
     }
 
     public void stop() {
+        this.memoryMonitor.stop();
+
         try {
             this.gremlinServer.stop().get();
             LOG.info("HugeGremlinServer stopped");

hugegraph-server/hugegraph-dist/src/main/java/org/apache/hugegraph/dist/MemoryMonitor.java

@@ -0,0 +1,119 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.hugegraph.dist;
+
+import java.lang.management.ManagementFactory;
+import java.lang.management.MemoryUsage;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+
+import org.apache.hugegraph.config.HugeConfig;
+import org.apache.hugegraph.config.ServerOptions;
+import org.apache.hugegraph.util.Log;
+import org.slf4j.Logger;
+
+import com.sun.management.ThreadMXBean;
+
+public class MemoryMonitor {
+
+    private static final Logger LOG = Log.logger(MemoryMonitor.class);
+    private final double MEMORY_MONITOR_THRESHOLD;
+    private final int MEMORY_MONITOR_PERIOD;
+    private final ScheduledExecutorService scheduler;
+
+    public MemoryMonitor(String restServerConf) {
+        HugeConfig restServerConfig = new HugeConfig(restServerConf);
+        MEMORY_MONITOR_THRESHOLD =
+                restServerConfig.get(ServerOptions.JVM_MEMORY_MONITOR_THRESHOLD);
+        MEMORY_MONITOR_PERIOD =
+                restServerConfig.get(ServerOptions.JVM_MEMORY_MONITOR_PERIOD);
+        scheduler = Executors.newScheduledThreadPool(1);
+    }
+
+    private void manageMemory() {
+        double memoryUsagePercentage = getMemoryUsagePercentage();
+
+        if (memoryUsagePercentage > MEMORY_MONITOR_THRESHOLD) {
+            LOG.info("JVM memory usage is '{}', exceeding the threshold of '{}'.",
+                     memoryUsagePercentage, MEMORY_MONITOR_THRESHOLD);
+            System.gc();
+            LOG.info("Trigger System.gc()");
+
+            double doubleCheckUsage = getMemoryUsagePercentage();
+            if (doubleCheckUsage > MEMORY_MONITOR_THRESHOLD) {
+                Thread targetThread = getHighestMemoryThread();
+                if (targetThread != null) {
+                    LOG.info("JVM memory usage is '{}', exceeding the threshold of '{}'.",
+                             doubleCheckUsage, MEMORY_MONITOR_THRESHOLD);
+                    targetThread.interrupt();
+                    LOG.info("Send interrupt to '{}' thread",
+                             targetThread.getName());
+                }
+            }
+        }
+    }
+
+    private double getMemoryUsagePercentage() {
+        MemoryUsage heapMemoryUsage =
+                ManagementFactory.getMemoryMXBean().getHeapMemoryUsage();
+        return (double) heapMemoryUsage.getUsed() / heapMemoryUsage.getMax();
+    }
+
+    private Thread getHighestMemoryThread() {
+        long highestMemory = 0;
+        Thread highestThread = null;
+
+        ThreadMXBean threadMXBean = (ThreadMXBean) ManagementFactory.getThreadMXBean();
+
+        for (Thread thread : Thread.getAllStackTraces().keySet()) {
+            if (thread.getState() != Thread.State.RUNNABLE ||
+                thread.getName() == null ||
+                !thread.getName().startsWith("grizzly-http-server-")) {
+                continue;
+            }
+
+            long threadMemory =
+                    threadMXBean.getThreadAllocatedBytes(thread.getId());
+            if (threadMemory > highestMemory) {
+                highestMemory = threadMemory;
+                highestThread = thread;
+            }
+        }
+
+        return highestThread;
+    }
+
+    public void start() {
+        if (MEMORY_MONITOR_THRESHOLD >= 1.0) {
+            return;
+        }
+        Runnable task = this::manageMemory;
+        scheduler.scheduleAtFixedRate(task, 0, MEMORY_MONITOR_PERIOD,
+                                      TimeUnit.MILLISECONDS);
+        LOG.info("Memory monitoring task started.");
+    }
+
+    public void stop() {
+        if (MEMORY_MONITOR_THRESHOLD >= 1.0) {
+            return;
+        }
+        scheduler.shutdownNow();
+        LOG.info("Memory monitoring task ended.");
+    }
+}