src: stats: Add hub messages stats pub interface

joaoantoniocardoso · Sep 17, 2024 · 5193155 · 5193155
1 parent 294c4e1
commit 5193155
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/stats/mod.rs b/src/stats/mod.rs
@@ -7,6 +7,7 @@ use std::sync::{Arc, Mutex};
 
 use accumulated::AccumulatedStatsInner;
 use anyhow::Result;
+use messages::HubMessagesStats;
 use tokio::sync::{mpsc, oneshot};
 
 use actor::StatsActor;
@@ -109,6 +110,16 @@ impl Stats {
         response_rx.await?
     }
 
+    pub async fn hub_messages_stats(&mut self) -> Result<HubMessagesStats> {
+        let (response_tx, response_rx) = oneshot::channel();
+        self.sender
+            .send(StatsCommand::GetHubMessagesStats {
+                response: response_tx,
+            })
+            .await?;
+        response_rx.await?
+    }
+
     pub async fn set_period(&mut self, period: tokio::time::Duration) -> Result<()> {
         let (response_tx, response_rx) = oneshot::channel();
         self.sender