Merge pull request #200 from kytos-ng/enhacement/link

Optimized `_get_link_or_create`
kytos-ng · May 21, 2024 · 695b3b5 · 695b3b5
2 parents 968845c + 8c6195b
commit 695b3b5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -95,9 +95,8 @@ def _get_link_or_create(self, endpoint_a, endpoint_b):
         """
         new_link = Link(endpoint_a, endpoint_b)
 
-        for link in self.links.values():
-            if new_link == link:
-                return (link, False)
+        if new_link.id in self.links:
+            return (self.links[new_link.id], False)
 
         self.links[new_link.id] = new_link
         return (new_link, True)