.NET Dictionary: nhận hoặc tạo mới

Question

Tôi thường thấy mình tạo ra một Dictionary với một lớp giá trị không tầm thường (ví dụ: List), và sau đó luôn viết cùng một mẫu mã khi điền dữ liệu.

Ví dụ:

var dict = new Dictionary<string, List<string>>(); 
string key = "foo"; 
string aValueForKey = "bar"; 

Đó là, tôi muốn chèn "bar" vào danh sách tương ứng với chìa khóa "foo", nơi chính "foo" có thể không được ánh xạ tới bất cứ điều gì.

Đây là nơi mà tôi sử dụng mô hình không ngừng lặp đi lặp lại:

List<string> keyValues; 
if (!dict.TryGetValue(key, out keyValues)) 
    dict.Add(key, keyValues = new List<string>()); 
keyValues.Add(aValueForKey); 

Có một cách thanh lịch hơn để làm điều này?

câu hỏi

liên quan mà không có câu trả lời cho câu hỏi này:

• Is there an IDictionary implementation that returns null on missing key instead of throwing?
• Find-or-insert with only one lookup in c# dictionary
• Dictionary returning a default value if the key does not exist

Answer 1

Chúng tôi có một hơi khác nhau về vấn đề này, nhưng hiệu quả là tương tự:

public static TValue GetOrCreate<TKey, TValue>(this IDictionary<TKey, TValue> dict, TKey key) 
    where TValue : new() 
{ 
    TValue val; 

    if (!dict.TryGetValue(key, out val)) 
    { 
     val = new TValue(); 
     dict.Add(key, val); 
    } 

    return val; 
} 

gọi:

var dictionary = new Dictionary<string, List<int>>(); 

List<int> numbers = dictionary.GetOrCreate("key"); 

Nó làm cho việc sử dụng hạn chế chung cho nhà xây dựng parameterless công cộng: where TValue : new().

Để giúp phát hiện, trừ khi phương pháp mở rộng là khá cụ thể cho một vấn đề hẹp, chúng ta có xu hướng đặt phương pháp khuyến nông trong không gian tên của các loại họ đang mở rộng, trong trường hợp này:

namespace System.Collections.Generic 

Hầu hết các thời gian, người sử dụng loại có tuyên bố using được xác định ở trên cùng, vì vậy IntelliSense cũng sẽ tìm thấy các phương pháp mở rộng cho nó được xác định trong mã của bạn.

Answer 2

Như với rất nhiều vấn đề lập trình, khi bạn thấy mình đang làm một cái gì đó rất nhiều, cấu trúc lại nó thành một phương pháp:

public static void MyAdd<TKey, TCollection, TValue>(
    this Dictionary<TKey, TCollection> dictionary, TKey key, TValue value) 
    where TCollection : ICollection<TValue>, new() 
{ 
    TCollection collection; 
    if (!dictionary.TryGetValue(key, out collection)) 
    { 
     collection = new TCollection(); 
     dictionary.Add(key, collection); 
    } 
    collection.Add(value); 
} 

Answer 3

Ok, cách tiếp cận khác nhau:

public static bool TryAddValue<TKey,TValue>(this System.Collections.Generic.IDictionary<TKey,List<TValue>> dictionary, TKey key, TValue value) 
    { 
     // Null check (useful or not, depending on your null checking approach) 
     if (value == null) 
      return false; 

     List<TValue> tempValue = default(List<TValue>); 

     try 
     { 
      if (!dictionary.TryGetValue(key, out tempValue)) 
      { 
       dictionary.Add(key, tempValue = new List<TValue>()); 
      } 
      else 
      { 
       // Double null check (useful or not, depending on your null checking approach) 
       if (tempValue == null) 
       { 
        dictionary[key] = (tempValue = new List<TValue>()); 
       } 
      } 

      tempValue.Add(value); 
      return true; 
     } 
     catch 
     { 
      return false; 
     } 
    } 

Bằng cách này bạn có để "cố gắng thêm" giá trị của bạn vào Danh sách chung chung (rõ ràng là có thể khái quát chung cho một bộ sưu tập chung), hãy kiểm tra và cố gắng lấy khóa/giá trị hiện có trong Từ điển của bạn. Cách sử dụng và ví dụ:

var x = new Dictionary<string,List<string>>(); 
x.TryAddValue("test", null); // return false due to null value. Doesn't add the key 
x.TryAddValue("test", "ok"); // it works adding the key/value 
x.TryAddValue("test", "ok again"); // it works adding the value to the existing list 

Hy vọng điều đó sẽ hữu ích.

Answer 4

Và còn điều này thì sao?

var keyValues = dictionary[key] = dictionary.ContainsKey(key) ? dictionary[key] : new List<string>(); 
keyValues.Add(aValueForKey);