Merge pull request #447 from joelverhagen/tryadd

Add TryAdd method to Dictionary
yanghuan · Dec 28, 2023 · f581a50 · f581a50
2 parents c8867b9 + 70207c4
commit f581a50
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/CSharp.lua/CoreSystem.Lua/CoreSystem/Collections/Dictionary.lua b/CSharp.lua/CoreSystem.Lua/CoreSystem/Collections/Dictionary.lua
@@ -356,6 +356,15 @@ local Dictionary = {
     end
     return remove(this, key)
   end,
+  TryAdd = function (this, key, value)
+    if key == nil then throw(ArgumentNullException("key")) end
+    local exists, currentValue = this:TryGetValue(key)
+    if exists then
+      return false
+    end
+    this:set(key, value)
+    return true
+  end,
   TryGetValue = function (this, key)
     if key == nil then throw(ArgumentNullException("key")) end
     local value = this[key]
@@ -633,6 +642,15 @@ local ArrayDictionary = (function ()
       end
       return false
     end,
+    TryAdd = function (this, key, value)
+      if key == nil then throw(ArgumentNullException("key")) end
+      local exists, currentValue = this:TryGetValue(key)
+      if exists then
+        return false
+      end
+      this:set(key, value)
+      return true
+    end,
     TryGetValue = function (this, key)
       if key == nil then throw(ArgumentNullException("key")) end
       local len = #this