Add stat_set_num function and implement latency tracking in client_inner

geph-official · Sep 18, 2024 · c189a1a · c189a1a
1 parent 54c4746
commit c189a1a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/binaries/geph5-client/src/client_inner.rs b/binaries/geph5-client/src/client_inner.rs
@@ -36,7 +36,7 @@ use crate::{
     client::CtxField,
     control_prot::{ConnectedInfo, CURRENT_CONN_INFO},
     route::{deprioritize_route, get_dialer},
-    stats::stat_incr_num,
+    stats::{stat_incr_num, stat_set_num},
     vpn::{fake_dns_backtranslate, vpn_whitelist},
     ConnInfo,
 };
@@ -236,7 +236,9 @@ async fn client_inner(ctx: AnyCtx<Config>, authed_pipe: impl Pipe) -> anyhow::Re
                 let mux = mux.clone();
                 let ctx = ctx.clone();
                 let (remote_addr, send_back) = ctx.get(CONN_REQ_CHAN).1.lock().await.recv().await?;
-
+                if let Some(latency) = mux.last_latency() {
+                    stat_set_num(&ctx, "ping", latency.as_secs_f64());
+                }
                 spawn!(async move {
                     tracing::debug!(remote_addr = display(&remote_addr), "opening tunnel");
                     let stream = mux.open(remote_addr.as_bytes()).await;

diff --git a/binaries/geph5-client/src/stats.rs b/binaries/geph5-client/src/stats.rs
@@ -12,12 +12,12 @@ use crate::{client::CtxField, Config};
 
 static NUM_STATS: CtxField<DashMap<SmolStr, AtomicF64>> = |_| DashMap::new();
 
-// pub fn stat_set_num(ctx: &AnyCtx<Config>, stat: &str, num: f64) {
-//     ctx.get(NUM_STATS)
-//         .entry(stat.into())
-//         .or_default()
-//         .store(num, Ordering::Relaxed);
-// }
+pub fn stat_set_num(ctx: &AnyCtx<Config>, stat: &str, num: f64) {
+    ctx.get(NUM_STATS)
+        .entry(stat.into())
+        .or_default()
+        .store(num, Ordering::Relaxed);
+}
 
 pub fn stat_incr_num(ctx: &AnyCtx<Config>, stat: &str, num: f64) {
     ctx.get(NUM_STATS)