我应该如何在C#中编写生产者/消费者代码?

时间:2010-09-02 16:14:39

标签: c# multithreading

我有1个线程流数据和第2个(线程池)处理数据。数据处理大约需要100毫秒,因此我使用第二个线程,所以不要占用第一个线程。

当第二个线程处理数据时,第一个线程将数据添加到字典缓存中,然后当第二个线程完成时,它处理缓存的值。

我的问题是如何用C#做生产者/消费者代码?

public delegate void OnValue(ulong value);

public class Runner
{
    public event OnValue OnValueEvent;
    private readonly IDictionary<string, ulong> _cache = new Dictionary<string, ulong>(StringComparer.InvariantCultureIgnoreCase);
    private readonly AutoResetEvent _cachePublisherWaitHandle = new AutoResetEvent(true);

    public void Start()
    {
        for (ulong i = 0; i < 500; i++)
        {
            DataStreamHandler(i.ToString(), i);
        }
    }

    private void DataStreamHandler(string id, ulong value)
    {
        _cache[id] = value;

        if (_cachePublisherWaitHandle.WaitOne(1))
        {
            IList<ulong> tempValues = new List<ulong>(_cache.Values);
            _cache.Clear();

            _cachePublisherWaitHandle.Reset();

            ThreadPool.UnsafeQueueUserWorkItem(delegate
            {
                try
                {
                    foreach (ulong value1 in tempValues)
                        if (OnValueEvent != null)
                            OnValueEvent(value1);
                }
                finally
                {
                    _cachePublisherWaitHandle.Set();
                }
            }, null);
        }
        else
        {
            Console.WriteLine(string.Format("Buffered value: {0}.", value));
        }
    }
}

class Program
{
    static void Main(string[] args)
    {
        Stopwatch sw = Stopwatch.StartNew();
        Runner r = new Runner();
        r.OnValueEvent += delegate(ulong x)
                              {
                                  Console.WriteLine(string.Format("Processed value: {0}.", x));
                                  Thread.Sleep(100);

                                  if(x == 499)
                                  {
                                      sw.Stop();
                                      Console.WriteLine(string.Format("Time: {0}.", sw.ElapsedMilliseconds));
                                  }
                              };
        r.Start();
        Console.WriteLine("Done");
        Console.ReadLine();
    }
}

2 个答案:

答案 0 :(得分:4)

设置生产者 - 消费者模式的最佳做法是使用.NET 4.0中提供的BlockingCollection类或单独下载Reactive Extensions框架。这个想法是生产者将使用Add方法排队,消费者将使用Take方法出列队列,如果队列为空,则会阻塞。就像SwDevMan81指出的那样,如果你想要手动路线,Albahari网站上有一篇关于如何让它正常工作的非常好的文章。

答案 1 :(得分:3)

MSDN上有一个关于同步Producer和Consumer的好articleAlbahari网站上还有一个很好的例子。