diff options
-rw-r--r-- | server/src/main.rs | 8 | ||||
-rw-r--r-- | server/src/websocket/server.rs | 1 |
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()); } |