bugfix delay remove

sj-distributor · Oct 21, 2024 · cec25c0 · cec25c0
1 parent 02754e8
commit cec25c0
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;
         }