refactor: Simplify PortGraph::port_links (#188)

CQCL · Feb 24, 2025 · 93a3a05 · 93a3a05
1 parent b0c0cfc
commit 93a3a05
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/src/portgraph.rs b/src/portgraph.rs
@@ -712,14 +712,9 @@ impl LinkView for PortGraph {
 
     fn port_links(&self, port: PortIndex) -> impl Iterator<Item = (PortIndex, PortIndex)> + Clone {
         self.port_meta_valid(port).unwrap();
-        match self.port_link[port.index()] {
-            Some(link) => std::iter::once((port, link)),
-            None => {
-                let mut iter = std::iter::once((port, port));
-                iter.next();
-                iter
-            }
-        }
+        self.port_link[port.index()]
+            .map(|link| (port, link))
+            .into_iter()
     }
 
     #[inline]