制作快照并清空 ConcurrentDictionary

Question

我有一个要清空的 ConcurrentDictionary，将其内容复制到字典（或任何其他容器）中，同时清空它。

一个关键的不变性是在操作过程中不会丢失源中的任何元素。

我当前的代码如下所示：

    private Dictionary<TKey, TValue> Foo<TKey, TValue>(ConcurrentDictionary<TKey, TValue> source)
    {
        var result = new Dictionary<TKey, TValue>();
        foreach (var item in source)
        {
            result[item.Key] = item.Value;
        }
        source.Clear();
        return result;
    }

据我了解，这段代码是线程安全的，但在 foreach 循环之后和 Clear() 之前同时添加的任何元素都将被清除。

编辑：更精确一些。在我的用例中，该代码是唯一删除某些字典键的代码；其他线程只使用 TryAdd 或 AddOrUpdate。

Answer 1

我能想到的唯一方法是TryRemove所有现有的键，一个一个：

public static Dictionary<TKey, TValue> RemoveAll<TKey, TValue>(
    this ConcurrentDictionary<TKey, TValue> source)
{
    var result = new Dictionary<TKey, TValue>();
    foreach (var key in source.Keys)
    {
        if (source.TryRemove(key, out var value)) result.Add(key, value);
    }
    return result;
}

Keys 属性返回ConcurrentDictionary 中键的快照，因此不可能包含两次相同的键。但它会产生争用，因为它acquires all internal locks。下面是一个使用 ConcurrentDictionary 的惰性枚举器的版本，因此它不会产生太多争用：

public static Dictionary<TKey, TValue> RemoveAll<TKey, TValue>(
    this ConcurrentDictionary<TKey, TValue> source)
{
    var result = new Dictionary<TKey, TValue>();
    foreach (var entry in source)
    {
        var (key, value) = entry;
        if (!result.ContainsKey(key) && source.TryRemove(key, out value))
            result.Add(key, value);
    }
    return result;
}

!result.ContainsKey(key) 的原因是枚举器产生两次相同键的（理论上）可能性。基于ConcurrentDictionary、this cannot happen的当前实现。