如何在遞歸爬取目錄結構時取得進展？

Question

我有一個函數，加載給定的文件夾中的所有* .txt和所有它的子文件夾。我想獲得實際的進展（例如15/35加載）。

但我想不出如何獲得加載到目錄下一級目錄結構的文件數目，以添加到當前索引。

* a 
    * b 
    - 1.txt (file in dir b) 
    - 1.txt (file in dir a) 
    - 2.txt _(index of this file is 3 - one file from dir below, one file in this dir)_ 

代碼：

public int getFilesInSubfolders(directory) 
    { 
     int count = 0; 

     foreach (subdirectory in directory) 
     { 
      count += getFilesInSubfolders(); 
     } 

     foreach (txtfile in folderFiles) 
     { 

      load(txtfile); 
      count++; 

      updateProgress(actualIndex); // how to get the actual index? e.g. 15/35 loaded, so that 15 
     } 
     return count; 
    } 

Answer 1

有解決這個問題的方法有兩種。

您可以將ref int count傳遞給每個遞歸調用。最外面的呼叫將初始化爲count爲零。

例如：

public int IterateDirectories(string root) 
{ 
    int count = 0; 
    iterateDirectories(root, ref count); 
    return count; 
} 

private void iterateDirectories(string root, ref int count) 
{ 
    foreach (string directory in Directory.EnumerateDirectories(root)) 
     iterateDirectories(directory, ref count); 

    foreach (string file in Directory.EnumerateFiles(root, "*.txt")) 
    { 
     // load(file); 

     ++count; 

     // Now count is the actual number of files processed, 
     // so you can use it for updateProgress() 
    } 
} 

---

或者，你可以用整個事情中的一類這樣的：

public sealed class DirectoryIterator 
{ 
    public static int Iterate(string root) 
    { 
     var iterator = new DirectoryIterator(); 
     iterator.iterate(root); 
     return iterator.count; 
    } 

    private void iterate(string root) 
    { 
     foreach (string directory in Directory.EnumerateDirectories(root)) 
      iterate(directory); 

     foreach (string file in Directory.EnumerateFiles(root, "*.txt")) 
     { 
      // load(file); 

      ++count; 

      // Now count is the actual number of files processed, 
      // so you can use it for updateProgress() 
     } 
    } 

    private int count; 

    private DirectoryIterator(){} 
} 

您可以使用這樣的：

int count = DirectoryIterator.Iterate("D:\\"); 

（雖然你可能不在乎關於返回的值）。

您需要修改此代碼以確切目的（它沒有計算您已經計算的文件的總數，因此您必須添加一個字段以用於那）。

注：我已經從這兩個例子中省略了錯誤處理。真正的代碼必須避免受保護的系統目錄。

Answer 2

我不明白你爲什麼要使用遞歸此。還有就是Directory.GetFiles方便的過載，讓您獲得所有子文件夾中的所有文件：

public int GetFilesInSubfolders(string directory) 
{ 
    var files = Directory.GetFiles(directory, "*.txt", SearchOption.AllDirectories)); 
    for (var i = 0; i < files.Length; i++) 
    { 
     load(files[i]); 
     updateProgress(i); 
    } 

    return files.Length; 
} 

Answer 3

明顯簡單的解決辦法是循環兩次 - 第一次計數的文件數量，然後使用計數器updateProgress。 爲了使其更具可讀性，您可以使用Stack<T>數據結構將其重構爲遠離遞歸，但這是另一回事。

確定您的解釋後，我承擔以下重構。 只要不使用遞歸，就有actualIndex。

int totalCounter = GetTheTotalTxtFilesNumber();//as you've mentioned you already have it 

Stack<Directory> directoryStack = new Stack<Directory>(); 
    directoryStack.Push(directory); 

    int actualIndex = 0; 
    while(directoryStack.Count > 0) 
    { 
    Dirextory current = directoryStack.Pop(); 

    foreach (txtfile in folderFiles) 
    { 
     load(txtfile); 
     actualIndex++; 
     updateProgress(actualIndex);//15 out of 35 
    } 
    foreach (subdirectory in current) 
    { 
     directoryStack.Push(subdirectory); 
    } 
    } 

Answer 4

我可以給你15/35中的15個部分，但不是35部分，因爲我不知道，直到這個迭代完成爲止。

公衆詮釋getFilesInSubfolders（目錄，詮釋計數）{

foreach (subdirectory in directory) 
    { 
     getFilesInSubfolders(subDirectory,count); 
    } 

    foreach (txtfile in folderFiles) 
    { 

     load(txtfile); 
     count++; 

     updateProgress(count); // how to get the actual index? e.g. 15/35 loaded, so that 15 
    } 
    return count; 
}