如何确保BatchBlock已经完成

Question

我试图了解如何使用 BatchBlock，因为它似乎是我正在从事的项目所需要的。然而，我特别关注一个部分，那就是如何确保批次完成。以下面的代码为例。它的批量大小为 5。五个值被发布到它

using System;
using System.Linq;
using System.Threading.Tasks.Dataflow;

namespace sandbox
{
    class Program
    {

        public static void Main(string[] args)
        {
            BatchBlock<int> batchBlock = new BatchBlock<int>(5);
            ActionBlock<int[]> actionBlock = new ActionBlock<int[]>(OutputAverage);

            batchBlock.LinkTo(actionBlock, new DataflowLinkOptions() { PropagateCompletion = true });

            batchBlock.Post(1);
            batchBlock.Post(2);
            batchBlock.Post(3);
            batchBlock.Post(4);
            batchBlock.Post(5);

            batchBlock.Complete();
            batchBlock.Completion.Wait();
        }

        private static void OutputAverage(int[] values) =>
            Console.WriteLine("The average is: " + values.Average());

    }
}

如果我运行上面的代码，控制台窗口不会输出任何内容。这是因为程序执行在批处理之前就完成了。如果我在最后添加一个超小的 Thread.Sleep(100); ，那么这足以让它完成批处理并将预期的文本输出到控制台。

我还考虑过其他解决方案，包括经典的 Console.ReadLine() 但最终，我需要它可以被调用并运行，直到所有批次正确完成无需有人观看即可进行处理，然后按 Enter 键将其关闭。

我想要使​​用的实际项目是读取一个大型（> 4GB）二进制编码文件并一次处理块以解析记录。 BatchBlock 部分将用于批量解析记录并将其输出到 JSON 文件，与解析并行。目前一切都按预期工作，除了我的测试文件，这些文件很小并且只包含约 400 条记录，程序在批处理线程完成之前退出。

所以最终，我的问题是，我该怎么做才能确保在程序退出之前所有批次都已通过操作处理完成？

Answer 1

您需要等待管道中的最后一个块，而不是第一个块。只需将其替换为：

batchBlock.Completion.Wait();

... 即可：

actionBlock.Completion.Wait();