如何在BlockingCollection上取消GetConsumingEnumerable()

时间:2012-02-02 02:36:31

标签: c# cancellation blockingcollection

在下面的代码中,当生产者没有生成时,我正在使用CancellationToken来唤醒GetConsumingEnumerable(),我想要脱离foreach并退出Task。但我没有看到IsCancellationRequested被记录,我的Task.Wait(timeOut)等待整个timeOut期间。我做错了什么?

userToken.Task = Task.Factory.StartNew(state =>
{
    userToken.CancelToken = new CancellationTokenSource();

    foreach (var broadcast in userToken.BroadcastQueue.GetConsumingEnumerable(userToken.CancelToken.Token))
    {
        if (userToken.CancelToken.IsCancellationRequested)
        {
            Log.Write("BroadcastQueue IsCancellationRequested");
            break;
            ...
        }
    }

    return 0;
}, "TaskSubscribe", TaskCreationOptions.LongRunning);

...后

UserToken.CancelToken.Cancel();          
try
{
    task.Wait(timeOut);
}
catch (AggregateException ar)
{
    Log.Write("AggregateException " + ar.InnerException, MsgType.InfoMsg);
}
catch (OperationCanceledException)
{
    Log.Write("BroadcastQueue Cancelled", MsgType.InfoMsg);
}

2 个答案:

答案 0 :(得分:10)

您可以使用CompleteAdding()表示不再向该集合添加任何项目。如果使用GetConsumingEnumerable,foreach将优雅地结束,因为它会知道等待更多项目没有意义。

基本上,一旦你完成了向BlockingCollection添加项目的操作,就行了:     myBlockingCollection.CompleteAdding()

任何使用GetConsumingEnumerable进行foreach循环的线程都将停止循环。

答案 1 :(得分:5)

我已经创建了快速原型,它似乎对我有用。

注意在令牌取消请求之前的Thread.Sleep(1000)。 您可能正在为Token变量创建race condition,因为您在不同的线程中创建并访问item.CancelToken变量。

E.g。用于取消任务的代码可能会在错误(先前或空)取消令牌上调用取消。

static void Main(string[] args)
{
    CancellationTokenSource token = null;
    BlockingCollection<string> coll = new BlockingCollection<string>();
    var t = Task.Factory.StartNew(state =>
    {
        token = new CancellationTokenSource();
        try
        {
            foreach (var broadcast in coll.GetConsumingEnumerable(token.Token))
            {
                if (token.IsCancellationRequested)
                {
                    return;
                }
            }
        }
        catch (OperationCanceledException)
        {
            Console.WriteLine("Cancel");
            return;
        }
    }, "TaskSubscribe", TaskCreationOptions.LongRunning);
    Thread.Sleep(1000);
    token.Cancel();
    t.Wait();
}