Refactor TPS metrics handling and introduce API

Author: renvins

fabric/src/main/java/it/renvins/serverpulse/fabric/ServerPulseFabric.java

@@ -2,16 +2,69 @@
 
 import java.util.logging.Logger;
 
+import it.renvins.serverpulse.api.ServerPulseProvider;
+import it.renvins.serverpulse.api.metrics.IDiskRetriever;
+import it.renvins.serverpulse.api.metrics.IPingRetriever;
+import it.renvins.serverpulse.api.metrics.ITPSRetriever;
+import it.renvins.serverpulse.api.service.IDatabaseService;
+import it.renvins.serverpulse.api.service.IMetricsService;
+import it.renvins.serverpulse.common.DatabaseService;
+import it.renvins.serverpulse.common.MetricsService;
+import it.renvins.serverpulse.common.config.DatabaseConfiguration;
+import it.renvins.serverpulse.common.config.MetricsConfiguration;
+import it.renvins.serverpulse.common.logger.PulseLogger;
+import it.renvins.serverpulse.common.metrics.DiskRetriever;
+import it.renvins.serverpulse.common.platform.Platform;
+import it.renvins.serverpulse.common.scheduler.TaskScheduler;
+import it.renvins.serverpulse.fabric.config.FabricConfiguration;
+import it.renvins.serverpulse.fabric.config.FabricDatabaseConfiguration;
+import it.renvins.serverpulse.fabric.config.FabricMetricsConfiguration;
+import it.renvins.serverpulse.fabric.logger.FabricLogger;
+import it.renvins.serverpulse.fabric.metrics.FabricPingRetriever;
+import it.renvins.serverpulse.fabric.metrics.FabricTPSRetriever;
+import it.renvins.serverpulse.fabric.platform.FabricPlatform;
+import it.renvins.serverpulse.fabric.task.FabricScheduler;
 import lombok.Getter;
 import net.fabricmc.api.ModInitializer;
 import net.fabricmc.fabric.api.event.lifecycle.v1.ServerLifecycleEvents;
