我們在開發(fā)中，經(jīng)常需要遍歷一個(gè)目錄下的所有文件，常用的辦法就是使用一個(gè)函數(shù)遞歸遍歷是常用的辦法。例如：
public static void iterateFile(File file) {
if (file.isDirectory()) {
if (file.getName().startsWith(".")) return;
for (File item : file.listFiles()) {
iterateFile(item);
}
return;
}
// do something 
}
但是遞歸函數(shù)的缺點(diǎn)就是擴(kuò)展不方便，當(dāng)然你對這個(gè)函數(shù)加入一個(gè)參數(shù)FileHandler，這樣擴(kuò)展性稍好一些，但是仍然不夠好，比如說，不能根據(jù)遍歷的
需要中途停止遍歷，加入Filter等等。我實(shí)現(xiàn)了一個(gè)FileIterator，使得遍歷一個(gè)目錄下的文件如何遍歷一個(gè)集合中的元素一般操作。
廢話少說，代碼如下：
package net.wenshao;
import java.io.File;
import java.util.Iterator;
import java.util.NoSuchElementException;
public class FileIterator implements Iterator<File> {
private static class State {
final State parent;
final File[] files;
int index = 0;
public State(State parent, File dir) {
this.parent = parent;
files = dir.listFiles();
}
}
private File current;
private State state;
public FileIterator(File file) {
if (file.isDirectory()) {
state = new State(null, file);
nextInternal();
} else {
this.current = file;
state = null;
}
}
@Override
public boolean hasNext() {
return current != null;
}
@Override
public File next() {
File rtValue = current;
if (rtValue == null) throw new NoSuchElementException();
nextInternal();
return rtValue;
}
private void nextInternal() {
current = null;
if (this.state == null) return;
for (;;) {
if (state.index >= state.files.length) {
state = state.parent;
if (state == null) return;
state.index++;
continue;
}
File file = state.files[state.index];
// 可以在此處加入Filters處理代碼
if (file.isDirectory()) {
state = new State(state, file);
continue;
}
current = file;
state.index++;
break;
}
}
@Override
public void remove() {
throw new UnsupportedOperationException();
}
}
使用FileIterator的例子：
File dir = new File("/home/wenshao/workspace");
Iterator<File> iter = new FileIterator(dir);
while (iter.hasNext()) {
File file = iter.next();
System.out.println(file.getPath());
}