From d5cb5f1ffc8c1a44997a1be19ff49032af526510 Mon Sep 17 00:00:00 2001
From: Tony Giorgio <tonygiorgio@protonmail.com>
Date: Tue, 21 Feb 2023 12:16:55 -0600
Subject: [PATCH] Handle websocket error

---
 src/lib.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 28d31c9..3a8db52 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -142,6 +142,10 @@ pub async fn main(req: Request, env: Env, _ctx: Context) -> Result<Response> {
                 let mut event_stream = server.events().expect("stream error");
                 console_log!("spawned event stream, waiting for first message..");
                 while let Some(event) = event_stream.next().await {
+                    if let Err(e) = event {
+                        console_log!("error parsing some event: {e}");
+                        continue;
+                    }
                     match event.expect("received error in websocket") {
                         WebsocketEvent::Message(msg) => {
                             if msg.text().is_none() {
@@ -221,6 +225,7 @@ pub async fn main(req: Request, env: Env, _ctx: Context) -> Result<Response> {
                         }
                         WebsocketEvent::Close(_) => {
                             console_log!("closing");
+                            break;
                         }
                     }
                 }