From 5cecc9f0bea3a5c6794182e98348c0b2b377887c Mon Sep 17 00:00:00 2001
From: Sergio Benitez <sb@sergio.bz>
Date: Thu, 9 May 2024 22:49:28 -0700
Subject: [PATCH] Detect and handle MessageStream WebSocket closure.

---
 contrib/ws/src/websocket.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/contrib/ws/src/websocket.rs b/contrib/ws/src/websocket.rs
index 6557a293..36155044 100644
--- a/contrib/ws/src/websocket.rs
+++ b/contrib/ws/src/websocket.rs
@@ -278,6 +278,7 @@ impl<'r, S> IoHandler for MessageStream<'r, S>
         rocket::tokio::pin!(stream);
         while let Some(msg) = stream.next().await {
             let result = match msg {
+                Ok(msg) if msg.is_close() => return Ok(()),
                 Ok(msg) => sink.send(msg).await,
                 Err(e) => Err(e)
             };