implement storage + improve code

server/src/broker/subscriber.rs

@@ -2,21 +2,23 @@ use rumqttd::{local::LinkRx, Broker, Notification};
 use common::{StatusMessage, MQTT_TOPIC};
 use std::sync::Arc;
 
-use crate::bridge::ClientManager;
+use crate::{bridge::ClientManager, storage::StorageRepository};
 
 pub struct MqttSubscriber {
     link_rx: LinkRx,
     clients: Arc<ClientManager>,
+    storage: Arc<dyn StorageRepository>,
 }
 
 impl MqttSubscriber {
-    pub fn new(broker: &Broker, clients: Arc<ClientManager>) -> Self {
+    pub fn new(broker: &Broker, clients: Arc<ClientManager>, storage: Arc<dyn StorageRepository>) -> Self {
         let (mut link_tx, link_rx) = broker.link("internal-subscriber").unwrap();
         link_tx.subscribe(MQTT_TOPIC).unwrap();
 
         Self {
             link_rx,
             clients,
+            storage,
         }
     }
 
@@ -27,6 +29,10 @@ impl MqttSubscriber {
 
                 if let Ok(payload_str) = payload.to_string() {
                     self.clients.broadcast(payload_str).await;
+
+                    if let Err(e) = self.storage.record_message(&payload.agent).await {
+                        eprintln!("failed to record message for {}: {}", &payload.agent, e);
+                    }
                 }
             }
         }