fix(backend): close connection even without handshake

ravenclaw900 · Nov 14, 2021 · 630d38d · 630d38d
1 parent 8c8b63c
commit 630d38d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/backend/src/sockets.rs b/src/backend/src/sockets.rs
@@ -288,7 +288,10 @@ pub async fn socket_handler(socket: warp::ws::WebSocket) {
         let mut first_message = true;
         let mut req: types::Request;
         loop {
-            let data = socket_recv.next().await.unwrap().unwrap();
+            let data = match socket_recv.next().await {
+                Some(Ok(data)) => data,
+                Some(Err(_)) | None => break,
+            };
             if data.is_close() {
                 break;
             }