一个FileIterator的实现
来源 :中华考试网 2020-11-30
中我们在开发中,经常需要遍历一个目录下的所有文件,常用的办法就是使用一个函数递归遍历是常用的办法。例如:
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;
填写下面表单即可预约申请免费试听java课程!害怕学不会?助教陪读,随时解惑!担心就业?一地学习,可全国推荐就业!
java课程免费学习,高薪触手可得
- 地区:
- 北京
- 天津
- 上海
- 江苏
- 浙江
- 山东
- 江西
- 安徽
- 广东
- 广西
- 海南
- 辽宁
- 吉林
- 黑龙江
- 内蒙古
- 山西
- 福建
- 河南
- 河北
- 湖南
- 湖北
- 四川
- 重庆
- 云南
- 贵州
- 新疆
- 西藏
- 陕西
- 青海
- 宁夏
- 甘肃
- 姓名:
- 手机:
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;
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
while (iter.hasNext()) {
File file = iter.next();
System.out.println(file.getPath());
}