More reorg

diff --git a/server/src/main.rs b/server/src/main.rs
index 3e51785b30bdc1fa47efc6f7ccc290baea5137bc..add17cfd6af26b6632b70a386b51b0b327e64f5e 100644 (file)
--- 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 a530b9a6594d3b3f1d79c212818f7485700d5dd3..07378d37aa671d4c4e6aa8abc5248a9df47ac18e 100644 (file)
--- 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());
     }