统计指定目录 题目描述: 统计指定目录中,各类文件(以文件后缀名区分)的数量和大小(单位MB)。 实现思路: 遍历指定目录中的每个子文件,获取该文件的后缀名,使用后缀名作为Map的key,并将数量和大小以Map的形式封装至value。
时间: 2024-02-25 12:57:01 浏览: 35
好的,这是一个很常见的文件统计功能,一般可以通过递归遍历目录中的所有文件,并使用HashMap来记录每个后缀名对应的文件数量和文件大小。具体代码实现如下:
```java
import java.io.File;
import java.util.HashMap;
import java.util.Map;
public class FileStats {
public static void main(String[] args) {
String dirPath = "D:\\test"; // 指定目录路径
File dir = new File(dirPath);
Map<String, FileStatsInfo> statsMap = new HashMap<>(); // 用于记录统计信息的Map
// 递归遍历目录中的所有文件,并进行统计
countFiles(dir, statsMap);
// 输出统计结果
for (Map.Entry<String, FileStatsInfo> entry : statsMap.entrySet()) {
String suffix = entry.getKey();
FileStatsInfo info = entry.getValue();
System.out.printf("后缀名为%s的文件数量为%d,总大小为%.2fMB\n", suffix, info.getCount(), info.getSize() / 1024.0 / 1024.0);
}
}
// 递归遍历目录中的所有文件,并进行统计
private static void countFiles(File dir, Map<String, FileStatsInfo> statsMap) {
if (!dir.exists()) {
return;
}
if (dir.isFile()) {
String suffix = getSuffix(dir.getName());
if (suffix != null) {
FileStatsInfo info = statsMap.getOrDefault(suffix, new FileStatsInfo());
info.setCount(info.getCount() + 1);
info.setSize(info.getSize() + dir.length());
statsMap.put(suffix, info);
}
} else if (dir.isDirectory()) {
File[] files = dir.listFiles();
if (files != null) {
for (File file : files) {
countFiles(file, statsMap);
}
}
}
}
// 获取文件的后缀名
private static String getSuffix(String fileName) {
int index = fileName.lastIndexOf('.');
if (index > 0 && index < fileName.length() - 1) {
return fileName.substring(index + 1);
}
return null;
}
// 用于记录文件统计信息的类
private static class FileStatsInfo {
private int count; // 文件数量
private long size; // 总大小
public int getCount() {
return count;
}
public void setCount(int count) {
this.count = count;
}
public long getSize() {
return size;
}
public void setSize(long size) {
this.size = size;
}
}
}
```
需要注意的是,这里使用了一个内部类`FileStatsInfo`来记录每个后缀名对应的文件数量和文件大小。另外,为了方便输出大小的单位为MB,需要将文件大小除以1024再除以1024。