feat: Add create new object if not exists when add to directions

Jandini · Mar 26, 2021 · a4892b5 · a4892b5
1 parent 612e99c
commit a4892b5
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/src/RightTurn/ITurnDirections.cs b/src/RightTurn/ITurnDirections.cs
@@ -2,6 +2,7 @@
 {
     public interface ITurnDirections
     {
+        T Add<T>() where T : new();
         T Add<T>(T value);
         T Get<T>();
         T TryGet<T>();

diff --git a/src/RightTurn/TurnDirections.cs b/src/RightTurn/TurnDirections.cs
@@ -4,13 +4,23 @@
 namespace RightTurn
 {
     internal class TurnDirections : Dictionary<Type, object>, ITurnDirections
-    {       
+    {      
         public T Add<T>(T value)
         {
             this[typeof(T)] = value;
             return value;
         }
 
+        public T Add<T>() where T : new()
+        {
+            if (ContainsKey(typeof(T)))
+                return (T)this[typeof(T)];
+
+            var value = new T();
+            this[typeof(T)] = value;
+            return value;
+        }
+
         public bool Have<T>()
         {
             return ContainsKey(typeof(T));
@@ -32,7 +42,6 @@ public T TryGet<T>()
                 return default;
 
             return (T)this[typeof(T)];
-        }
-
+        }      
     }
 }