More reorg

2 files changed, 8 insertions, 1 deletions

diff --git a/server/src/main.rs b/server/src/main.rs
index 3e51785b..add17cfd 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -29,7 +29,13 @@ fn chat_route(req: &HttpRequest<WsChatSessionState>) -> Result<HttpResponse, Err
     WSSession {
       id: 0,
       hb: Instant::now(),
-      ip: req.connection_info().remote().unwrap_or("127.0.0.1:12345").split(":").next().unwrap_or("127.0.0.1").to_string()
+      ip: req.connection_info()
+        .remote()
+        .unwrap_or("127.0.0.1:12345")
+        .split(":")
+        .next()
+        .unwrap_or("127.0.0.1")
+        .to_string()
     },
     )
 }
diff --git a/server/src/websocket/server.rs b/server/src/websocket/server.rs
index a530b9a6..07378d37 100644
--- a/server/src/websocket/server.rs
+++ b/server/src/websocket/server.rs
@@ -1233,6 +1233,7 @@ impl Perform for GetPost {
       sessions.remove(&addr);
     }
 
+    // If the room doesn't exist yet
     if chat.rooms.get_mut(&self.id).is_none() {
       chat.rooms.insert(self.id, HashSet::new());
     }