Java遍历目录实现

一、为什么需要遍历目录

在程序开发中，经常需要读取指定路径下的所有文件或者文件夹。而对于大规模的文件读取任务，手动一个一个的去查找并读取文件显然是不可取的。

此时就需要一种方法可以自动遍历指定目录下的所有文件和文件夹，并返回一个包含目录下所有文件绝对路径的列表或集合。

二、Java中遍历目录的方法

Java中实现遍历目录的方法有多种，本文将讲解三种常见的方法。

1. 递归算法

递归算法是一种常见的遍历目录的方式，其思路为：如果当前文件是文件夹，则递归该文件夹；反之，将文件路径加入列表。具体实现代码如下：

public static List getFiles(String path) {
    List fileList = new ArrayList();
    File[] files = new File(path).listFiles();
    if (files != null) {
        for (File file : files) {
            if (file.isDirectory()) {
                fileList.addAll(getFiles(file.getAbsolutePath()));
            } else {
                fileList.add(file.getAbsolutePath());
            }
        }
    }
    return fileList;
}

2. 广度优先算法

广度优先算法则是先将根目录下的所有文件和文件夹压入队列，然后不断从队首取出文件夹并将其子文件夹和文件入队，直到队列为空。具体实现代码如下：

public static List getFiles(String path) {
    List fileList = new ArrayList();
    Queue queue = new LinkedList();
    queue.offer(new File(path));
    while (!queue.isEmpty()) {
        File t = queue.poll();
        if (t.isDirectory()) {
            File[] files = t.listFiles();
            if (files != null) {
                for (File file : files) {
                    queue.offer(file);
                }
            }
        } else {
            fileList.add(t.getAbsolutePath());
        }
    }
    return fileList;
}

3. 深度优先算法

深度优先算法是一种借助栈实现的遍历方式，其基本思路是先将根目录压入栈中，然后取出栈顶的文件，如果是文件夹则将其子文件夹和文件倒序压入栈中，直到栈为空为止。具体实现代码如下：

public static List getFiles(String path) {
    List fileList = new ArrayList();
    Stack stack = new Stack();
    stack.push(new File(path));
    while (!stack.empty()) {
        File t = stack.pop();
        if (t.isDirectory()) {
            File[] files = t.listFiles();
            if (files != null) {
                for (int i = files.length - 1; i >= 0; i--) {
                    stack.push(files[i]);
                }
            }
        } else {
            fileList.add(t.getAbsolutePath());
        }
    }
    return fileList;
}

三、总结

对于小文件读取任务，以上三种遍历目录的方法都可以应对；但对于文件数较多的情况，广度优先算法和深度优先算法可能会遭遇栈内存溢出或队列过大的问题，此时递归算法较为稳妥。

Java提供了多种实现遍历目录的方法，根据不同的需求可以灵活选择对应的方法。