使用File类创建简单的文件目录检索小工具

File类的常用方法

工具类源码

package com.xy;

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

/**
 * @author 谢阳
 * @version 1.8.0_131
 * @date 2023/1/11 13:22
 * @description
 */
public class FileTreeShow {
    private static Long total_length = 0L;
    private static Integer total_file = 0;
    private static Boolean inquireFile = false;
    private static String fileName;
    private static ArrayList<String> filesPath;

    public static void main(String[] args) {
        String path = menuView();
        System.out.println("--------------------------------------------------------------------");
        File file = new File(path);
        showFlies(file, 0, 0);
        System.out.println("--------------------------------------------------------------------");
        System.out.println("共有子文件 " + total_file + " 个");
        System.out.println("文件总大小 " + total_length / 1024 / 1024 + " MB");
        if (inquireFile) {
            System.out.println("共查询到类似文件 " + filesPath.size() + " 个");
            filesPath.forEach(System.out::println);
        }
        System.out.println("--------------------------------------------------------------------");
    }

    /**
     * 输入条件
     * @return 路径
     */
    private static String menuView() {
        Scanner scanner = new Scanner(System.in);
        System.out.print("请输入文件路径:");
        String path = scanner.next();
        System.out.print("是否需查询文件或文件夹路径(y/n):");
        inquireFile = scanner.next().toLowerCase().charAt(0) == 'y';
        if (inquireFile) {
            System.out.print("请输入文件夹或文件名:");
            fileName = scanner.next();
        }
        return path;
    }

    /**
     * 展示文件目录结构
     * @param file  当前文件或文件夹
     * @param cur 当前目录
     * @param father 父级层数
     */
    private static void showFlies(File file, int cur, int father) {
        // 获取文件及文件夹大小
        total_length += file.length();

        if (inquireFile) {
            quireFile(file);
        }

        String pre = prefix(cur, father);
        System.out.println(pre + file.getName());
        File[] files = file.listFiles();

        // 递归查询
        if (files != null && files.length != 0) {
            for (File subFile : files) {
                showFlies(subFile, cur + 1, cur);
            }
        } else {
            // 获取文件数量
            total_file += 1;
        }
    }

    /**
     * 查询文件或文件夹是否存在, 存在则放入数组中
     * @param file 当前文件或文件夹
     */
    private static void quireFile(File file) {
        if(file != null && file.getName().contains(fileName)) {
            if (filesPath == null) {
                filesPath = new ArrayList<>();
            }
            filesPath.add(file.getAbsolutePath());
        }
    }

    /**
     * 添加前缀方便展示层级关系
     * @param cur 当前目录
     * @param father 父级层数
     * @return 前缀所需空格
     */
    private static String prefix(int cur,int father) {
        StringBuilder prefix = new StringBuilder("");
        if (cur >= 1) {
            prefix.append("  ");
        }
        for (int i = 0; i < cur; i++) {
            if (i == father) {
                prefix.append("|__");
            }
            if (i < father) {
                prefix.append("    ");
            }
        }

        return prefix.toString();
    }
}

测试结果如下：