feat(mgmt): add iterators for peerings

Signed-off-by: Fredi Raspall <[email protected]>

Author: Fredi-raspall

mgmt/src/rpc/overlay/tests.rs

@@ -37,6 +37,7 @@ pub mod test {
         vpc1.add_expose(expose).expect("Should succeed");
         vpc1
     }
+
     /* build sample peering between VPC-1 and VPC-2 */
     fn build_vpc_peering() -> VpcPeering {
         // build vpc manifests
@@ -140,4 +141,55 @@ pub mod test {
         assert_eq!(overlay.validate(), Ok(()));
         println!("{overlay:#?}");
     }
+
+    #[test]
+    fn test_peering_iter() {
+        let mut peering_table = VpcPeeringTable::new();
+
+        let m1 = VpcExposeManifest::new("VPC-1");
+        let m2 = VpcExposeManifest::new("VPC-2");
+        let mut peering = VpcPeering::new("Peering-1");
+        peering.set_one(m1);
+        peering.set_two(m2);
+        peering_table.add(peering).unwrap();
+
+        let m1 = VpcExposeManifest::new("VPC-1");
+        let m2 = VpcExposeManifest::new("VPC-3");
+        let mut peering = VpcPeering::new("Peering-2");
+        peering.set_one(m1);
+        peering.set_two(m2);
+        peering_table.add(peering).unwrap();
+
+        let m1 = VpcExposeManifest::new("VPC-2");
+        let m2 = VpcExposeManifest::new("VPC-4");
+        let mut peering = VpcPeering::new("Peering-3");
+        peering.set_one(m1);
+        peering.set_two(m2);
+        peering_table.add(peering).unwrap();
+
+        let m1 = VpcExposeManifest::new("VPC-1");
+        let m2 = VpcExposeManifest::new("VPC-4");
+        let mut peering = VpcPeering::new("Peering-4");
+        peering.set_one(m1);
+        peering.set_two(m2);
+        peering_table.add(peering).unwrap();
+
+        // all peerings of VPC-1
+        let x: Vec<String> = peering_table
+            .peerings_vpc("VPC-1")
+            .map(|p| p.name.clone())
+            .collect();
+
+        assert!(x.contains(&"Peering-1".to_owned()));
+        assert!(x.contains(&"Peering-2".to_owned()));
+        assert!(x.contains(&"Peering-4".to_owned()));
+        assert!(!x.contains(&"Peering-3".to_owned()), "not there");
+
+        // all peerings of VPC-2 with VPC-4
+        let x: Vec<String> = peering_table
+            .peerings_between("VPC-2", "VPC-4")
+            .map(|p| p.name.clone())
+            .collect();
+        assert!(x.contains(&"Peering-3".to_owned()));
+    }
 }

mgmt/src/rpc/overlay/vpcpeering.rs

@@ -115,4 +115,31 @@ impl VpcPeeringTable {
     pub fn values(&self) -> impl Iterator<Item = &VpcPeering> {
         self.0.values()
     }
+    /// Produce iterator of [`VpcPeering`]s that involve the vpc with the provided name
+    pub fn peerings_vpc(&self, vpc: &str) -> impl Iterator<Item = &VpcPeering> {
+        self.0.values().filter(|peering| {
+            // VPCs are options to ease builders but should always be there
+            let name1 = peering.vpc1.as_ref().map(|m| m.name.as_str());
+            let name2 = peering.vpc2.as_ref().map(|m| m.name.as_str());
+            if name1.is_none() || name2.is_none() {
+                false
+            } else {
+                name1 == Some(vpc) || name2 == Some(vpc)
+            }
+        })
+    }
+    /// Produce iterator of [`VpcPeering`]s between the two provided vpcs.
+    /// In principle there should be one peering at the most between two vpcs ?
+    pub fn peerings_between(&self, vpc1: &str, vpc2: &str) -> impl Iterator<Item = &VpcPeering> {
+        self.0.values().filter(|peering| {
+            let name1 = peering.vpc1.as_ref().map(|m| m.name.as_str());
+            let name2 = peering.vpc2.as_ref().map(|m| m.name.as_str());
+            if name1.is_none() || name2.is_none() {
+                false
+            } else {
+                name1 == Some(vpc1) && name2 == Some(vpc2)
+                    || name1 == Some(vpc2) && name2 == Some(vpc1)
+            }
+        })
+    }
 }