非阻塞并发收集?

时间:2010-07-19 14:35:38

标签: c# multithreading collections nonblocking

System.Collections.Concurrent有一些在多线程环境中运行良好的新集合。但是,它们有点受限。它们会阻塞,直到某个项目可用,或者它们返回default(T)(TryXXX方法)。

我需要一个线程安全的集合,但它不是阻塞调用线程,而是使用回调通知我至少有一个项目可用。

我目前的解决方案是使用BlockingCollection,但要将APM与委托一起使用以获取下一个元素。换句话说,我创建了一个Take来自集合的方法的委托,并使用BeginInvoke执行该委托。

不幸的是,为了实现这一目标,我必须在课堂上保持很多状态。更糟糕的是,这个类不是线程安全的;它只能由单个线程使用。我正在避开可维护性的边缘,我不想这样做。

我知道有一些库可以让我在这里做的很简单(我相信Reactive Framework就是其中之一),但是我想在没有添加版本4之外的任何引用的情况下实现我的目标框架。

我可以使用哪种更好的模式,不需要外部引用来实现我的目标吗?


TL; DR:

是否有任何模式满足要求:

“我需要发信号通知我已准备好接收下一个元素的集合,并让集合在下一个元素到达时执行回调,而不会阻塞任何线程。”

2 个答案:

答案 0 :(得分:4)

我想我有两种可能的解决方案。我对这两者都不是特别满意,但他们至少提供了APM方法的合理替代方案。

第一个不满足你没有阻塞线程的要求,但我认为它相当优雅,因为你可以注册回调,它们会以循环方式被调用,但你仍然可以调用{{1}像往常一样TakeTryTake。此代码强制每次请求项时都注册回调。这是集合的信令机制。关于这种方法的好处是,对BlockingCollection的调用不会像我在第二个解决方案中那样被饿死。

Take

第二个确实符合您的无阻塞线程的要求。注意它如何将回调的调用传递给线程池。我之所以这样做,是因为我认为如果它同步执行,那么锁会保持更长时间,从而导致public class NotifyingBlockingCollection<T> : BlockingCollection<T> { private Thread m_Notifier; private BlockingCollection<Action<T>> m_Callbacks = new BlockingCollection<Action<T>>(); public NotifyingBlockingCollection() { m_Notifier = new Thread(Notify); m_Notifier.IsBackground = true; m_Notifier.Start(); } private void Notify() { while (true) { Action<T> callback = m_Callbacks.Take(); T item = Take(); callback.BeginInvoke(item, null, null); // Transfer to the thread pool. } } public void RegisterForTake(Action<T> callback) { m_Callbacks.Add(callback); } } Add的瓶颈。我仔细研究了一下,我不认为它可以实时锁定(项目和回调都可用,但回调永远不会被执行)但你可能想要自己查看来验证。这里唯一的问题是,对RegisterForTake的调用会因为回调始终优先而受到匮乏。

Take

答案 1 :(得分:3)

这样的事情怎么样? (命名可能会使用一些工作。请注意,这是未经测试的。)

public class CallbackCollection<T>
{
    // Sychronization object to prevent race conditions.
    private object _SyncObject = new object();

    // A queue for callbacks that are waiting for items.
    private ConcurrentQueue<Action<T>> _Callbacks = new ConcurrentQueue<Action<T>>();

    // A queue for items that are waiting for callbacks.
    private ConcurrentQueue<T> _Items = new ConcurrentQueue<T>();

    public void Add(T item)
    {
        Action<T> callback;
        lock (_SyncObject)
        {
            // Try to get a callback. If no callback is available,
            // then enqueue the item to wait for the next callback
            // and return.
            if (!_Callbacks.TryDequeue(out callback))
            {
                _Items.Enqueue(item);
                return;
            }
        }

        ExecuteCallback(callback, item);
    }

    public void TakeAndCallback(Action<T> callback)
    {
        T item;
        lock(_SyncObject)
        {
            // Try to get an item. If no item is available, then
            // enqueue the callback to wait for the next item
            // and return.
            if (!_Items.TryDequeue(out item))
            {
                _Callbacks.Enqueue(callback);
                return;
            }
        }
        ExecuteCallback(callback, item);
    }

    private void ExecuteCallback(Action<T> callback, T item)
    {
        // Use a new Task to execute the callback so that we don't
        // execute it on the current thread.
        Task.Factory.StartNew(() => callback.Invoke(item));
    }
}