DictionaryExtensions.cs

Nixill · Dec 12, 2024 · 8119dd4 · 8119dd4
1 parent bfefd06
commit 8119dd4
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/CSharp.Nixill/src/Utils/Extensions/DictionaryExtensions.cs b/CSharp.Nixill/src/Utils/Extensions/DictionaryExtensions.cs
@@ -0,0 +1,20 @@
+namespace Nixill.Utils.Extensions;
+
+public static class DictionaryExtensions
+{
+  public static V GetOrSet<K, V>(this IDictionary<K, V> dictionary, K key, V value)
+    => dictionary.GetOrSet(key, () => value);
+
+  public static V GetOrSet<K, V>(this IDictionary<K, V> dictionary, K key, Func<V> value)
+  {
+    V returnValue;
+
+    if (!dictionary.TryGetValue(key, out returnValue!))
+    {
+      returnValue = value();
+      dictionary[key] = returnValue;
+    }
+
+    return returnValue;
+  }
+}