是否存在类似ConcurrentDictionary与Interlocked.Exchange的原子键值替换数据结构？

内置类型的局限 #

ConcurrentDictionary<TKey,TValue>的AddOrUpdate系列方法确实无法直接满足需求：它的逻辑是“键不存在则添加，存在则更新”，且更新逻辑依赖委托回调，无法原子性地直接替换值并返回旧值；而Interlocked.Exchange<T>仅能操作单个变量，无法关联字典的键值对。

自定义实现方案 #

可以基于ConcurrentDictionary和Interlocked封装一个满足需求的原子字典：

方案1：使用原子引用包装类

通过ConcurrentDictionary<TKey, AtomicReference<TValue>>存储键对应的原子引用，结合Interlocked.Exchange实现原子替换：

public class AtomicDictionary<TKey, TValue> where TKey : notnull
{
    private readonly ConcurrentDictionary<TKey, AtomicReference<TValue>> _innerDict = new();

    // 原子性设置新值并返回旧值，键不存在时默认值为TValue的默认值
    public TValue Exchange(TKey key, TValue newValue)
    {
        // 原子获取或创建键对应的原子引用
        var atomicRef = _innerDict.GetOrAdd(key, _ => new AtomicReference<TValue>(default));
        // 原子替换值并返回旧值
        return Interlocked.Exchange(ref atomicRef.Value, newValue);
    }
}

// 辅助类：用于包装值类型/引用类型，支持Interlocked操作
public class AtomicReference<T>
{
    public T Value;

    public AtomicReference(T initialValue)
    {
        Value = initialValue;
    }
}

方案2：针对非blittable值类型的适配

如果TValue是非blittable值类型（比如自定义结构体），Interlocked.Exchange无法直接操作，可改用Interlocked.CompareExchange配合循环实现原子替换：

public TValue ExchangeNonBlittable(TKey key, TValue newValue)
{
    var atomicRef = _innerDict.GetOrAdd(key, _ => new AtomicReference<TValue>(default));
    TValue oldValue;
    do
    {
        oldValue = atomicRef.Value;
    } while (!ReferenceEquals(Interlocked.CompareExchange(ref atomicRef.Value, newValue, oldValue), oldValue));
    return oldValue;
}

注意事项 #

• 如果需要在键不存在时返回特定标识（而非默认值），可修改AtomicReference的初始值，或在Exchange方法中结合_innerDict.TryGetValue做额外判断（需注意原子性边界）。
• 对于引用类型，上述方案天然支持；对于值类型，需确保类型满足Interlocked操作的要求，或使用包装类适配。

内容的提问来源于stack exchange，提问作者Kyle McClellan