From 2f2f4fa44fdc04de1d69051dc274ae1c0c81bc2e Mon Sep 17 00:00:00 2001
From: Alex <alex@roese.dev>
Date: Mon, 27 Dec 2021 23:52:22 +0100
Subject: [PATCH] auto restart system

---
 .../proxy/listener/SocketClientMessageListener.java | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

diff --git a/src/main/java/net/krakatoa/proxy/listener/SocketClientMessageListener.java b/src/main/java/net/krakatoa/proxy/listener/SocketClientMessageListener.java
index 1399e6c..a4d0152 100644
--- a/src/main/java/net/krakatoa/proxy/listener/SocketClientMessageListener.java
+++ b/src/main/java/net/krakatoa/proxy/listener/SocketClientMessageListener.java
@@ -73,11 +73,9 @@ public class SocketClientMessageListener implements KraSocketClientEventInterfac
             ServiceTask serviceTask = CloudNetDriver.getInstance().getServiceTaskProvider()
                 .getServiceTask("Lobby");
 
-            if (!playersAutoRestartServerSystem.containsKey("lobby")) {
-              playersAutoRestartServerSystem.put("lobby", null);
-            }
-
             if (serviceTask != null) {
+              playersAutoRestartServerSystem.put(serviceTask.getName(), null);
+
               ServiceInfoSnapshot serviceInfoSnapshot = ServiceConfiguration.builder(serviceTask)
                   .build().createNewService();
 
@@ -106,8 +104,6 @@ public class SocketClientMessageListener implements KraSocketClientEventInterfac
             // stop target server
             CloudNetDriver.getInstance().getCloudServiceProvider(serviceInfoSnapshot).stop();
           }
-
-
         }
         break;
       }
@@ -115,15 +111,14 @@ public class SocketClientMessageListener implements KraSocketClientEventInterfac
         String serverName = kraProtocolMessage.getArgs().toLowerCase();
 
         if (serverName.startsWith("lobby") && playersAutoRestartServerSystem.containsKey(
-            "lobby")) { // close old lobbies
+            kraProtocolMessage.getArgs())) { // close old lobbies
           Collection<ServiceInfoSnapshot> serviceInfoSnapshots = CloudNetDriver.getInstance()
               .getCloudServiceProvider()
               .getCloudServices("Lobby");
 
           if (serviceInfoSnapshots != null) {
             serviceInfoSnapshots.forEach(serviceInfoSnapshot -> {
-              System.out.println("lobby " + serviceInfoSnapshot.getName());
-              if (!serverName.equals(serviceInfoSnapshot.getName())) {
+              if (!playersAutoRestartServerSystem.containsKey(serviceInfoSnapshot.getName())) {
                 serviceInfoSnapshot.provider().stop();
               }
             });