Readded message if user lost connection.

magefree · Jan 28, 2018 · 34a8845 · 34a8845
1 parent 66eb1de
commit 34a8845
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/Mage.Server/src/main/java/mage/server/ChatManager.java b/Mage.Server/src/main/java/mage/server/ChatManager.java
@@ -295,6 +295,15 @@ public void sendReconnectMessage(UUID userId) {
 
     }
 
+    public void sendLostConnectionMessage(UUID userId, DisconnectReason reason) {
+        UserManager.instance.getUser(userId).ifPresent(user
+                -> getChatSessions()
+                        .stream()
+                        .filter(chat -> chat.hasUser(userId))
+                        .forEach(chatSession -> chatSession.broadcast(null, user.getName() + reason.getMessage(), MessageColor.BLUE, true, MessageType.STATUS, null)));
+
+    }
+
     public void removeUser(UUID userId, DisconnectReason reason) {
         for (ChatSession chatSession : getChatSessions()) {
             if (chatSession.hasUser(userId)) {

diff --git a/Mage.Server/src/main/java/mage/server/UserManager.java b/Mage.Server/src/main/java/mage/server/UserManager.java
@@ -139,6 +139,8 @@ public void disconnect(UUID userId, DisconnectReason reason) {
             if (reason == DisconnectReason.Disconnected) {
                 removeUserFromAllTablesAndChat(userId, reason);
                 user.get().setUserState(UserState.Offline);
+            } else {
+                ChatManager.instance.sendLostConnectionMessage(userId, reason);
             }
         }
     }