More reorg

pull/283/head
Dessalines 2019-05-04 08:53:29 -07:00
parent 23157e2e8d
commit 1eb0937190
2 changed files with 8 additions and 1 deletions

View File

@ -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()
},
)
}

View File

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