src: hub: Add hub messages stats pub interface

joaoantoniocardoso · Sep 16, 2024 · eba69be · eba69be
1 parent 679a629
commit eba69be
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/hub/mod.rs b/src/hub/mod.rs
@@ -12,7 +12,10 @@ use tokio::sync::{broadcast, mpsc, oneshot, RwLock};
 use crate::{
     drivers::{Driver, DriverInfo},
     protocol::Protocol,
-    stats::driver::{AccumulatedDriverStats, AccumulatedStatsInner},
+    stats::{
+        driver::{AccumulatedDriverStats, AccumulatedStatsInner},
+        messages::AccumulatedHubMessagesStats,
+    },
 };
 
 use actor::HubActor;
@@ -103,6 +106,17 @@ impl Hub {
         Ok(res)
     }
 
+    pub async fn hub_messages_stats(&self) -> Result<AccumulatedHubMessagesStats> {
+        let (response_tx, response_rx) = oneshot::channel();
+        self.sender
+            .send(HubCommand::GetHubMessagesStats {
+                response: response_tx,
+            })
+            .await?;
+        let res = response_rx.await?;
+        Ok(res)
+    }
+
     pub async fn reset_all_stats(&self) -> Result<()> {
         let (response_tx, response_rx) = oneshot::channel();
         self.sender