c# - 如何使用 'Directory.getFiles' 获取磁盘上具有特定扩展名的所有文件并将它们保存在列表中

Question

我正在做一个控制台项目，其目标是在整个磁盘中搜索扩展名为“.config”的所有文件

我试过类似的方法:

foreach (string file in Directory.GetFiles("C:\\", "*.config", SearchOption.AllDirectories))
  {
   Console.WriteLine(file);
   Console.ReadLine();
}

但给了我一个错误“拒绝访问路径 (...)”。

我在网上找到这段代码:

Stack<string> pending = new Stack<string>();
        pending.Push("C:\\");

        while (pending.Count != 0)
        {
            var path = pending.Pop();
            string[] next = null;

            try
            {
                next = Directory.GetFiles(path, "*.config");
            }
            catch { }

            if (next != null && next.Length != 0)
                foreach (var file in next)
                {
                    Console.WriteLine(file);
                    Console.ReadLine();
                }
            try
            {
                next = Directory.GetDirectories(path);
                foreach (var subdir in next) pending.Push(subdir);
            }
            catch { }
        }

但它只显示总是在“输入”中单击的路径，我想将这些文件/路径保存在列表中。

有人可以帮忙吗？

Answer 1

您可以做两件事来改进该代码:

1. 使用Directory.EnumerateFiles()和 Directory.EnumerateDirectories()以避免复制每个目录中所有文件的名称。
2. 将方法的返回类型设置为IEnumerable<string>使其更易于消费。

我们还需要非常小心因试图访问 protected 文件和目录而导致的异常。下面的代码也很复杂，因为您不允许 yield return从内部 try/catch block ，因此我们必须稍微重新安排代码。

(另请注意，我们必须处理从 .GetEnumerator() 返回的枚举器；通常这会在您使用 foreach 时自动完成，但在这种情况下我们不能 - 因为必须避免在中执行 yield return try/catch - 所以我们必须使用 using 来处理它。)

这是对您的原始代码的修改以执行此操作:

public static IEnumerable<string> GetFiles(string root, string spec)
{
    var pending = new Stack<string>(new []{root});

    while (pending.Count > 0)
    {
        var path = pending.Pop();
        IEnumerator<string> fileIterator = null;

        try
        {
            fileIterator = Directory.EnumerateFiles(path, spec).GetEnumerator();
        }

        catch {}

        if (fileIterator != null)
        {
            using (fileIterator)
            {
                while (true)
                {
                    try
                    {
                        if (!fileIterator.MoveNext()) // Throws if file is not accessible.
                            break;
                    }

                    catch { break; }

                    yield return fileIterator.Current;
                }
            }
        }

        IEnumerator<string> dirIterator = null;

        try
        {
            dirIterator = Directory.EnumerateDirectories(path).GetEnumerator();
        }

        catch {}

        if (dirIterator != null)
        {
            using (dirIterator)
            {
                while (true)
                {
                    try
                    {
                        if (!dirIterator.MoveNext()) // Throws if directory is not accessible.
                            break;
                    }

                    catch { break; }

                    pending.Push(dirIterator.Current);
                }
            }
        }
    }
}

例如，您可以使用控制台应用列出“C:\”驱动器上所有可访问的“.txt”文件:

static void Main()
{
    foreach (var file in GetFiles("C:\\", "*.txt"))
    {
        Console.WriteLine(file);
    }
}