Remove unwrap in connection_lost

This commit is contained in:
Daniel García 2018-11-21 15:07:18 +01:00
parent 8b3e87cfe0
commit 3d36ac4601
No known key found for this signature in database
GPG Key ID: FC8A7D14C3CD543A

View File

@ -219,7 +219,6 @@ impl Factory for WSFactory {
type Handler = WSHandler; type Handler = WSHandler;
fn connection_made(&mut self, out: Sender) -> Self::Handler { fn connection_made(&mut self, out: Sender) -> Self::Handler {
println!("WS: Connection made");
WSHandler { WSHandler {
out, out,
user_uuid: None, user_uuid: None,
@ -228,12 +227,11 @@ impl Factory for WSFactory {
} }
fn connection_lost(&mut self, handler: Self::Handler) { fn connection_lost(&mut self, handler: Self::Handler) {
println!("WS: Connection lost");
// Remove handler // Remove handler
let user_uuid = &handler.user_uuid.unwrap(); if let Some(user_uuid) = &handler.user_uuid {
if let Some(mut user_conn) = self.users.map.get_mut(user_uuid) { if let Some(mut user_conn) = self.users.map.get_mut(user_uuid) {
user_conn.remove_item(&handler.out); user_conn.remove_item(&handler.out);
}
} }
} }
} }