c# - 等到 BlockingCollection 队列被后台线程清除，如果时间太长则超时？

Question

在 C# 中，我想知道是否可以等到 BlockingCollection 被后台线程清除，如果时间太长则超时。

我现在的临时代码让我觉得有些不雅(从什么时候开始使用 Thread.Sleep 是个好习惯？):

while (_blockingCollection.Count > 0 || !_blockingCollection.IsAddingCompleted)
{
    Thread.Sleep(TimeSpan.FromMilliseconds(20));
    // [extra code to break if it takes too long]
}

Answer 1

您可以在消费线程中使用GetConsumingEnumerable() 和foreach 来确定队列何时为空，然后设置一个ManualResetEvent，它主线程可以检查队列是否为空。 GetConsumingEnumerable() 返回一个枚举器，该枚举器检查 CompleteAdding() 在空队列上终止之前是否已被调用。

示例代码:

using System;
using System.Collections.Concurrent;
using System.Threading;
using System.Threading.Tasks;

namespace Demo
{
    internal class Program
    {
        private void run()
        {
            Task.Run(new Action(producer));
            Task.Run(new Action(consumer));

            while (!_empty.WaitOne(1000))
                Console.WriteLine("Waiting for queue to empty");

            Console.WriteLine("Queue emptied.");
        }

        private void producer()
        {
            for (int i = 0; i < 20; ++i)
            {
                _queue.Add(i);
                Console.WriteLine("Produced " + i);
                Thread.Sleep(100);
            }

            _queue.CompleteAdding();
        }

        private void consumer()
        {
            foreach (int n in _queue.GetConsumingEnumerable())
            {
                Console.WriteLine("Consumed " + n);
                Thread.Sleep(200);
            }

            _empty.Set();
        }

        private static void Main()
        {
            new Program().run();
        }

        private BlockingCollection<int> _queue = new BlockingCollection<int>();

        private ManualResetEvent _empty = new ManualResetEvent(false);
    }
}