package util.file;
import java.io.File;
/**
* 搜索文件.
*
* @author koveer
* @since 1.0 - 2023/4/9 18:53
*/
public class SearchFile {
public static void main(String[] args) {
String directoryPath = "D:\\CS\\Markdown笔记"; // 指定目录路径
File rootDir = new File(directoryPath);
printTree(rootDir,1);
}
/**
* 返回文件大小的字符串形式,level指定文件大小格式 0:Byte, 1:KB, 2:MB, 3:GB.
*
* @param file 文件路径
* @param level 文件大小格式
* @return java.lang.String string
* @author koveer -2023/4/11 21:54
* @since 1.0
*/
public static String FileLevelSizeString(File file, int level) {
String[] levelString = new String[]{"Byte", "KB", "MB", "GB"};
return "(%d %s)".formatted(FileLevelSize(file, level), levelString[level]);
}
/**
* 格式化输出文件或文件夹大小,level指定文件大小格式 0:Byte, 1:KB, 2:MB, 3:GB.
*
* @param file 文件路径
* @param level 文件大小格式
* @return long
* @author koveer
* -2023/4/9 20:33
* @since 1.0
*/
public static long FileLevelSize(File file, int level) {
switch (level) {
case 0 -> {
return FileSize(file);
}
case 1 -> {
// return FileSize(file) / 1024;
return FileSize(file) >> 10;
}
case 2 -> {
// return FileSize(file) / 1048576;
return FileSize(file) >> 20;
}
case 3 -> {
// return FileSize(file) / 1073741824;
return FileSize(file) >> 30;
}
default -> throw new IllegalStateException("Unexpected value: " + level);
}
}
/**
* 递归获取文件或文件夹大小.
*
* @param file 文件路径
* @return long
* @author koveer
* -2023/4/9 20:32
* @since 1.0
*/
public static long FileSize(File file) {
if (file.isFile()) {
return file.length();
} else {
long size = 0;
File[] fileList = file.listFiles();
if (fileList != null)
for (File f : fileList) {
if (f.isFile()) {
size += f.length();
} else {
size += FileSize(f);
}
}
return size;
}
}
/**
* 以树状图形式打印文件夹下文件及大小,默认MB.
*
* @param file 文件路径
* @author koveer
* -2023/4/9 20:40
* @since 1.0
*/
public static void printTree(File file) {
System.out.println(file);
printTree(file, "", 0, 2);
}
/**
* 以树状图形式打印文件夹下文件及大小.
*
* @param file 文件路径
* @param level 大小格式
* @author koveer
* -2023/4/9 20:40
* @since 1.0
*/
public static void printTree(File file, int level) {
System.out.println(file);
printTree(file, "", 0, level);
}
/**
* 以树状图形式打印文件夹下文件及大小,默认MB.
*
* @param file 文件路径
* @param prefix 文件前缀
* @param deep 文件深度
* @author koveer
* -2023/4/9 20:39
* @since 1.0
*/
private static void printTree(File file, String prefix, int deep, int level) {
if (!file.isDirectory()) {
System.out.println("无效的文件路径.");
} else {
// 生成目录下的子文件和子目录列表
File[] fileList = file.listFiles();
String thisPrefix = "";
String nextPrefix = "";
for (int i = 0; i < fileList.length; i++) {
if (deep >= 0) {
// 如果不是最后一个元素
if ((i + 1 < fileList.length)) {
// 不是最后一个文件/目录都打印这个符号
thisPrefix = prefix + "├─";
// 下一个打印这符号表示展开目录
nextPrefix = prefix + "│ ";
} else {
// 最后一个子文件/子目录项
thisPrefix = prefix + "└─";
//
nextPrefix = prefix + " ";
}
}
// 如果是文件,则输出文件名和文件大小
if (fileList[i].isFile()) {
System.out.printf("%s%s %s%n", thisPrefix, fileList[i].getName(), FileLevelSizeString(file, level));
}
// 如果是目录,则输出目录名
if (fileList[i].isDirectory()) {
System.out.printf("%s\u001B[36m%s %s\u001B[0m%n", thisPrefix, fileList[i].getName(), FileLevelSizeString(file, level));
printTree(fileList[i], nextPrefix, deep + 1, level); // 递归输出目录下的文件和目录
}
}
}
}
}
具体效果:
以树状图的形式展现文件夹下所有文件及其大小,文件夹会标记为蓝色,实测效率不如windows自带的tree命令,但可拓展性比较高。
在控制台显示颜色的效果在WINDOWS10及以下的CMD窗口无法实现,请使用更高级的控制台,如terminal或者IDEA等。
请使用JDK12及以上版本。每个方法的具体功能已经详细注释。