检查某个项目是否存在于“阻止收集”中的最佳方法是什么?在尝试添加新的之前?基本上我不希望将重复项添加到BlockingCollection中。
答案 0 :(得分:5)
您必须实现自己的IProducerConsumerCollection<T>
,其行为类似于一组(例如,不允许重复)。这是一个简单的版本,它使用一个关键部分(C#lock
)来使其成为线程安全的。对于高并发性方案,您可以使用类似SpinWait
的类来提高性能,与ConcurrentQueue<T>
的方式相同。
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
。这会导致InvalidOperationException
方法抛出BlockingCollection<T>.Add
,因此您可能需要包装代码以将项添加到以下内容中:
bool AddItem<T>(BlockingCollection<T> blockingCollection, T item) {
try {
blockingCollection.Add(item);
return true;
}
catch (InvalidOperationException) {
return false;
}
}
请注意,如果您将项目添加到已完成的集合中,您还将获得InvalidOperationException
,并且您必须检查异常消息以确定异常的根本原因。
答案 1 :(得分:-2)
使用TryAdd(data)
方法。您还可以传入timespan
对象或int
表示超时时间。返回true
或false
。
请注意,如果基础集合类型无法处理重复项,并且您尝试添加的数据是重复的,则会引发InvalidOperationException
。