检查“阻止收集”中的重复项

Question

在尝试添加新项目之前检查BlockingCollection中是否存在项目的最佳方法是什么？基本上我不希望将重复项添加到BlockingCollection中。

Answer 1

您必须实现自己的行为类似于集合的IProducerConsumerCollection<T>（例如，不允许重复）。这是一个使用关键部分（C# lock）的简单版本，以使其线程安全。对于高并发方案，可以通过像ConcurrentQueue<T>一样使用类似SpinWait类来提高性能。

public class ProducerConsumerSet<T> : IProducerConsumerCollection<T> {
  readonly object gate = new object();
  readonly Queue<T> queue = new Queue<T>();
  readonly HashSet<T> hashSet = new HashSet<T>();
  public void CopyTo(T[] array, int index) {
    if (array == null)
      throw new ArgumentNullException("array");
    if (index < 0)
      throw new ArgumentOutOfRangeException("index");
    lock (gate)
      queue.CopyTo(array, index);
  }
  public bool TryAdd(T item) {
    lock (gate) {
      if (hashSet.Contains(item))
        return false;
      queue.Enqueue(item);
      hashSet.Add(item);
      return true;
    }
  }
  public bool TryTake(out T item) {
    lock (gate) {
      if (queue.Count == 0) {
        item = default(T);
        return false;
      }
      item = queue.Dequeue();
      hashSet.Remove(item);
      return true;
    }
  }
  public T[] ToArray() {
    lock (gate)
      return queue.ToArray();
  }
  public void CopyTo(Array array, int index) {
    if (array == null)
      throw new ArgumentNullException("array");
    lock (gate)
      ((ICollection) queue).CopyTo(array, index);
  }
  public int Count {
    get { return queue.Count; }
  }
  public object SyncRoot {
    get { return gate; }
  }
  public bool IsSynchronized {
    get { return true; }
  }
  public IEnumerator<T> GetEnumerator() {
    List<T> list = null;
    lock (gate)
      list = queue.ToList();
    return list.GetEnumerator();
  }
  IEnumerator IEnumerable.GetEnumerator() {
    return GetEnumerator();
  }
}

如果需要，您可以详细说明此类以通过提供可选IEqualityComparer<T>来自定义相等性，然后使用该初始化HashSet<T>。

当尝试插入重复项时，IProducerConsumerCollection<T>.Add方法返回false。这会导致 BlockingCollection<T>.Add 方法引发InvalidOperationException，因此您可能需要包装代码才能将项目添加到如下所示的内容中：

bool AddItem<T>(BlockingCollection<T> blockingCollection, T item) {
  try {
    blockingCollection.Add(item);
    return true;
  }
  catch (InvalidOperationException) {
    return false;
  }
}

请注意，如果将项添加到已完成的集合中，则还将获得InvalidOperationException，并且必须检查异常消息以确定异常的根本原因。

Answer 2

使用TryAdd(data)方法。还可以传入指示超时期限的timespan对象或 int。返回true或false。请注意，如果基础集合类型无法处理重复项，并且您尝试添加的数据是重复项，则会引发InvalidOperationException。