src: hub: Add hub messages stats pub interface

joaoantoniocardoso · Sep 17, 2024 · d92b449 · d92b449
1 parent eb83a0a
commit d92b449
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::accumulated::{driver::AccumulatedDriverStats, AccumulatedStatsInner},
+    stats::accumulated::{
+        driver::AccumulatedDriverStats, messages::AccumulatedHubMessagesStats,
+        AccumulatedStatsInner,
+    },
 };
 
 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