From 385352c2dec1ef898b90db87ca840aab91aa1673 Mon Sep 17 00:00:00 2001 From: lucalise Date: Thu, 12 Feb 2026 11:48:35 -0800 Subject: [PATCH] fix: cleanup --- registry/endpoints/ws/peers.rs | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/registry/endpoints/ws/peers.rs b/registry/endpoints/ws/peers.rs index d0dc779..daaacb0 100644 --- a/registry/endpoints/ws/peers.rs +++ b/registry/endpoints/ws/peers.rs @@ -2,6 +2,7 @@ use crate::{AppState, error::Error, storage::StorageImpl}; use actix_web::{HttpRequest, HttpResponse, rt, web}; use actix_ws::AggregatedMessage; use futures_util::StreamExt; +use tokio::sync::broadcast::error::RecvError; pub async fn peers( req: HttpRequest, @@ -22,7 +23,7 @@ pub async fn peers( return Ok(res); } tracing::info!( - "sent initial peer list ({} peers) to new WebSocket client", + "sent initial peer list ({} peers) to new client", initial_peers.len() ); } @@ -67,20 +68,20 @@ pub async fn peers( if session.text(json).await.is_err() { break; } - tracing::info!("sent peer update to WebSocket client: {}", peer_update.peer.public_key); + tracing::info!("sent peer update to client: {}", peer_update.peer.public_key); } - Err(tokio::sync::broadcast::error::RecvError::Lagged(n)) => { - tracing::warn!("WebSocket client lagged, missed {} updates", n); + Err(RecvError::Lagged(n)) => { + tracing::warn!("client lagged, missed {} updates", n); } - Err(tokio::sync::broadcast::error::RecvError::Closed) => { + Err(RecvError::Closed) => { break; } } } } } - let _ = session.close(None).await; - tracing::info!("WebSocket client disconnected"); + session.close(None).await.ok(); + tracing::info!("client disconnected"); }); Ok(res)