fix: cleanup

This commit is contained in:
2026-02-12 11:48:35 -08:00
parent b614def39d
commit 385352c2de

View File

@@ -2,6 +2,7 @@ use crate::{AppState, error::Error, storage::StorageImpl};
use actix_web::{HttpRequest, HttpResponse, rt, web}; use actix_web::{HttpRequest, HttpResponse, rt, web};
use actix_ws::AggregatedMessage; use actix_ws::AggregatedMessage;
use futures_util::StreamExt; use futures_util::StreamExt;
use tokio::sync::broadcast::error::RecvError;
pub async fn peers( pub async fn peers(
req: HttpRequest, req: HttpRequest,
@@ -22,7 +23,7 @@ pub async fn peers(
return Ok(res); return Ok(res);
} }
tracing::info!( tracing::info!(
"sent initial peer list ({} peers) to new WebSocket client", "sent initial peer list ({} peers) to new client",
initial_peers.len() initial_peers.len()
); );
} }
@@ -67,20 +68,20 @@ pub async fn peers(
if session.text(json).await.is_err() { if session.text(json).await.is_err() {
break; 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)) => { Err(RecvError::Lagged(n)) => {
tracing::warn!("WebSocket client lagged, missed {} updates", n); tracing::warn!("client lagged, missed {} updates", n);
} }
Err(tokio::sync::broadcast::error::RecvError::Closed) => { Err(RecvError::Closed) => {
break; break;
} }
} }
} }
} }
} }
let _ = session.close(None).await; session.close(None).await.ok();
tracing::info!("WebSocket client disconnected"); tracing::info!("client disconnected");
}); });
Ok(res) Ok(res)