+import net.fabricmc.loader.api.FabricLoader;
 import net.minecraft.server.MinecraftServer;
 
 public class ServerPulseFabric implements ModInitializer {
 
     public static final String MOD_ID = "serverpulse";
     public static final Logger LOGGER = Logger.getLogger(MOD_ID);
 
+    private final FabricConfiguration config;
+
+    private final Platform platform;
+
+    private final IDatabaseService databaseService;
+    private final IMetricsService metricsService;
+
+    private final ITPSRetriever tpsRetriever;
+    private final IDiskRetriever diskRetriever;
+    private final IPingRetriever pingRetriever;
+
+    public ServerPulseFabric() {
+        this.config = new FabricConfiguration(FabricLoader.getInstance().getConfigDir(), "config.yml");
+
+        PulseLogger logger = new FabricLogger();
+
+        this.platform = new FabricPlatform(this);
+        TaskScheduler scheduler = new FabricScheduler();
+
+        DatabaseConfiguration dbConfig = new FabricDatabaseConfiguration(config);
+        MetricsConfiguration metricsConfig = new FabricMetricsConfiguration(config);
+
+        this.databaseService = new DatabaseService(logger, platform, dbConfig, scheduler);
+        this.metricsService = new MetricsService(logger, platform, metricsConfig, scheduler);
+
+        this.tpsRetriever = new FabricTPSRetriever(scheduler);
+        this.diskRetriever = new DiskRetriever(FabricLoader.getInstance().getGameDir().toFile());
+        this.pingRetriever = new FabricPingRetriever(this);
+    }
+
     @Getter private MinecraftServer server;
 
     @Override
@@ -25,10 +78,28 @@ public void onInitialize() {
     private void onServerStarting(MinecraftServer server) {
         LOGGER.info("ServerPulse is starting...");
         this.server = server;
+
+        config.load();
+
+        ServerPulseProvider.register(new ServerPulseFabricAPI(databaseService, metricsService, tpsRetriever, diskRetriever, pingRetriever));
+
+        databaseService.load();
+        if (!platform.isEnabled()) {
+            return;
+        }
+        metricsService.load();
+
+        LOGGER.info("Starting tick monitoring task...");
+        ((FabricTPSRetriever) tpsRetriever).startTickMonitor();
     }
 
     private void onServerStopped(MinecraftServer server) {
         LOGGER.info("ServerPulse is stopping...");
+
+        databaseService.unload();
+        metricsService.unload();
+
+        ServerPulseProvider.unregister();
         this.server = null;
     }
 }

fabric/src/main/java/it/renvins/serverpulse/fabric/ServerPulseFabricAPI.java

@@ -0,0 +1,44 @@
+package it.renvins.serverpulse.fabric;
+
+import it.renvins.serverpulse.api.ServerPulseAPI;
+import it.renvins.serverpulse.api.metrics.IDiskRetriever;
+import it.renvins.serverpulse.api.metrics.IPingRetriever;
+import it.renvins.serverpulse.api.metrics.ITPSRetriever;
+import it.renvins.serverpulse.api.service.IDatabaseService;
+import it.renvins.serverpulse.api.service.IMetricsService;
+import lombok.RequiredArgsConstructor;
+
+@RequiredArgsConstructor
+public class ServerPulseFabricAPI implements ServerPulseAPI {
+
+    private final IDatabaseService databaseService;
+    private final IMetricsService metricsService;
+    private final ITPSRetriever tpsRetriever;
+    private final IDiskRetriever diskRetriever;
+    private final IPingRetriever pingRetriever;
+
+    @Override
+    public IDatabaseService getDatabaseService() {
+        return databaseService;
+    }
+
+    @Override
+    public IMetricsService getMetricsService() {
+        return metricsService;
+    }
+
+    @Override
+    public ITPSRetriever getTPSRetriever() {
+        return tpsRetriever;
+    }
+
+    @Override
+    public IDiskRetriever getDiskRetriever() {
+        return diskRetriever;
+    }
+
+    @Override
+    public IPingRetriever getPingRetriever() {
+        return pingRetriever;
+    }
+}

fabric/src/main/java/it/renvins/serverpulse/fabric/metrics/FabricTPSRetriever.java

@@ -4,8 +4,7 @@
 import java.util.Queue;
 
 import it.renvins.serverpulse.api.metrics.ITPSRetriever;
-import it.renvins.serverpulse.fabric.task.FabricScheduler;
-import it.renvins.serverpulse.fabric.task.FabricTask;
+import it.renvins.serverpulse.common.scheduler.TaskScheduler;
 import lombok.RequiredArgsConstructor;
 
 @RequiredArgsConstructor
@@ -20,7 +19,7 @@ public class FabricTPSRetriever implements ITPSRetriever {
 
     private final Queue<Long> tickHistory = new LinkedList<>();
 
-    private final FabricScheduler scheduler;
+    private final TaskScheduler scheduler;
 
     private long lastTickTimeNano = -1;
 
@@ -30,7 +29,8 @@ public class FabricTPSRetriever implements ITPSRetriever {
 
     @Override
     public double[] getTPS() {
-        return new double[0];
+        calculateAverages();
+        return new double[]{tps1m, tps5m, tps15m};
     }
 
     public void startTickMonitor() {
@@ -49,7 +49,32 @@ public void startTickMonitor() {
     }
 
     public void calculateAverages() {
-        // Calculate TPS for 1m, 5m, and 15m
+        long sum1m = 0, sum5m = 0, sum15m = 0;
+        int count1m = 0, count5m = 0, count15m = 0;
+
+        Object[] durations = tickHistory.toArray();
+        for (int i = durations.length - 1; i >= 0; i--) {
+            if (!(durations[i] instanceof Long)) continue;
+            long durationNano = (Long) durations[i];
+            if (i < TICK_PER_MIN) {
+                sum1m += durationNano;
+                count1m++;
+            }
+            if (i < TICK_FIVE_MIN) {
+                sum5m += durationNano;
+                count5m++;
+            }
+            if (i < TICK_FIFTEEN_MIN) {
+                sum15m += durationNano;
+                count15m++;
+            } else {
+                break;
+            }
+        }
+
+        tps1m = calculateTPSFromAvgNano(sum1m, count1m);
+        tps5m = calculateTPSFromAvgNano(sum5m, count5m);
+        tps15m = calculateTPSFromAvgNano(sum15m, count15m);
     }
 
     public double calculateTPSFromAvgNano(long totalNano, int count) {