我们在开发中,经常需要遍历一个目录下的所有文件,常用的办法就是使用一个函数递归遍历是常用的办法。例如:
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 == nullthrow new NoSuchElementException();

        nextInternal();

        
return rtValue;
    }

    
private void nextInternal() {
        current 
= null;

        
if (this.state == nullreturn;

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

posted on 2008-06-05 07:56 温少的日志 阅读(1866) 评论(2)  编辑  收藏
Comments
  • # re: 一个FileIterator的实现
    kingslee
    Posted @ 2008-08-27 15:15
    State对象的声明,有些“为了对象而对象”的味道。可以直接用ArrayList或者Stack作为存储结构,语义更加简明。

    遍历大的文件目录时,有时候深度优先好些,有时候广度优先好些。上述代码中如果推算不错的话,应当是深度优先。  回复  更多评论   
  • # re: 一个FileIterator的实现
    温少的日志
    Posted @ 2008-08-30 16:12
    回kingslee
    1、State有三个field,parent、index和files,不声明一个对象,使用ArrayList不好,声明一个对象语义更明确。
    2、这是深度优先的,广度优先的实现方式会有所不同。  回复  更多评论   

只有注册用户登录后才能发表评论。


网站导航: