src: stats: Add hub stats to the protocol

joaoantoniocardoso · Sep 16, 2024 · 928410e · 928410e
1 parent 9f41887
commit 928410e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/stats/actor.rs b/src/stats/actor.rs
@@ -80,6 +80,10 @@ impl StatsActor {
                     let result = self.drivers_stats().await;
                     let _ = response.send(result);
                 }
+                StatsCommand::GetHubStats { response } => {
+                    let result = self.hub_stats().await;
+                    let _ = response.send(result);
+                }
             }
         }
 

diff --git a/src/stats/protocol.rs b/src/stats/protocol.rs
@@ -1,7 +1,7 @@
 use anyhow::Result;
 use tokio::sync::oneshot;
 
-use crate::stats::DriversStats;
+use crate::stats::{DriversStats, StatsInner};
 
 pub enum StatsCommand {
     SetPeriod {
@@ -14,4 +14,7 @@ pub enum StatsCommand {
     GetDriversStats {
         response: oneshot::Sender<Result<DriversStats>>,
     },
+    GetHubStats {
+        response: oneshot::Sender<Result<StatsInner>>,
+    },
 }