Question

我有一个流程生成工作，第二个流程使用BlockingCollection<>消耗该工作。当我关闭我的程序时，我需要我的消费者停止消费工作，但我仍然需要快速记录待处理但尚未消耗的工作。

现在，我的消费者产生了一个具有foreach (<object> in BlockingCollection.GetConsumingEnumerable())循环的线程。当我停止我的程序时，我的制作人会调用Consumer.BlockingCollection.CompleteAdding()。我发现我的消费者继续处理队列中的所有内容。

谷歌搜索问题告诉我，我需要使用CancellationToken。所以我试了一下：

private void Process () { // This method runs in a separate thread
    try {
        foreach (*work* in BlockingCollection.GetConsumingEnumerable(CancellationToken)) {
            // Consume
        }
    }
    catch (OperationCancelledException) {
        foreach (*work* in BlockingCollection.GetConsumingEnumerable()) {
            // quickly log
        }
    }
}

我的制作人有：

private CancellationTokenSource StopFlag = new CancellationTokenSource ();
MyConsumer.CancellationToken = StopFlag.Token;
// Make the consumer spawn it's consuming thread...
StopFlag.Cancel ();
MyConsumer.BlockingCollection.CompleteAdding ();

当我尝试这个时，我没有发现OperationCancelledException曾经发生过。

This question尝试解释使用取消令牌，但似乎没有正确使用它。（论证：如果它有效，那么它“足够正确”。）而this question似乎与我的问题完全相同，但没有例子。（相同here。）

重申一下：如何在CancellationToken上正确使用BlockingCollection.GetConsumingEnumerable()时需要在使用其他方法取消队列后处理队列中的其余项目时需要注意？

（我认为我的问题集中在正确使用CancellationToken。我的测试都没有表明该过程实际上被取消。（StopFlag.IsCancellationRequested总是等于false。））

Answer 1

当您将CancellationToken传递给GetConsumingEnumerable时，它不会引发取消请求的例外，它只会停止吐出项目。而不是捕获异常只需检查令牌：

foreach (var item in BlockingCollection.
    GetConsumingEnumerable(CancellationToken))
{
    //consume item
}
if (CancellationToken.IsCancellationRequested)
    foreach (var item in BlockingCollection)
    {
        //log item
    }

另请注意，如果请求取消，并且可能未调用CompletedAdding，那么您应该只是迭代集合，而不是调用GetConsumingEnumerable。如果您知道生产者将在取消操作时完成添加，那么这不是问题。

Answer 2

我的问题在于我是如何尝试取消操作的。我没有让我的制作人拥有CancellationTokenSource，而是把它全部放在消费者手中。

public class cProducer {
    private cConsumer myConsumer = new cConsumer ();

    public void onStart () {
        myConsumer.OnStart ();
    }

    public void onStop () {
        myConsumer.OnStop ();
    }

    public void OnOrderReceived (cOrder newOrder) {
        myConsumer.orderQueue.Add (cOrder);
    }
}

public class cConsumer {
    private CancellationTokenSource stopFlag;
    public BlockingCollection<cOrder> orderQueue = new BlockingCollection<cOrder> ();
    private Task processingTask;

    public void OnStart () {
        stopFlag = new CancellationTokenSource ();
        processingTask = Task.Factory.StartNew (() => Process ());
    }

    public void OnStop () {
        stopFlag.Cancel ();
        orderQueue.CompleteAdding ();
        processingTask.Wait ();
    }

    private void Process () {
        try {
            foreach (cOrder newOrder in orderQueue.GetConsumingEnumerable (stopFlag.Token)) {
                // process
            }
        }
        catch (OperationCanceledException) {
            foreach (cOrder cancelledOrder in orderQueue.GetConsumingEnumerable ()) {
                // log it
            }
        }
    }
}

取消BlockingCollection.GetConsumingEnumerable（）并处理剩下的内容

2 个答案: