re-introduce some special dictionary methods

sebas77 · Jun 4, 2022 · d753000 · d753000
1 parent a8e52bc
commit d753000
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/DataStructures/Dictionaries/FasterDictionary.cs b/DataStructures/Dictionaries/FasterDictionary.cs
@@ -197,6 +197,12 @@ public bool Remove(TKey key)
             return _dictionary.Remove(key);
         }
 
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public bool Remove(TKey key, out TValue val)
+        {
+            return _dictionary.Remove(key, out _, out val);
+        }
+
         [MethodImpl(MethodImplOptions.AggressiveInlining)]
         public void Trim()
         {

diff --git a/DataStructures/Dictionaries/ThreadSafeDictionary.cs b/DataStructures/Dictionaries/ThreadSafeDictionary.cs
@@ -307,6 +307,20 @@ public bool TryRemove(TKey key)
                 _lockQ.ExitWriteLock();
             }
         }
+
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public bool TryRemove(TKey key, out TValue val)
+        {
+            _lockQ.EnterWriteLock();
+            try
+            {
+                return _dict.Remove(key, out val);
+            }
+            finally
+            {
+                _lockQ.ExitWriteLock();
+            }
+        }
 
         [MethodImpl(MethodImplOptions.AggressiveInlining)]
         public void Trim()