This is based on Alexey Drobyshevsky's excellent article, "Problems with Iteration". With Alexey's suggestion, I implemented his solution using a ReaderWriterLockSlim. Then I fleshed out a ThreadSafeList<T> : IList<T> implementation.
The idea here is to have a collection that multiple threads can iterate over, execute Linq queries against, and modify at the same time without data inconsistencies, InvalidOperationExceptions, or deadlocks.
using System.Collections;
using System.Collections.Generic;
using System.Threading;
namespace ThreadSafeEnumeration
{
public class SafeEnumerator<T> : IEnumerator<T>
{
private readonly ReaderWriterLockSlim _readerWriterLock;
private readonly IEnumerator<T> _innerCollection;
public SafeEnumerator(IEnumerator<T> innerCollection, ReaderWriterLockSlim readerWriterLock)
{
_innerCollection = innerCollection;
_readerWriterLock = readerWriterLock;
_readerWriterLock.EnterReadLock();
}
public void Dispose()
{
_readerWriterLock.ExitReadLock();
}
public bool MoveNext()
{
return _innerCollection.MoveNext();
}
public void Reset()
{
_innerCollection.Reset();
}
public T Current
{
get { return _innerCollection.Current; }
}
object IEnumerator.Current
{
get { return Current; }
}
}
public class ThreadSafeList<T> : IList<T>
{
private readonly ReaderWriterLockSlim _readerWriterLock = new ReaderWriterLockSlim(LockRecursionPolicy.SupportsRecursion);
private readonly List<T> _innerList = new List<T>();
IEnumerator<T> IEnumerable<T>.GetEnumerator()
{
_readerWriterLock.EnterReadLock();
try
{
return new SafeEnumerator<T>(_innerList.GetEnumerator(), _readerWriterLock);
}
finally
{
_readerWriterLock.ExitReadLock();
}
}
public IEnumerator GetEnumerator()
{
return (this as IEnumerable<T>).GetEnumerator();
}
public void Add(T item)
{
try
{
_readerWriterLock.EnterWriteLock();
_innerList.Add(item);
}
finally
{
_readerWriterLock.ExitWriteLock();
}
}
public void Clear()
{
try
{
_readerWriterLock.EnterWriteLock();
_innerList.Clear();
}
finally
{
_readerWriterLock.ExitWriteLock();
}
}
public bool Contains(T item)
{
try
{
_readerWriterLock.EnterReadLock();
return _innerList.Contains(item);
}
finally
{
_readerWriterLock.EnterReadLock();
}
}
public void CopyTo(T[] array, int arrayIndex)
{
try
{
_readerWriterLock.EnterWriteLock();
_innerList.CopyTo(array, arrayIndex);
}
finally
{
_readerWriterLock.ExitWriteLock();
}
}
public bool Remove(T item)
{
try
{
_readerWriterLock.EnterWriteLock();
return _innerList.Remove(item);
}
finally
{
_readerWriterLock.ExitWriteLock();
}
}
public int Count
{
get
{
try
{
_readerWriterLock.EnterReadLock();
return _innerList.Count;
}
finally
{
_readerWriterLock.ExitReadLock();
}
}
}
bool ICollection<T>.IsReadOnly
{
get
{
try
{
_readerWriterLock.EnterReadLock();
return ((ICollection<T>)_innerList).IsReadOnly;
}
finally
{
_readerWriterLock.ExitReadLock();
}
}
}
public int IndexOf(T item)
{
try
{
_readerWriterLock.EnterReadLock();
return _innerList.IndexOf(item);
}
finally
{
_readerWriterLock.ExitReadLock();
}
}
public void Insert(int index, T item)
{
try
{
_readerWriterLock.EnterWriteLock();
_innerList.Insert(index, item);
}
finally
{
_readerWriterLock.ExitWriteLock();
}
}
public void RemoveAt(int index)
{
try
{
_readerWriterLock.EnterWriteLock();
_innerList.RemoveAt(index);
}
finally
{
_readerWriterLock.ExitWriteLock();
}
}
public T this[int index]
{
get
{
try
{
_readerWriterLock.EnterReadLock();
return _innerList[index];
}
finally
{
_readerWriterLock.ExitReadLock();
}
}
set
{
try
{
_readerWriterLock.EnterWriteLock();
_innerList[index] = value;
}
finally
{
_readerWriterLock.ExitWriteLock();
}
}
}
}
}