vw_small

commit 3d36ac460192cc2e0a95f33c2ef102ce1da97cab
parent 8b3e87cfe05ec6fee8556a71fd03045210bede59
Author: Daniel García <dani-garcia@users.noreply.github.com>
Date:   Wed, 21 Nov 2018 15:07:18 +0100

Remove unwrap in connection_lost

Diffstat:
M
src/api/notifications.rs
 | 
10
++++------

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