option to restart at specific tps

moom0o · Jul 1, 2021 · 5b80320 · 5b80320
1 parent 07ba7ae
commit 5b80320
Show file tree

Hide file tree

Showing 4 changed files with 49 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -20,4 +20,10 @@ finalstring: "&e[SERVER] Server restarting..."
 minutestring: "minutes..."
 secondsstring: "seconds..."
 secondstring: "second..."
+
+# REBOOT AT LOW TPS SETTINGS
+RebootFromLowTPS: false
+TPSToStartCounting: 10
+HowLongShouldTheServerGoWithLowTPS: 300 # seconds, the counter is reset if the server is above the specified tps for one second.
+InstantRestart: false # Should the server send the 15 minute countdown messages?
 ```
diff --git a/pom.xml b/pom.xml
@@ -6,7 +6,7 @@
 
     <groupId>me.moomoo</groupId>
     <artifactId>RestartPlugin</artifactId>
-    <version>2.0</version>
+    <version>3.0</version>
     <packaging>jar</packaging>
 
     <name>RestartPlugin</name>

diff --git a/src/main/java/me/moomoo/restartplugin/RestartPlugin.java b/src/main/java/me/moomoo/restartplugin/RestartPlugin.java
@@ -17,6 +17,8 @@
 
 public class RestartPlugin extends JavaPlugin implements Listener {
     FileConfiguration config = getConfig();
+    int secondsServerWasAtLowTPS = 0;
+    Boolean serverIsRestarting = false;
 
     public void onEnable() {
         saveDefaultConfig();
@@ -48,10 +50,43 @@ public void onEnable() {
                 });
                 t.start();
             }, initalDelay, TimeUnit.DAYS.toSeconds(1), TimeUnit.SECONDS);
+
+            if (getConfig().getBoolean("RebootFromLowTPS")) {
+                ScheduledExecutorService schedulerTPS = Executors.newScheduledThreadPool(1);
+                schedulerTPS.scheduleAtFixedRate(() -> {
+                    Thread t = new Thread(() -> {
+                        double tps = Bukkit.getServer().getTPS()[0];
+                        if (secondsServerWasAtLowTPS > getConfig().getInt("HowLongShouldTheServerGoWithLowTPS")) {
+                            if (!serverIsRestarting) {
+                                getLogger().warning("The server is rebooting because the tps was lower than " + getConfig().getDouble("TPSToStartCounting") + " for " + getConfig().getInt("HowLongShouldTheServerGoWithLowTPS") + " seconds.");
+                            }
+                            if (getConfig().getBoolean("InstantRestart")) {
+                                Bukkit.shutdown();
+                            } else {
+                                try {
+                                    restart();
+                                } catch (InterruptedException e) {
+                                    e.printStackTrace();
+                                }
+                            }
+                        }
+                        if (tps < getConfig().getDouble("TPSToStartCounting")) {
+                            secondsServerWasAtLowTPS++;
+                        } else {
+                            secondsServerWasAtLowTPS = 0;
+                        }
+                    });
+                    t.start();
+                }, 1, 1, TimeUnit.SECONDS);
+            }
         });
     }
 
     public void restart() throws InterruptedException {
+        if (serverIsRestarting) {
+            return;
+        }
+        serverIsRestarting = true;
         String s = get("string");
         // Pull req. If you find a better way to do this!
         b(translate(s.replace("%time%", "15").replace("%timeword%", get("minutestring"))));

diff --git a/src/main/resources/config.yml b/src/main/resources/config.yml
@@ -12,4 +12,10 @@ string: "&e[SERVER] Server restarting in %time% %timeword%"
 finalstring: "&e[SERVER] Server restarting..."
 minutestring: "minutes..."
 secondsstring: "seconds..."
-secondstring: "second..."
+secondstring: "second..."
+
+# REBOOT AT LOW TPS SETTINGS
+RebootFromLowTPS: false
+TPSToStartCounting: 10
+HowLongShouldTheServerGoWithLowTPS: 300 # seconds, the counter is reset if the server is above the specified tps for one second.
+InstantRestart: false # Should the server send the 15 minute countdown messages?