Handle close messages

ProjectLighthouseCAU · Jul 17, 2024 · 6791919 · 6791919
1 parent 87d7b14
commit 6791919
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lighthouse-client/src/error.rs b/lighthouse-client/src/error.rs
@@ -19,6 +19,8 @@ pub enum Error {
     Server { code: i32, message: Option<String>, warnings: Vec<String> },
     #[error("No next message available")]
     NoNextMessage,
+    #[error("The connection was closed")]
+    ConnectionClosed,
     #[error("Custom error")]
     Custom(String),
 }

diff --git a/lighthouse-client/src/lighthouse.rs b/lighthouse-client/src/lighthouse.rs
@@ -111,8 +111,8 @@ impl<S> Lighthouse<S>
             let message = ws_stream.next().await.ok_or_else(|| Error::NoNextMessage)??;
             match message {
                 Message::Binary(bytes) => break Ok(bytes),
-                // We ignore pings for now
-                Message::Ping(_) => {},
+                Message::Ping(_) => {}, // Ignore pings for now
+                Message::Close(_) => break Err(Error::ConnectionClosed),
                 _ => warn!("Got non-binary message: {:?}", message),
             }
         }