From e682a3b0e0566943a01fa01745886d47fba616b4 Mon Sep 17 00:00:00 2001
From: Mike Dilger <mike@mikedilger.com>
Date: Mon, 19 Feb 2024 20:41:25 +1300
Subject: [PATCH] Try to close the websocket when ending, maybe we weren't
 closing it nicely, let's see

---
 src/main.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 7fbc8c7..b7f8f1e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -346,7 +346,12 @@ impl WebSocketService {
                     match message_option {
                         Some(message) => {
                             let message = message?;
-                            self.handle_websocket_message(message).await?;
+                            if let Err(e) = self.handle_websocket_message(message).await {
+                                if let Err(e) = self.websocket.close(None).await {
+                                    log::info!("Err on websocket close: {e}");
+                                }
+                                return Err(e);
+                            }
                         },
                         None => break, // the websocket is closed
                     }