非递归遍历文件夹下指定类型的文件-Java栈实现

这几天突然想写一个非递归遍历文件夹的小例子，苦思冥想后终于出炉了。

我测试了下，递归D:/programfile（30G左右），非递归性能是 递归 的30倍，表现不错。

后来在网上也搜了下，发现网上的例子都是用list，而我想的是栈，不错，我这也算是原创了吧。

public class Main {

/**
* 遍历文件夹下指定类型的文件
* @param srcDir 源文件
* @param parten 模式，可以多个，比如".xls" ".xlsx" ".mdb"，只要匹配一个就会输出
*/
public static void count(String srcDir,String... parten){

Dir root = new Dir(new File(srcDir)); //创建自定义目录

FILES_STACK.push(root); //将根目录压入栈中

while(!FILES_STACK.empty()){ //循环处理栈顶目录

Dir currentDir = FILES_STACK.peek();
if(currentDir.peekNum > 0){ //若栈顶目录之前peek过，则略过该目录(因为其子文件已全部被处理过)
FILES_STACK.pop();
continue;
}

File children[] = currentDir.dir.listFiles();

if(children == null || children.length == 0){ //若该目录是空目录，则略过该目录
FILES_STACK.pop();
continue;
}else{ //定义统计子目录数的变量
int countDir = 0;
for( File f : children){
if(f.isDirectory()) {
FILES_STACK.push(new Dir(f)); //对于子目录，不处理，直接压入栈中，然后重新执行循环
countDir++;
} else{

boolean flag = false;
for(String p : parten){ //匹配模式，只要匹配一种模式，则退出匹配

if(f.getName().endsWith(p)){
flag = true;
break;
}

}

if(flag){

// 在这里处理匹配到的文件

}

}
}

if(countDir == 0){ //如果子目录数为0，则该目录是叶子节点，将其剔出栈
FILES_STACK.pop();
continue;
}
else{ //如果子目录数不为0，则标示下该目录已经被peek过
currentDir.peekNum++;
}
}
}

}

}

class Dir {

File dir;
int peekNum;

Dir(File dir){
this.dir = dir;
this.peekNum = 0;
}

}