From 42ece4bdcc90689d032d5f2c2569a841d119d62e Mon Sep 17 00:00:00 2001
From: a3510377 <a102009102009@gmail.com>
Date: Sun, 5 Nov 2023 20:47:55 +0800
Subject: [PATCH] feat player metrics

---
 .../java/monkey/info/command/Prometheus.java  |  4 ++-
 .../monkey/info/command/metrics/Mobcaps.java  |  7 ++--
 .../monkey/info/command/metrics/Player.java   | 32 +++++++++++++++++++
 3 files changed, 39 insertions(+), 4 deletions(-)
 create mode 100644 src/main/java/monkey/info/command/metrics/Player.java

diff --git a/src/main/java/monkey/info/command/Prometheus.java b/src/main/java/monkey/info/command/Prometheus.java
index 04303f7..3c2e56e 100644
--- a/src/main/java/monkey/info/command/Prometheus.java
+++ b/src/main/java/monkey/info/command/Prometheus.java
@@ -5,6 +5,7 @@
 import io.prometheus.metrics.model.registry.PrometheusRegistry;
 import monkey.info.command.metrics.Metric;
 import monkey.info.command.metrics.Mobcaps;
+import monkey.info.command.metrics.Player;
 import monkey.info.command.metrics.Tick;
 import org.jetbrains.annotations.Nullable;
 
@@ -39,8 +40,9 @@ public void run() {
 
     public void start() {
         this.stop();
-        this.registerMetric(new Tick());
         this.registerMetric(new Mobcaps());
+        this.registerMetric(new Player());
+        this.registerMetric(new Tick());
 
         if (metricUpdateLoop == null) metricUpdateLoop = new Timer("Metric Update Loop");
         metricUpdateLoop.scheduleAtFixedRate(new TimerTask() {
diff --git a/src/main/java/monkey/info/command/metrics/Mobcaps.java b/src/main/java/monkey/info/command/metrics/Mobcaps.java
index efb4462..e0a265f 100644
--- a/src/main/java/monkey/info/command/metrics/Mobcaps.java
+++ b/src/main/java/monkey/info/command/metrics/Mobcaps.java
@@ -18,15 +18,16 @@ public Mobcaps() {
     public void update(MinecraftServer server) {
         // edit from: https://github.com/gnembon/fabric-carpet/blob/7486670918f733aaf6ae3b595290947802024d02/src/main/java/carpet/utils/SpawnReporter.java#L105
         for (ServerWorld world : server.getWorlds()) {
-            RegistryKey<World> dim = world.getRegistryKey();
             SpawnHelper.Info lastSpawner = world.getChunkManager().getSpawnInfo();
-            int chunkcount = SpawnReporter.chunkCounts.getOrDefault(dim, -1);
 
             if (lastSpawner != null) {
+                RegistryKey<World> dim = world.getRegistryKey();
+                String worldName = world.getRegistryKey().getValue().toString();
+                int chunkcount = SpawnReporter.chunkCounts.getOrDefault(dim, -1);
+
                 for (SpawnGroup enumcreaturetype : SpawnGroup.values()) {
                     Object2IntMap<SpawnGroup> dimCounts = lastSpawner.getGroupToCount();
                     String enumcreature = enumcreaturetype.toString();
-                    String worldName = world.getRegistryKey().getValue().toString();
 
                     int cur = dimCounts.getOrDefault(enumcreaturetype, -1);
                     int max = (int) (chunkcount * ((double) enumcreaturetype.getCapacity() / SpawnReporter.currentMagicNumber()));
diff --git a/src/main/java/monkey/info/command/metrics/Player.java b/src/main/java/monkey/info/command/metrics/Player.java
new file mode 100644
index 0000000..6e8c208
--- /dev/null
+++ b/src/main/java/monkey/info/command/metrics/Player.java
@@ -0,0 +1,32 @@
+package monkey.info.command.metrics;
+
+import carpet.patches.EntityPlayerMPFake;
+import net.minecraft.server.MinecraftServer;
+import net.minecraft.server.network.ServerPlayerEntity;
+import net.minecraft.server.world.ServerWorld;
+import net.minecraft.util.registry.RegistryKey;
+import net.minecraft.world.World;
+
+import java.util.List;
+
+public class Player extends Metric {
+    public Player() {
+        super("players_online", "Online players", "type", "world");
+    }
+
+    @Override
+    public void update(MinecraftServer server) {
+        // edit from: https://github.com/gnembon/fabric-carpet/blob/7486670918f733aaf6ae3b595290947802024d02/src/main/java/carpet/utils/SpawnReporter.java#L105
+        for (ServerWorld world : server.getWorlds()) {
+            RegistryKey<World> dim = world.getRegistryKey();
+            List<ServerPlayerEntity> players = world.getPlayers();
+            String worldName = world.getRegistryKey().getValue().toString();
+
+            long fakePlayers = players.stream().filter(player -> player instanceof EntityPlayerMPFake).count();
+            long realPlayers = players.size() - fakePlayers;
+
+            this.getGauge().labelValues("FAKE", worldName).set(fakePlayers);
+            this.getGauge().labelValues("REAL", worldName).set(realPlayers);
+        }
+    }
+}