FileIterator

我们在开发中，经常需要遍历一个目录下的所有文件，常用的办法就是使用一个函数递归遍历是常用的办法。例如：

public static void iterateFile(File file) {
    if (file.isDirectory()) {
        if (file.getName().startsWith(".")) return;

        for (File item : file.listFiles()) {
            iterateFile(item);
        }
        return;
    }

    // do something 
}

但是递归函数的缺点就是扩展不方便，当然你对这个函数加入一个参数FileHandler，这样扩展性稍好一些，但是仍然不够好，比如说，不能根据遍历的需要中途停止遍历，加入Filter等等。我实现了一个FileIterator，使得遍历一个目录下的文件如何遍历一个集合中的元素一般操作。

废话少说，代码如下：

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());
}