session count on server

ShadowsocksR-Live · May 15, 2024 · f1c82ff · f1c82ff
1 parent a9f748c
commit f1c82ff
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -5,6 +5,7 @@ nginx_signing.key
 overtls-daemon.sh
 project.xcworkspace/
 xcuserdata/
+.vs/
 .vscode/
 .VSCodeCounter/
 .env

diff --git a/src/server.rs b/src/server.rs
@@ -83,6 +83,8 @@ pub async fn run_server(config: &Config, exiting_flag: crate::CancellationToken)
         }
     };
 
+    let session_count = Arc::new(std::sync::atomic::AtomicUsize::new(0));
+
     loop {
         tokio::select! {
             _ = exiting_flag.cancelled() => {
@@ -105,10 +107,16 @@ pub async fn run_server(config: &Config, exiting_flag: crate::CancellationToken)
                     Ok::<_, Error>(())
                 };
 
+                let session_count = session_count.clone();
+
                 tokio::spawn(async move {
+                    let count = session_count.fetch_add(1, std::sync::atomic::Ordering::SeqCst) + 1;
+                    log::debug!("{}: {} sessions", peer_addr, count);
                     if let Err(e) = incoming_task.await {
                         log::debug!("{peer_addr}: {e}");
                     }
+                    let count = session_count.fetch_sub(1, std::sync::atomic::Ordering::SeqCst) - 1;
+                    log::debug!("{}: {} sessions", peer_addr, count);
                 });
             }
         }