From 1eb0937190cf4a1d928bada4e1ecc95d8d728dab Mon Sep 17 00:00:00 2001
From: Dessalines <happydooby@gmail.com>
Date: Sat, 4 May 2019 08:53:29 -0700
Subject: [PATCH] More reorg

---
 server/src/main.rs             | 8 +++++++-
 server/src/websocket/server.rs | 1 +
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/server/src/main.rs b/server/src/main.rs
index 3e51785..add17cf 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 a530b9a..07378d3 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());
     }