- fix: panic on network client disconnect before actually logging in

This commit is contained in:
2024-12-21 18:27:34 -05:00
parent f55ca79410
commit 7c8e5a52c6

View File

@@ -66,9 +66,13 @@ async fn handle_connection(stream: &mut TcpStream, pool: Arc<BufferPool>, auth_c
} }
if let Some(state) = connection_service.get_connection(&connection_id) { if let Some(state) = connection_service.get_connection(&connection_id) {
let session_id = state.session_id.unwrap(); let session_id = state.session_id.unwrap_or_default();
let mut auth_client = auth_client.lock().await; if !session_id.is_empty() {
auth_client.logout(&session_id).await?; let mut auth_client = auth_client.lock().await;
auth_client.logout(&session_id).await?;
} else {
warn!("No session found for {}", stream.peer_addr()?);
}
} }
ACTIVE_CONNECTIONS.dec(); ACTIVE_CONNECTIONS.dec();
Ok(()) Ok(())