sync locations from discovery (#524)

Author: JannikSt

crates/orchestrator/src/discovery/monitor.rs

@@ -338,6 +338,19 @@ impl DiscoveryMonitor {
                     node.ip_address = discovery_node.node.ip_address.clone();
                     let _ = self.store_context.node_store.add_node(node.clone()).await;
                 }
+                if existing_node.location.is_none() && discovery_node.location.is_some() {
+                    info!(
+                        "Node {} location changed from None to {:?}",
+                        node_address, discovery_node.location
+                    );
+                    if let Some(location) = &discovery_node.location {
+                        let _ = self
+                            .store_context
+                            .node_store
+                            .update_node_location(&node_address, location)
+                            .await;
+                    }
+                }
 
                 if existing_node.status == NodeStatus::Dead {
                     if let (Some(last_change), Some(last_updated)) = (

crates/orchestrator/src/store/domains/node_store.rs

@@ -7,6 +7,7 @@ use log::info;
 use redis::AsyncCommands;
 use redis::Value;
 use shared::models::heartbeat::TaskDetails;
+use shared::models::node::NodeLocation;
 use shared::models::task::TaskState;
 use std::sync::Arc;
 
@@ -145,6 +146,21 @@ impl NodeStore {
         Ok(())
     }
 
+    pub async fn update_node_location(
+        &self,
+        node_address: &Address,
+        location: &NodeLocation,
+    ) -> Result<()> {
+        let mut con = self.redis.client.get_multiplexed_async_connection().await?;
+        let node_key: String = format!("{}:{}", ORCHESTRATOR_BASE_KEY, node_address);
+        let node_string: String = con.get(&node_key).await?;
+        let mut node: OrchestratorNode = serde_json::from_str(&node_string)?;
+        node.location = Some(location.clone());
+        let node_string = node.to_string();
+        let _: () = con.set(&node_key, node_string).await?;
+        Ok(())
+    }
+
     pub async fn update_node_p2p_id(&self, node_address: &Address, p2p_id: &str) -> Result<()> {
         let mut con = self.redis.client.get_multiplexed_async_connection().await?;
         let node_key: String = format!("{}:{}", ORCHESTRATOR_BASE_KEY, node_address);