如何限制递归子目录搜索的深度

Question

我有一个功能，当前可以抓取所有文件夹和子文件夹来检查我正在构建的小工具的 ACL，但我正在绞尽脑汁试图弄清楚如何限制它可以达到的深度。例如，您有一个 4 层深的文件夹，但我希望只能获取其中 3 层的 ACL。

目前我已经这样编码了：

private void StepThroughDirectories(string dir)
{
    string[] directories = Directory.GetDirectories(dir);
    try
    {
        foreach (string d in Directory.GetDirectories(dir))
        {
            if (recCount < (int)Depth)
            {
                GetACLs(d, new DirectoryInfo(d));
                pBar.Value += 1;
                //MessageBox.Show("Recursive Level: " + counter.ToString());
                recCount++;
                StepThroughDirectories(d);
            }
            else
            {
                recCount--;
            }
        }
    }
    catch (System.Exception e)
    {
        Console.WriteLine(e.Message);
    }
}

显然，这并不像以前那么好，因为我已经研究这个问题一段时间了，但如果有人能指出我正确的方向来解决这个问题，我会非常高兴！

Answer 1

首先，避免在外部将 recCount 字段声明为“全局”变量。在递归场景中，沿着递归调用传递状态通常更容易管理。

其次，将深度测试移出 foreach，以消除对文件系统子目录的不必要查询。

第三，将实际的处理逻辑放在方法的开头，再次放在子目录处理循环之外。

您的代码将如下所示：

void StepThroughDirectories(string dir)
{
    StepThroughDirectories(dir, 0)
}

void StepThroughDirectories(string dir, int currentDepth)
{
    // process 'dir'
    ...

    // process subdirectories
    if (currentDepth < MaximumDepth)
    {
        foreach (string subdir in Directory.GetDirectories(dir))
            StepThroughDirectories(subdir, currentDepth + 1);
    }
}

Answer 2

一种可能的方法是在方法外部添加一个类字段和一个变量来指示最大深度有多少层。

整数级别；

private void StepThroughDirectories(string dir, int depth)
{
    levels ++;
    if (levels > depth)
        return;
    string[] directories = Directory.GetDirectories(dir);
    try
    { ...

Answer 3

当您从 StepThroughDirectories 返回时减少 recCount，但这会更好......

    private void StepThroughDirectories(string dir, int depth)
    {
        if (depth < 0)
            return;
        string[] directories = Directory.GetDirectories(dir);
        try
        {
            foreach (string d in Directory.GetDirectories(dir))
            {
                // your code here
                Console.WriteLine("{0}", d);
                StepThroughDirectories(d, depth-1);
            }
        }
        catch (System.Exception e)
        {
            Console.WriteLine(e.Message);
        }
    }