如何使用多个条件过滤Directory.EnumerateFiles？

Z时代
2024-01-10
分类：问答

我有以下代码：

List<string> result = new List<string>();
foreach (string file in Directory.EnumerateFiles(path,"*.*",  
      SearchOption.AllDirectories)
      .Where(s => s.EndsWith(".mp3") || s.EndsWith(".wma")))
       {
          result.Add(file);                 
       }

它工作正常，可以满足我的需求。除了一件小事。我想找到一种更好的方法来过滤多个扩展。我想使用带有这样的过滤器的字符串数组：

string[] extensions = { "*.mp3", "*.wma", "*.mp4", "*.wav" };

使用NET Framework 4.0 / LINQ进行此操作的最有效方法是什么？有什么建议么？

我非常感谢偶尔的程序员提供的帮助:-)

回答：

我创建了一些辅助方法来解决这个问题，这是我在今年早些时候写的博客。

一个版本使用regex模式\.mp3|\.mp4，另一个版本使用字符串列表并并行运行。

public static class MyDirectory
{   // Regex version
   public static IEnumerable<string> GetFiles(string path, 
                       string searchPatternExpression = "",
                       SearchOption searchOption = SearchOption.TopDirectoryOnly)
   {
      Regex reSearchPattern = new Regex(searchPatternExpression, RegexOptions.IgnoreCase);
      return Directory.EnumerateFiles(path, "*", searchOption)
                      .Where(file =>
                               reSearchPattern.IsMatch(Path.GetExtension(file)));
   }
   // Takes same patterns, and executes in parallel
   public static IEnumerable<string> GetFiles(string path, 
                       string[] searchPatterns, 
                       SearchOption searchOption = SearchOption.TopDirectoryOnly)
   {
      return searchPatterns.AsParallel()
             .SelectMany(searchPattern => 
                    Directory.EnumerateFiles(path, searchPattern, searchOption));
   }
}