Add shutdown handling to sponge

WiIIiam278 · May 22, 2023 · 85daac9 · 85daac9
1 parent 66d0b9f
commit 85daac9
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/sponge/src/main/java/net/william278/huskhomes/SpongeHuskHomes.java b/sponge/src/main/java/net/william278/huskhomes/SpongeHuskHomes.java
@@ -63,6 +63,7 @@
 import org.spongepowered.api.event.lifecycle.ConstructPluginEvent;
 import org.spongepowered.api.event.lifecycle.RegisterCommandEvent;
 import org.spongepowered.api.event.lifecycle.StartedEngineEvent;
+import org.spongepowered.api.event.lifecycle.StoppingEngineEvent;
 import org.spongepowered.api.network.EngineConnection;
 import org.spongepowered.api.network.channel.ChannelBuf;
 import org.spongepowered.api.network.channel.raw.RawDataChannel;
@@ -201,6 +202,17 @@ public void onServerStarted(final StartedEngineEvent<org.spongepowered.api.Serve
         initialize("metrics", (plugin) -> this.registerMetrics(METRICS_ID));
     }
 
+    @Listener
+    public void onShutdown(final StoppingEngineEvent<org.spongepowered.api.Server> event) {
+        if (database != null) {
+            database.terminate();
+        }
+        if (broker != null) {
+            broker.close();
+        }
+        cancelAllTasks();
+    }
+
     @Listener
     public void onRegisterCommands(final RegisterCommandEvent<Raw> event) {
         initialize("commands", (plugin) -> this.commands = registerCommands(event));