nik/chorus
1
0
Fork 0
mirror of https://github.com/mikedilger/chorus.git synced 2026-05-03 06:51:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Log number of active websockets

Browse Source
This commit is contained in:
Mike Dilger 2024-02-19 11:06:26 +13:00
parent 65070109ac
commit 87f8624075
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/main.rs
View File

@ -233,8 +233,6 @@ async fn handle_http_request(
// Await the websocket upgrade process // Await the websocket upgrade process
match websocket.await { match websocket.await {
Ok(websocket) => { Ok(websocket) => {
log::info!("{}: websocket started", peer);
// Build a websocket service // Build a websocket service
let mut ws_service = WebSocketService { let mut ws_service = WebSocketService {
peer, peer,
@ -247,7 +245,13 @@ async fn handle_http_request(
}; };
// Increment count of active websockets // Increment count of active websockets
let _ = GLOBALS.num_clients.fetch_add(1, Ordering::SeqCst); let old_num_websockets = GLOBALS.num_clients.fetch_add(1, Ordering::SeqCst);
log::info!(
"{}: websocket started (making {} active websockets)",
peer,
old_num_websockets + 1
);
// Handle the websocket // Handle the websocket
if let Err(e) = ws_service.handle_websocket_stream().await { if let Err(e) = ws_service.handle_websocket_stream().await {
@ -264,9 +268,13 @@ async fn handle_http_request(
} }
// DecrementIncrement count of active websockets // DecrementIncrement count of active websockets
let _ = GLOBALS.num_clients.fetch_sub(1, Ordering::SeqCst); let old_num_websockets = GLOBALS.num_clients.fetch_sub(1, Ordering::SeqCst);
log::info!("{}: websocket ended", peer); log::info!(
"{}: websocket ended (making {} active websockets)",
peer,
old_num_websockets - 1
);
} }
Err(e) => { Err(e) => {
log::error!("{}", e); log::error!("{}", e);