src: stats: Add hub messages stats to the protocol

src/stats/actor.rs

@@ -113,6 +113,10 @@ impl StatsActor {
                     let result = self.hub_stats().await;
                     let _ = response.send(result);
                 }
+                StatsCommand::GetHubMessagesStats { response } => {
+                    let result = self.hub_messages_stats().await;
+                    let _ = response.send(result);
+                }
             }
         }
 
@@ -153,6 +157,13 @@ impl StatsActor {
         Ok(hub_stats)
     }
 
+    #[instrument(level = "debug", skip(self))]
+    async fn hub_messages_stats(&self) -> Result<HubMessagesStats> {
+        let hub_messages_stats = self.hub_messages_stats.read().await.clone();
+
+        Ok(hub_messages_stats)
+    }
+
     #[instrument(level = "debug", skip(self))]
     async fn drivers_stats(&mut self) -> Result<DriversStats> {
         let drivers_stats = self.drivers_stats.read().await.clone();

src/stats/protocol.rs

@@ -3,6 +3,8 @@ use tokio::sync::oneshot;
 
 use crate::stats::{DriversStats, StatsInner};
 
+use super::messages::HubMessagesStats;
+
 pub enum StatsCommand {
     SetPeriod {
         period: tokio::time::Duration,
@@ -17,4 +19,7 @@ pub enum StatsCommand {
     GetHubStats {
         response: oneshot::Sender<Result<StatsInner>>,
     },
+    GetHubMessagesStats {
+        response: oneshot::Sender<Result<HubMessagesStats>>,
+    },
 }