Merge pull request #347 from fraukappayuwe/patch-1

Close websocket when keepalive are no longer received
whisperfish · Nov 20, 2024 · e6affcc · e6affcc
2 parents 268e0c4 + e73644b
commit e6affcc
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/websocket/mod.rs b/src/websocket/mod.rs
@@ -201,6 +201,14 @@ impl SignalWebSocketProcess {
             futures::select! {
                 _ = ka_interval.tick().fuse() => {
                     use prost::Message;
+                    if self.outgoing_keep_alive_set.len() > 0 {
+                        tracing::warn!("Websocket will be closed due to failed keepalives.");
+                        if let Err(e) = self.ws.close(reqwest_websocket::CloseCode::Away, None).await {
+                            tracing::debug!("Could not close WebSocket: {:?}", e);
+                        }
+                        self.outgoing_keep_alive_set.clear();
+                        break;
+                    }
                     tracing::debug!("sending keep-alive");
                     let request = WebSocketRequestMessage::new(Method::GET)
                         .id(self.next_request_id())