src: hub: Add hub messages stats to the protocol

joaoantoniocardoso · Sep 16, 2024 · 679a629 · 679a629
1 parent 607a3ac
commit 679a629
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/hub/actor.rs b/src/hub/actor.rs
@@ -54,6 +54,10 @@ impl HubActor {
                     let hub_stats = self.get_hub_stats().await;
                     let _ = response.send(hub_stats);
                 }
+                HubCommand::GetHubMessagesStats { response } => {
+                    let hub_messages_stats = self.get_hub_messages_stats().await;
+                    let _ = response.send(hub_messages_stats);
+                }
                 HubCommand::ResetAllStats { response } => {
                     let _ = response.send(self.reset_all_stats().await);
                 }

diff --git a/src/hub/protocol.rs b/src/hub/protocol.rs
@@ -6,7 +6,10 @@ use tokio::sync::{broadcast, oneshot};
 use crate::{
     drivers::{Driver, DriverInfo},
     protocol::Protocol,
-    stats::driver::{AccumulatedDriverStats, AccumulatedStatsInner},
+    stats::{
+        driver::{AccumulatedDriverStats, AccumulatedStatsInner},
+        messages::AccumulatedHubMessagesStats,
+    },
 };
 
 pub enum HubCommand {
@@ -27,6 +30,9 @@ pub enum HubCommand {
     GetHubStats {
         response: oneshot::Sender<AccumulatedStatsInner>,
     },
+    GetHubMessagesStats {
+        response: oneshot::Sender<AccumulatedHubMessagesStats>,
+    },
     GetDriversStats {
         response: oneshot::Sender<Vec<(String, AccumulatedDriverStats)>>,
     },