add function to get the distance between routers

nsg-ethz · Oct 7, 2024 · 0c2d8d5 · 0c2d8d5
1 parent d780bda
commit 0c2d8d5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/bgpsim/Cargo.toml b/bgpsim/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "bgpsim"
-version = "0.17.6"
+version = "0.17.7"
 edition = "2021"
 license-file = "LICENSE"
 description = "A network control-plane simulator"

diff --git a/bgpsim/src/event/rand_queue.rs b/bgpsim/src/event/rand_queue.rs
@@ -308,6 +308,11 @@ impl<P: Prefix> GeoTimingModel<P> {
         self.processing_params.insert(router, params);
     }
 
+    /// Get the distance between two routers
+    pub fn get_distance(&mut self, src: RouterId, dst: RouterId) -> Option<f64> {
+        self.distances.get(&(src, dst)).map(|x| x.into_inner())
+    }
+
     /// Set the distance between two nodes in light seconds
     pub fn set_distance(&mut self, src: RouterId, dst: RouterId, dist: f64) {
         let dist = NotNan::new(dist).unwrap();