refactor: include pings in WebSocket activity

BoltzExchange · Feb 13, 2025 · 40b107a · 40b107a
1 parent e76e74c
commit 40b107a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/boltzr/src/api/ws/status.rs b/boltzr/src/api/ws/status.rs
@@ -228,6 +228,8 @@ 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;
@@ -301,6 +303,13 @@ where
                 },
             }
         }
+
+        if let Err(err) = ws_sender.send(Message::Close(None)).await {
+            trace!("Could not send close message: {}", err);
+        };
+        if let Err(err) = ws_sender.close().await {
+            trace!("Could not close WebSocket: {}", err);
+        };
     }
 
     async fn handle_message(