refactor: include pings in WebSocket activity (#821)

BoltzExchange · Feb 13, 2025 · 8e69f5b · 8e69f5b
1 parent 050f97f
commit 8e69f5b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/boltzr/src/api/ws/status.rs b/boltzr/src/api/ws/status.rs
@@ -228,11 +228,16 @@ where
                                 }
                             },
                             Message::Ping(payload) => {
+                                last_activity = Instant::now();
+
                                 if let Err(err) = ws_sender.send(Message::Pong(payload)).await {
                                     trace!("Could not respond to ping: {}", err);
                                     break;
                                 }
                             },
+                            Message::Pong(_) => {
+                                last_activity = Instant::now();
+                            },
                             Message::Close(_) => {
                                 break;
                             },