使用Java搜索特定后缀名的文件

最新推荐文章于 2023-02-10 11:59:29 发布

原创最新推荐文章于 2023-02-10 11:59:29 发布 · 901 阅读

0 ·

CC 4.0 BY-SA版权

Java 专栏收录该内容

11 篇文章

订阅专栏

本文详细介绍了如何使用Java编程语言，在指定目录内递归搜索具有特定后缀名的文件，并生成包含这些文件的列表。通过定义FileListGenerator类，实现了根据路径和后缀名筛选文件的功能。

之前写过一个Java程序，其中以特定后缀名为条件，在指定目录内递归搜索文件，再生成文件列表。这里记录下其中的关键代码：

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class FileListGenerator {
    // FileList类可寻找指定路径下所有指定文件后缀名的文件，并生成文件列表 － List<File>

    FileListGenerator(File path) { // 如果未指定文件后缀名
        if (path.isDirectory() && path.isAbsolute()) {
            this.entryPath = path;
        } else {
            throw new IllegalArgumentException("The given path is not absolute directory!!");
        }
        this.specifySuffix = false; // 没有传入指定后缀名的参数，所以将specifySuffix设为false
        fileList = new ArrayList<File>(); // 初始化文件列表
    }

    FileListGenerator(File path, String fileNameSuffix) { // 如果指定了文件后缀名
        if (path.isDirectory() && path.isAbsolute()) {
            this.entryPath = path;
        } else {
            throw new IllegalArgumentException("The given path is not absolute directory!!");
        }
        this.specifySuffix = true;
        this.fileNameSuffix = fileNameSuffix;
        fileList = new ArrayList<File>(); // 初始化文件列表
    }

    private File entryPath; // 搜索文件的入口路径
    private List<File> fileList; // 对象类型为File的列表，该列表的初始化在构造方法中完成
    private boolean specifySuffix; // 是否指定搜索目标文件的后缀名，以开启搜索特定文件功能
    private String fileNameSuffix;

    public List<File> getFileList() {
        this.fillFileList(entryPath);
        return fileList;
    }

    private void fillFileList(File absolutePath) {
        // 寻找指定后缀名或者普通文件，并填充文件列表fileList
        File[] fileListInDir = absolutePath.listFiles(); // 用以保存指定路径下所有文件&目录列表，注意目录也是File类型

        if (this.specifySuffix) {
            for (int i = 0; i < fileListInDir.length; i++) {
                if (fileListInDir[i].isFile()) { // 判断当前项是否是文件，或者是目录
                    if (fileListInDir[i].getAbsolutePath().endsWith(this.fileNameSuffix)) {
                        this.fileList.add(fileListInDir[i]);
                    }
                } else {
                    this.fillFileList(fileListInDir[i]); // 如果数组fileListInDir中的当前项不是文件，而是目录时，递归调用方法fillFileList()
                }
            }
        } else {
            for (File x : fileListInDir) { // 如果不指定文件后缀名的话，寻找所有的文件的过程就简单了...
                if (x.isFile()) {
                    this.fileList.add(x);
                } else {
                    this.fillFileList(x);
                }
            }
        }

        fileListInDir = null; // 此时该临时文件列表已完成使用
    }