Add live

1 files changed, 12 insertions, 0 deletions

diff --git a/structure/Router.java b/structure/Router.java
index d977471..fce5977 100644
--- a/structure/Router.java
+++ b/structure/Router.java
@@ -15,7 +15,15 @@ public class Router {
 		this.name=""+id;
 	}
 
+	public void resetLinks(){
+		this.links=new HashMap<>();
+	}
+	
 	public void buildLink(Router router, int weight){
+		this.links.remove(router);
+		router.removeLink(this);
+
+		
 		this.links.put(router, weight);
 		router.addLink(this, weight);
 	}
@@ -32,6 +40,10 @@ public class Router {
 		return links;
 	}
 
+	public void removeLink(Router router){
+		this.links.remove(router);
+	}
+	
 	public void setLinks(HashMap<Router, Integer> links) {
 		this.links = links;
 	}