From: Dessalines <tyhou13@gmx.com>
Date: Sat, 4 May 2019 15:53:29 +0000 (-0700)
Subject: More reorg
X-Git-Url: http://these/git/readmes/%7B%60%24%7BarchiveTodayUrl%7D/static/gitweb.js?a=commitdiff_plain;h=779e5964fadc51fd59de1ca53d9a65e7c4f367ad;p=lemmy.git

More reorg
---

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());
     }