From 14a801f4d411a7e7a2b4a251c9f3b771802d8e8a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Frosteg=C3=A5rd?= <joakim.frostegard@gmail.com>
Date: Tue, 3 Jan 2023 18:27:56 +0100
Subject: [PATCH] ws socket worker: replace unwrap with error return

---
 aquatic_ws/src/workers/socket.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/aquatic_ws/src/workers/socket.rs b/aquatic_ws/src/workers/socket.rs
index 5d730a1..81190c5 100644
--- a/aquatic_ws/src/workers/socket.rs
+++ b/aquatic_ws/src/workers/socket.rs
@@ -455,7 +455,11 @@ impl<S: futures::AsyncRead + futures::AsyncWrite + Unpin> ConnectionReader<S> {
                 yield_if_needed().await;
             }
 
-            let message = self.ws_in.next().await.unwrap()?;
+            let message = self
+                .ws_in
+                .next()
+                .await
+                .ok_or_else(|| anyhow::anyhow!("Stream ended"))??;
 
             match InMessage::from_ws_message(message) {
                 Ok(in_message) => {