C# 使用 BlockingCollection

最新推荐文章于 2024-02-15 15:40:55 发布

zls365365

最新推荐文章于 2024-02-15 15:40:55 发布

阅读量541

点赞数

文章标签： c# 数据库 java python 编程语言

原文链接：https://mp.weixin.qq.com/s?__biz=MzI0NDE0MDgyNg==&mid=2649596613&idx=1&sn=1869c419854e6b3ec837c6fece07c1c7&chksm=f17b5cb0c60cd5a6fc30370ef8e5c23ef6361814efd8f668cd947bb7cfeb47a4d1d85579efd9&scene=126&&sessionid=0

版权

现在介绍管道的第一阶段。ReadFilenamesAsync 接收 BlockingCollection<T>为参数，在其中写入输出。该方法的实现使用枚举器来迭代指定目录及其子目录中的C# 文件。这些文件的文件名用 Add 方法添加到 BlockingCollection<T>中。完成添加文件名的操作后，调用 CompleteAdding 方法，以通知所有读取器不应再等待集合中的任何额外项：

public static class PipelineStages
{
  public static Task ReadFilenamesAsync(string path, 
    BlockingCollection<string> output)
  {
    return Task.Factory.StartNew(() =>
    {
      foreach (string filename in Directory.EnumerateFiles(path,"*.cs",
        SearchOption.AllDirectories))
      {
        output.Add(filename);
        ColoredConsole.WriteLine($"stage 1: added {filename}");
      }
      output.CompleteAdding();
    }，TaskCreationOptions.LongRunning);
  }
  //...

注意

如果在写入器添加项的同时，读取器从 BlockingCollection<T>中读取，那么调用CompleteAdding 方法是很重要的。否则，读取器会在 foreach 循环中等待更多的项被添加。

下一个阶段是读取文件并将其内容添加到另一个集合中，这由 LoadContentAsync 方法完成。该方法使用了输入集合传递的文件名，打开文件，然后把文件中的所有行添加到输出集合中。在 foreach 循环中，用输入阻塞集合调用GetConsumingEnumerable，以迭代各项。直接使用 input 变量而不调用GetConsumingEnumerable 是可以的，但是这只会迭代当前状态的集合，而不会迭代以后添加的项。

public static async Task LoadContentAsync(BlockingCollection<string> input, 
  BlockingCollection<string> output)
{
  foreach (var filename in input.GetConsumingEnumerable()) 
  {
    using (FileStream stream = File.OpenRead(filename))
    {
      var reader = new StreamReader(stream); 
      string line = null;
      while ((line = await reader.ReadLineAsync()) != null)
      {
        output.Add(line);
        ColoredConsole.WriteLine($"stage 2: added {line}");
      }
    }
  }
  output.CompleteAdding();
}

注意

如果在填充集合的同时，使用读取器读取集合，则需要使用GetConsumingEnumerable 方法获取阻塞集合的枚举器，而不是直接选代集合。