C# 使用 BlockingCollection
原创
©著作权归作者所有:来自51CTO博客作者biyusr216的原创作品,请联系作者获取转载授权,否则将追究法律责任
现在介绍管道的第一阶段。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 方法获取阻塞集合的枚举器,而不是直接选代集合。