close websocket when keepalive are no longer received

whisperfish · Nov 19, 2024 · 9502d4c · 9502d4c
1 parent c6b6eac
commit 9502d4c
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() > 6 {
+                        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())