在添加之前检查字典中是否存在键的最佳方法？

Question

当从字典中获取一个你不确定是否存在的键时，你通常会使用TryGetValue 而不是ContainsKey + get indexer 以避免检查键两次的开销。换句话说，这是：

string password;
if (accounts.TryGetValue(username, out password))
{
    // use the password
}

会更喜欢这个：

if (accounts.ContainsKey(username))
{
    string password = accounts[username];
}

如果我想在将某个键设置为某个值之前检查它是否已经存在，该怎么办？例如，我想在用新密码覆盖之前检查用户名是否存在：

if (!accounts.ContainsKey(username))
{
    accounts.Add(username, password);
}
else
{
    Console.WriteLine("Username is taken!");
}

对

// this doesn't exist
if (!accounts.TrySetValue(username, password))
{
    Console.WriteLine("Username is taken!");
}

有没有比ContainsKey 和Add 更性能 的替代方案？

Answer 1

如果您不想覆盖，我认为最好编写自己的扩展方法，例如TryGetValue。没有标准方法。

或

使用 CuncurrentDictionary，它有 TryAdd 方法，但同步会有开销。

所以，简单的回答——不，没有这样的方法。

Answer 2

我倾向于根据需要编写自己的扩展。

例如，GetValueOrDefault 是这样的：

public static V GetValueOrDefault<K, V>(this IDictionary<K, V> @this, K key, Func<V> @default)
{
    return @this.ContainsKey(key) ? @this[key] : @default();
}

可以这样使用：

var password = accounts.GetValueOrDefault(username, () => null);
if (password != null)
{
    //do stuff
}

或SetValueIfExists:

public static V SetValueIfExists<K, V>(this IDictionary<K, V> @this, K key, V value)
{
    if (@this.ContainsKey(key))
    {
        @this[key] = value;
    }
}

或SetValueIfNotExists:

public static V SetValueIfNotExists<K, V>(this IDictionary<K, V> @this, K key, V value)
{
    if (!@this.ContainsKey(key))
    {
        @this[key] = value;
    }
}

Answer 3

如果您认为插入新名称将是常见情况，而尝试插入重复名称将是极少数情况，您可能只想使用捕获异常的开销。

try
{
    accounts.Add(username, password);
}
catch (ArgumentException)
{
    Console.WriteLine("Username is taken!");
}

如果您使用现有密钥调用Add，则会抛出ArgumentException。即使您经常重复，这仍然可能比您的ContainsKey 检查更有效。

Answer 4

我知道我迟到了，但是您可以使用一个技巧并在索引器设置之前存储计数并在索引器设置之后检查计数。如果计数相同，则您已覆盖键，否则您已添加新映射：

public static bool AddOrUpdate<TKey, TValue>(this IDictionary<TKey, TValue> 
    dictionary, TKey key, TValue value)
{
    var countBefore = dictionary.Count;
    dictionary[key] = value;
    return countBefore != dictionary.Count;
}