ws socket worker: replace unwrap with error return

This commit is contained in:
Joakim Frostegård 2023-01-03 18:27:56 +01:00
parent 8bf0e12cb4
commit 14a801f4d4

View file

@ -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) => {