Merge pull request #24 from sj-distributor/Enhancement-Can-update-mem…

…ory-cache bugfix delay remove
sj-distributor · Oct 21, 2024 · ffee204 · ffee204
2 parents 2d40751 + cec25c0
commit ffee204
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/FastCache.InMemory/Extension/DictExtension.cs b/FastCache.InMemory/Extension/DictExtension.cs
@@ -12,7 +12,14 @@ public static bool TryRemove<TKey, TValue>(this ConcurrentDictionary<TKey, TValu
             var res = dictionary.TryRemove(key, out value);
             if (res && seconds > 0)
             {
-                Task.Delay(TimeSpan.FromSeconds(seconds)).ContinueWith(_ => { dictionary.TryRemove(key, out var _); });
+                Task.Run(async () =>
+                {
+                    await Task.Delay(TimeSpan.FromSeconds(seconds));
+                    lock (dictionary)
+                    {
+                        dictionary.TryRemove(key, out var _);
+                    }
+                });
             }
             return res;
         }