From f70cc988e92d883ec2c956052e4a62f1d21de52a Mon Sep 17 00:00:00 2001
From: Gabriel Kaszewski <gabrielkaszewski@gmail.com>
Date: Tue, 14 May 2024 18:23:14 +0200
Subject: [PATCH] remove disconnected users

---
 src/main.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index 03684fc..e7640c9 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -83,6 +83,15 @@ async fn on_connect(socket: SocketRef, state: Arc<AppState>) {
         info!("Socket disconnected: {:?}", socket.id);
         let mut soldiers = app_state_for_disconnect.soldiers.lock().unwrap();
         soldiers.remove(&socket.id.to_string());
+
+        info!("Soldiers: {:?}", soldiers.len());
+
+        let mut last_update = app_state_for_disconnect.last_update.lock().unwrap();
+        let socket_id = socket.id.to_string();
+        last_update.remove(&socket_id);
+
+        info!("Last update: {:?}", last_update.len());
+
         let soldiers_num = soldiers.len();
         socket
             .broadcast()