文件迭代器

我们在开发中，经常需要遍历一个目录下的所有文件，常用的办法就是使用一个函数递归遍历是常用的办法。例如：
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 com.apusic.io;

import java.io.File;
import java.util.Iterator;
import java.util.NoSuchElementException;

public class FileIterator implements Iterator {
	private static class State{
		final State parent;
		final File[] files;
		
		int index = 0;
		public State(State parent, File dir){
			this.parent = parent;
			this.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());
}

  转载自网友的... ...