Added dictionary extensions

2025-07-16 10:13:33 +01:00
parent a268050c8d
commit 38b1f648ce
1 changed files with 37 additions and 0 deletions
@@ -0,0 +1,37 @@
+namespace SPTarkov.Server.Core.Extensions
+{
+    public static class DictionaryExtensions
+    {
+        /// <summary>
+        /// Add a value by key to a dictionary, if the key doesn't exist, create it
+        /// </summary>
+        /// <typeparam name="T">Dictionary key type</typeparam>
+        /// <param name="dict">Dictionary to add/update</param>
+        /// <param name="key">Key to update by</param>
+        /// <param name="value">Value to add to key</param>
+        public static void AddOrUpdate<T>(this IDictionary<T, double> dict, T key, double value)
+            where T : notnull
+        {
+            if (!dict.TryAdd(key, value))
+            {
+                dict[key] += value;
+            }
+        }
+
+        /// <summary>
+        /// Add a value by key to a dictionary, if the key doesn't exist, create it
+        /// </summary>
+        /// <typeparam name="T">Dictionary key type</typeparam>
+        /// <param name="dict">Dictionary to add/update</param>
+        /// <param name="key">Key to update by</param>
+        /// <param name="value">Value to add to key</param>
+        public static void AddOrUpdate<T>(this IDictionary<T, int> dict, T key, int value)
+            where T : notnull
+        {
+            if (!dict.TryAdd(key, value))
+            {
+                dict[key] += value;
+            }
+        }
+    }
+}