Add ping statistics and command

2 files changed, 193 insertions, 0 deletions

diff --git a/spark-common/src/main/java/me/lucko/spark/common/util/RollingAverage.java b/spark-common/src/main/java/me/lucko/spark/common/util/RollingAverage.java
index 87c41a4..57dfdff 100644
--- a/spark-common/src/main/java/me/lucko/spark/common/util/RollingAverage.java
+++ b/spark-common/src/main/java/me/lucko/spark/common/util/RollingAverage.java
@@ -39,6 +39,12 @@ public class RollingAverage implements DoubleAverageInfo {
         this.samples = new ArrayDeque<>(this.windowSize + 1);
     }
 
+    public int getSamples() {
+        synchronized (this) {
+            return this.samples.size();
+        }
+    }
+
     public void add(BigDecimal num) {
         synchronized (this) {
             this.total = this.total.add(num);
diff --git a/spark-common/src/main/java/me/lucko/spark/common/util/StatisticFormatter.java b/spark-common/src/main/java/me/lucko/spark/common/util/StatisticFormatter.java
new file mode 100644
index 0000000..9a2850e
--- /dev/null
+++ b/spark-common/src/main/java/me/lucko/spark/common/util/StatisticFormatter.java
@@ -0,0 +1,187 @@
+/*
+ * This file is part of spark.
+ *
+ *  Copyright (c) lucko (Luck) <luck@lucko.me>
+ *  Copyright (c) contributors
+ *
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package me.lucko.spark.common.util;
+
+import com.google.common.base.Strings;
+
+import net.kyori.adventure.text.Component;
+import net.kyori.adventure.text.TextComponent;
+import net.kyori.adventure.text.format.TextColor;
+
+import java.lang.management.MemoryUsage;
+
+import static net.kyori.adventure.text.Component.text;
+import static net.kyori.adventure.text.format.NamedTextColor.DARK_GRAY;
+import static net.kyori.adventure.text.format.NamedTextColor.GRAY;
+import static net.kyori.adventure.text.format.NamedTextColor.GREEN;
+import static net.kyori.adventure.text.format.NamedTextColor.RED;
+import static net.kyori.adventure.text.format.NamedTextColor.YELLOW;
+
+public enum StatisticFormatter {
+    ;
+
+    private static final String BAR_CHARACTER = "|";
+
+    public static TextComponent formatTps(double tps) {
+        TextColor color;
+        if (tps > 18.0) {
+            color = GREEN;
+        } else if (tps > 16.0) {
+            color = YELLOW;
+        } else {
+            color = RED;
+        }
+
+        return text((tps > 20.0 ? "*" : "") + Math.min(Math.round(tps * 100.0) / 100.0, 20.0), color);
+    }
+
+    public static TextComponent formatTickDurations(RollingAverage average) {
+        return text()
+                .append(formatTickDuration(average.min()))
+                .append(text('/', GRAY))
+                .append(formatTickDuration(average.median()))
+                .append(text('/', GRAY))
+                .append(formatTickDuration(average.percentile95th()))
+                .append(text('/', GRAY))
+                .append(formatTickDuration(average.max()))
+                .build();
+    }
+
+    public static TextComponent formatTickDuration(double duration) {
+        TextColor color;
+        if (duration >= 50d) {
+            color = RED;
+        } else if (duration >= 40d) {
+            color = YELLOW;
+        } else {
+            color = GREEN;
+        }
+
+        return text(String.format("%.1f", duration), color);
+    }
+
+    public static TextComponent formatCpuUsage(double usage) {
+        TextColor color;
+        if (usage > 0.9) {
+            color = RED;
+        } else if (usage > 0.65) {
+            color = YELLOW;
+        } else {
+            color = GREEN;
+        }
+
+        return text(FormatUtil.percent(usage, 1d), color);
+    }
+
+    public static TextComponent formatPingRtts(double min, double median, double percentile95th, double max) {
+        return text()
+                .append(formatPingRtt(min))
+                .append(text('/', GRAY))
+                .append(formatPingRtt(median))
+                .append(text('/', GRAY))
+                .append(formatPingRtt(percentile95th))
+                .append(text('/', GRAY))
+                .append(formatPingRtt(max))
+                .build();
+    }
+
+    public static TextComponent formatPingRtt(double ping) {
+        TextColor color;
+        if (ping >= 200) {
+            color = RED;
+        } else if (ping >= 100) {
+            color = YELLOW;
+        } else {
+            color = GREEN;
+        }
+
+        return text((int) Math.ceil(ping), color);
+    }
+
+    public static TextComponent generateMemoryUsageDiagram(MemoryUsage usage, int length) {
+        double used = usage.getUsed();
+        double committed = usage.getCommitted();
+        double max = usage.getMax();
+
+        int usedChars = (int) ((used * length) / max);
+        int committedChars = (int) ((committed * length) / max);
+
+        TextComponent.Builder line = text().content(Strings.repeat(BAR_CHARACTER, usedChars)).color(YELLOW);
+        if (committedChars > usedChars) {
+            line.append(text(Strings.repeat(BAR_CHARACTER, (committedChars - usedChars) - 1), GRAY));
+            line.append(Component.text(BAR_CHARACTER, RED));
+        }
+        if (length > committedChars) {
+            line.append(text(Strings.repeat(BAR_CHARACTER, (length - committedChars)), GRAY));
+        }
+
+        return text()
+                .append(text("[", DARK_GRAY))
+                .append(line.build())
+                .append(text("]", DARK_GRAY))
+                .build();
+    }
+
+    public static TextComponent generateMemoryPoolDiagram(MemoryUsage usage, MemoryUsage collectionUsage, int length) {
+        double used = usage.getUsed();
+        double collectionUsed = used;
+        if (collectionUsage != null) {
+            collectionUsed = collectionUsage.getUsed();
+        }
+        double committed = usage.getCommitted();
+        double max = usage.getMax();
+
+        int usedChars = (int) ((used * length) / max);
+        int collectionUsedChars = (int) ((collectionUsed * length) / max);
+        int committedChars = (int) ((committed * length) / max);
+
+        TextComponent.Builder line = text().content(Strings.repeat(BAR_CHARACTER, collectionUsedChars)).color(YELLOW);
+
+        if (usedChars > collectionUsedChars) {
+            line.append(Component.text(BAR_CHARACTER, RED));
+            line.append(text(Strings.repeat(BAR_CHARACTER, (usedChars - collectionUsedChars) - 1), YELLOW));
+        }
+        if (committedChars > usedChars) {
+            line.append(text(Strings.repeat(BAR_CHARACTER, (committedChars - usedChars) - 1), GRAY));
+            line.append(Component.text(BAR_CHARACTER, YELLOW));
+        }
+        if (length > committedChars) {
+            line.append(text(Strings.repeat(BAR_CHARACTER, (length - committedChars)), GRAY));
+        }
+
+        return text()
+                .append(text("[", DARK_GRAY))
+                .append(line.build())
+                .append(text("]", DARK_GRAY))
+                .build();
+    }
+
+    public static TextComponent generateDiskUsageDiagram(double used, double max, int length) {
+        int usedChars = (int) ((used * length) / max);
+        int freeChars = length - usedChars;
+        return text()
+                .append(text("[", DARK_GRAY))
+                .append(text(Strings.repeat(BAR_CHARACTER, usedChars), YELLOW))
+                .append(text(Strings.repeat(BAR_CHARACTER, freeChars), GRAY))
+                .append(text("]", DARK_GRAY))
+                .build();
+    }
+}