Graph: +RemoveNode, +RemoveEdge

Author: gluschenko

src/Wishmaster.Helpers.ContainerDependency.Console/Program.cs

@@ -16,6 +16,9 @@ public static int Main()
             graph.SetEdge(b, c, new EdgeData(EdgeType.ProjectReference, ""));
             graph.SetEdge(c, a, new EdgeData(EdgeType.ProjectReference, ""));
 
+            graph.RemoveNode(b);
+            graph.RemoveEdge(c, a);
+
             return 0;
         }
     }

src/Wishmaster.Helpers.ContainerDependency/Models/Edge.cs

@@ -8,14 +8,12 @@ namespace Wishmaster.Helpers.ContainerDependency.Models
 {
     public class Edge<T>
     {
-        public Guid Uid { get; private set; }
         public Guid From { get; private set; }
         public Guid To { get; private set; }
         public T Value { get; private set; }
 
         public Edge(T value, Guid from, Guid to)
         {
-            Uid = Guid.NewGuid();
             From = from;
             To = to;
             Value = value;
@@ -28,7 +26,7 @@ public void SetValue(T value)
 
         public override int GetHashCode()
         {
-            return HashCode.Combine(Uid, From, To, Value);
+            return HashCode.Combine(From, To, Value);
         }
     }
 }

src/Wishmaster.Helpers.ContainerDependency/Models/Graph.cs

@@ -42,9 +42,31 @@ public Node<TNode> AddNode(Node<TNode> node)
             return node;
         }
 
-        public void RemoveNode()
+        public void RemoveNode(Node<TNode> node)
         {
+            if (node is null)
+            {
+                throw new ArgumentNullException(nameof(node));
+            }
 
+            var identity = new NodeIdentity(node.Uid);
+
+            if (!_nodes.ContainsKey(identity))
+            {
+                throw new Exception($"Node {node.Uid} is not found");
+            }
+
+            var edges = _edges
+                .Where(x => x.Key.From == node.Uid || x.Key.To == node.Uid)
+                .Select(x => x.Key)
+                .ToArray();
+
+            foreach (var edge in edges)
+            {
+                RemoveEdge(edge.From, edge.To);
+            }
+
+            _nodes.Remove(identity);
         }
 
         public void SetEdge(Node<TNode> from, Node<TNode> to, TEdge value)
@@ -68,9 +90,31 @@ public void SetEdge(Guid from, Guid to, TEdge value)
             _edges[identity] = new Edge<TEdge>(value, from, to);
         }
 
-        public void RemoveEdge()
+        public void RemoveEdge(Node<TNode> from, Node<TNode> to)
+        {
+            if (from is null)
+            {
+                throw new ArgumentNullException(nameof(from));
+            }
+
+            if (to is null)
+            {
+                throw new ArgumentNullException(nameof(to));
+            }
+
+            RemoveEdge(from.Uid, to.Uid);
+        }
+
+        public void RemoveEdge(Guid from, Guid to)
         {
+            var identity = new EdgeIdentity(from, to);
+
+            if (!_edges.ContainsKey(identity))
+            {
+                throw new Exception($"Edge {identity} is not found");
+            }
 
+            _edges.Remove(identity);
         }
 
         public override int GetHashCode()