如何取消BlockingCollection上的GetConsumingEnumerable()

本文关键字：BlockingCollection 上的 GetConsumingEnumerable 取消何取消 | 更新日期: 2023-09-27 18:22:16

在下面的代码中，当生产者不生产时，我使用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);
}

如何取消BlockingCollection上的GetConsumingEnumerable()

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

基本上，一旦你完成了向BlockingCollection添加项目，只需执行：

myBlockingCollection.CompleteAdding()

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

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

注意线程。在令牌取消请求之前立即休眠（1000）。您可能正在为Token变量创建竞争条件，因为您在不同的线程中创建和访问item.CancelToken变量。

例如，打算取消任务的代码可能会在错误的（以前的或空的）取消令牌上调用取消。

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();
}