最近发现磁盘空间严重不足，都弹窗提示我了：
想想看到底哪个文件夹占的空间比较大，好做针对性的删除和清理。奈何Windows系统没有查看文件夹大小的工具，只能鼠标放在某个文件夹上，等提示：
AI时代都来临了，就让ChatGPT帮我写一个程序吧：计算指定文件夹中所有文件夹的大小并排序。
初版
于是，我就向ChatGPT提问：
完整的代码是这样的：
import java.io.File;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
public class FolderSizeCalculator {
public static void main(String[] args) {
// 指定文件夹路径
String folderPath = "/path/to/folder";
// 获取文件夹列表
File folder = new File(folderPath);
File[] folderList = folder.listFiles(File::isDirectory);
// 计算每个文件夹的大小并存储到ArrayList中
ArrayList<Folder> folders = new ArrayList<>();
for (File f : folderList) {
long size = getFolderSize(f);
Folder folderObj = new Folder(f.getName(), size);
folders.add(folderObj);
}
// 根据文件夹大小进行排序
Collections.sort(folders, Comparator.comparingLong(Folder::getSize));
// 输出结果
for (Folder f : folders) {
System.out.println(f.getName() + " - " + formatSize(f.getSize()));
}
}
// 递归计算文件夹大小
public static long getFolderSize(File folder) {
long size = 0;
File[] fileList = folder.listFiles();
if (fileList != null) {
for (File f : fileList) {
if (f.isFile()) {
size += f.length();
} else {
size += getFolderSize(f);
}
}
}
return size;
}
// 格式化文件大小
public static String formatSize(long size) {
String[] units = {
"B", "KB", "MB", "GB", "TB"};
int i = 0;
double s = size;
while (s >= 1024 && i < units.length - 1) {
s /= 1024;
i++;
}
return String.format("%.2f %s", s, units[i]);
}
}
// 文件夹类
class Folder {
private String name;
private long size;
public Folder(String name, long size) {
this.name = name;
this.size = size;
}
public String getName() {
return name;
}
public long getSize() {
return size;
}
}
使用File类获取指定文件夹中的所有子文件夹，然后使用递归函数getFolderSize计算每个文件夹的大小，最后将结果存储到一个ArrayList中。程序使用Collections.sort方法根据文件夹大小对ArrayList进行排序，并使用formatSize方法格式化文件大小并输出结果。
修改一下指定文件夹的路径就可以运行，效果还不错：
社 - 646.13 KB
学 - 141.15 MB
猫 - 587.99 MB
万 - 1.52 GB
唯一有一点小瑕疵，排序是从小到大的，稍微做一下修改，把
// 根据文件夹大小进行排序
Collections.sort(folders, Comparator.comparingLong(Folder::getSize));
修改为：
// 根据文件夹大小进行排序
Collections.sort(folders, (f1, f2) -> Long.compare(f2.getSize(), f1.getSize()));
运行一下，看看效果：
万 - 1.52 GB
猫 - 587.99 MB
学 - 141.15 MB
社 - 646.13 KB
进化
每次修改需要计算的文件夹路径时，都需要修改代码重新运行，很不方便。我想到了Swing，可以用UI进行交互。于是，我又向ChatGPT重新提问：
完整的代码如下：
import java.awt.*;
import java.awt.event.*;
import java.io.*;
import java.util.*;
import javax.swing.*;
public class FolderSizeSorter extends JFrame {
private JLabel directoryLabel;
private JTextField directoryTextField;
private JButton calculateButton;
private JTextArea resultTextArea;
public FolderSizeSorter() {
super("Folder Size Sorter");
// set up UI components
directoryLabel = new JLabel("Directory:");
directoryTextField = new JTextField(20);
calculateButton = new JButton("Calculate");
resultTextArea = new JTextArea(20, 40);
JScrollPane scrollPane = new JScrollPane(resultTextArea);
// set up layout
JPanel inputPanel = new JPanel();
inputPanel.add(directoryLabel);
inputPanel.add(directoryTextField);
inputPanel.add(calculateButton);
add(inputPanel, BorderLayout.NORTH);
add(scrollPane, BorderLayout.CENTER);
// set up event listeners
calculateButton.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
calculateFolderSizes();
}
});
// set up window properties
setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
setSize(500, 400);
setVisible(true);
}
private void calculateFolderSizes() {
String directoryPath = directoryTextField.getText();
File directory = new File(directoryPath);
if (!directory.isDirectory()) {
JOptionPane.showMessageDialog(this, "Please enter a valid directory path.");
return;
}
Map<File, Long> folderSizes = new HashMap<>();
calculateFolderSizesRecursive(directory, folderSizes);
List<Map.Entry<File, Long>> folderSizesList = new ArrayList<>(folderSizes.entrySet());
Collections.sort(folderSizesList, new Comparator<Map.Entry<File, Long>>() {
public int compare(Map.Entry<File, Long> o1, Map.Entry<File, Long> o2) {
return o2.getValue().compareTo(o1.getValue());
}
});
resultTextArea.setText("");
for (Map.Entry<File, Long> folderSize : folderSizesList) {
resultTextArea.append(folderSize.getKey().getPath() + " - " + formatSize(folderSize.getValue()) + "\n");
}
}
private void calculateFolderSizesRecursive(File folder, Map<File, Long> folderSizes) {
File[] files = folder.listFiles();
if (files == null) {
return;
}
long folderSize = 0;
for (File file : files) {
if (file.isDirectory()) {
calculateFolderSizesRecursive(file, folderSizes);
folderSize += folderSizes.get(file);
} else {
folderSize += file.length();
}
}
folderSizes.put(folder, folderSize);
}
private String formatSize(long size) {
String[] units = {
"B", "KB", "MB", "GB", "TB"};
int unitIndex = 0;
double dSize = size;
while (dSize >= 1024 && unitIndex < units.length - 1) {
dSize /= 1024;
unitIndex++;
}
return String.format("%.2f %s", dSize, units[unitIndex]);
}
public static void main(String[] args) {
new FolderSizeSorter();
}
}
创建了一个GUI窗口，包括一个标签、一个文本框、一个按钮，在文本框输入文件夹的路径，点击“Calculate”就可以了，效果如下图：
再进化
在使用过程中，又有了新的问题：我看到某一个文件夹比较大，想再看看这个文件夹下的文件夹大小情况，还得手动复制粘贴文件夹路径重新计算，操作起来不是很方便。于是，我又向ChatGPT重新提问：
完整的代码如下：
import java.awt.BorderLayout;
import java.awt.EventQueue;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
public class FolderSizeCalculator extends JFrame {
private static final long serialVersionUID = 1L;
private JPanel contentPane;
private JButton btnSelectFolder;
private JTree tree;
private DefaultTreeModel treeModel;
/**
* Launch the application.
*/
public static void main(String[] args) {
EventQueue.invokeLater(new Runnable() {
public void run() {
try {
FolderSizeCalculator frame = new FolderSizeCalculator();
frame.setVisible(true);
} catch (Exception e) {
e.printStackTrace();
}
}
});
}
/**
* Create the frame.
*/
public FolderSizeCalculator() {
setTitle("Folder Size Calculator");
setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
setBounds(100, 100, 450, 300);
contentPane = new JPanel();
contentPane.setLayout(new BorderLayout(0, 0));
setContentPane(contentPane);
btnSelectFolder = new JButton("Select Folder");
btnSelectFolder.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
JFileChooser fileChooser = new JFileChooser();
fileChooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
int result = fileChooser.showOpenDialog(FolderSizeCalculator.this);
if (result == JFileChooser.APPROVE_OPTION) {
File folder = fileChooser.getSelectedFile();
DefaultMutableTreeNode root = new DefaultMutableTreeNode(folder.getName());
treeModel = new DefaultTreeModel(root);
tree.setModel(treeModel);
calculateFolderSize(folder, root);
tree.expandRow(0);
}
}
});
contentPane.add(btnSelectFolder, BorderLayout.NORTH);
JScrollPane scrollPane = new JScrollPane();
contentPane.add(scrollPane, BorderLayout.CENTER);
tree = new JTree();
scrollPane.setViewportView(tree);
}
private void calculateFolderSize(File folder, DefaultMutableTreeNode node) {
long size = 0;
List<File> subfolders = new ArrayList<>();
for (File file : folder.listFiles()) {
if (file.isDirectory()) {
subfolders.add(file);
} else {
size += file.length();
}
}
Collections.sort(subfolders);
for (File subfolder : subfolders) {
DefaultMutableTreeNode subnode = new DefaultMutableTreeNode(subfolder.getName());
node.add(subnode);
calculateFolderSize(subfolder, subnode);
}
node.setUserObject(node.getUserObject() + " (" + humanReadableByteCount(size, false) + ")");
}
private String humanReadableByteCount(long bytes, boolean si) {
int unit = si ? 1000 : 1024;
if (bytes < unit)
return bytes + " B";
int exp = (int) (Math.log(bytes) / Math.log(unit));
String pre = (si ? "kMGTPE" : "KMGTPE").charAt(exp - 1) + (si ? "" : "i");
return String.format("%.1f %sB", bytes / Math.pow(unit, exp), pre);
}
}
这个程序中，创建了一个继承自JFrame的FolderSizeCalculator类，并在其中创建了一个用户界面。
我们可以通过点击“Select Folder”按钮来选择要计算大小的文件夹。一旦用户选择了一个文件夹，程序将会计算该文件夹中所有文件夹的大小，并将结果以树形结构展示出来。
计算文件夹大小的方法是calculateFolderSize，它使用递归来遍历文件夹及其子文件夹，计算它们的大小。该方法将每个子文件夹作为一个节点添加到树中，并将文件夹大小添加到节点名称中。
humanReadableByteCount是一个辅助方法，用于将文件大小从字节转换为人类可读的格式（如MB、GB等）。
运行一下，看看效果：
发现了问题，子文件夹的大小反而比父文件夹的大，仔细阅读了一下代码发现：在calculateFolderSize方法中累加了文件的大小，没有累加子文件夹的大小。于是，我又向ChatGPT重新提问：
ChatGPT增加了一个calculateTotalSize方法：
private long calculateTotalSize(File folder) {
long totalSize = 0;
if (folder.isDirectory()) {
for (File file : folder.listFiles()) {
if (file.isDirectory()) {
totalSize += calculateTotalSize(file);
} else {
totalSize += file.length();
}
}
} else {
totalSize = folder.length();
}
return totalSize;
}
在该方法中，首先检查传递的文件夹是否是文件夹类型。如果是文件夹，则迭代遍历其所有子项，如果子项是文件夹，则递归调用该方法以计算其大小；如果子项是文件，则将其大小加到总大小中。如果传递的文件夹是文件，则直接返回其大小。最终，该方法返回总大小。
将以上方法集成到原有方法中，用于计算文件夹下所有文件和文件夹的大小总和。以下是修改后的示例代码：
private void calculateFolderSize(File folder, DefaultMutableTreeNode node) {
long size = calculateTotalSize(folder);
List<File> subfolders = new ArrayList<>();
for (File file : folder.listFiles()) {
if (file.isDirectory()) {
subfolders.add(file);
}
}
Collections.sort(subfolders);
for (File subfolder : subfolders) {
DefaultMutableTreeNode subnode = new DefaultMutableTreeNode(subfolder.getName());
node.add(subnode);
calculateFolderSize(subfolder, subnode);
}
node.setUserObject(node.getUserObject() + " (" + humanReadableByteCount(size, false) + ")");
}
我们可以发现ChatGPT修改后的代码性能不是很好，因为calculateFolderSize方法就是一个递归方法，在这个递归方法里还要对每个文件夹再做一次递归。
我们完全可以在一次递归中完成整个计算过程。于是，我又向ChatGPT重新提问：
很遗憾，我虽然给ChatGPT扣上了卓越的帽子，但回答仍然不理想。
只能自己动手了，把calculateFolderSize方法修改为：
private long calculateFolderSize(File folder, DefaultMutableTreeNode node) {
long size = 0;
Map<DefaultMutableTreeNode, Long> nodeMap = new HashMap<>();
for (File file : folder.listFiles()) {
if (file.isDirectory()) {
DefaultMutableTreeNode subnode = new DefaultMutableTreeNode(file.getName());
long subSize = calculateFolderSize(file, subnode);
nodeMap.put(subnode, subSize);
size += subSize;
} else {
size += file.length();
}
}
List<Entry<DefaultMutableTreeNode, Long>> nodes = new ArrayList<>(nodeMap.entrySet());
Collections.sort(nodes, (o1,o2)-> o2.getValue().compareTo(o1.getValue()));
for (Map.Entry<DefaultMutableTreeNode, Long> entry : nodes) {
node.add(entry.getKey());
}
node.setUserObject(node.getUserObject() + " (" + humanReadableByteCount(size, false) + ")");
return size;
}
运行一下，看看效果：
结尾
自从ChatGPT等AI技术风靡全球以来，有不少人开始担忧：我会不会被AI所替代？虽然AI在许多领域展现出了惊人的能力，但要想完全替代人类仍需克服许多难题。
在本文中，即便是在AI目前已经显示出较高的技术水平下，它仍然需要我的改进和优化来完成我所提供的任务。AI可以被看作是一种工具，协助我解决一些复杂的问题，提高工作效率。
因此，替代你的并不是AI，而是那些能够运用AI的人。
附：最终完整代码
import javax.swing.*;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
public class FolderSizeCalculator extends JFrame {
private static final long serialVersionUID = 1L;
private JPanel contentPane;
private JButton btnSelectFolder;
private JTree tree;
private DefaultTreeModel treeModel;
/**
* Launch the application.
*/
public static void main(String[] args) {
EventQueue.invokeLater(() -> {
try {
FolderSizeCalculator frame = new FolderSizeCalculator();
frame.setVisible(true);
} catch (Exception e) {
e.printStackTrace();
}
});
}
/**
* Create the frame.
*/
public FolderSizeCalculator() {
setTitle("Folder Size Calculator");
setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
setBounds(100, 100, 450, 300);
contentPane = new JPanel();
contentPane.setLayout(new BorderLayout(0, 0));
setContentPane(contentPane);
btnSelectFolder = new JButton("Select Folder");
btnSelectFolder.addActionListener(new ActionListener() {
@Override
public void actionPerformed(ActionEvent e) {
JFileChooser fileChooser = new JFileChooser();
fileChooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
int result = fileChooser.showOpenDialog(FolderSizeCalculator.this);
if (result == JFileChooser.APPROVE_OPTION) {
File folder = fileChooser.getSelectedFile();
DefaultMutableTreeNode root = new DefaultMutableTreeNode(folder.getName());
treeModel = new DefaultTreeModel(root);
tree.setModel(treeModel);
calculateFolderSize(folder, root);
tree.expandRow(0);
}
}
});
contentPane.add(btnSelectFolder, BorderLayout.NORTH);
JScrollPane scrollPane = new JScrollPane();
contentPane.add(scrollPane, BorderLayout.CENTER);
tree = new JTree();
scrollPane.setViewportView(tree);
}
private long calculateFolderSize(File folder, DefaultMutableTreeNode node) {
long size = 0;
Map<DefaultMutableTreeNode, Long> nodeMap = new HashMap<>();
for (File file : folder.listFiles()) {
if (file.isDirectory()) {
DefaultMutableTreeNode subnode = new DefaultMutableTreeNode(file.getName());
long subSize = calculateFolderSize(file, subnode);
nodeMap.put(subnode, subSize);
size += subSize;
} else {
size += file.length();
}
}
List<Entry<DefaultMutableTreeNode, Long>> nodes = new ArrayList<>(nodeMap.entrySet());
nodes.sort((o1, o2) -> o2.getValue().compareTo(o1.getValue()));
for (Map.Entry<DefaultMutableTreeNode, Long> entry : nodes) {
node.add(entry.getKey());
}
node.setUserObject(node.getUserObject() + " (" + humanReadableByteCount(size, false) + ")");
return size;
}
private String humanReadableByteCount(long bytes, boolean si) {
int unit = si ? 1000 : 1024;
if (bytes < unit) {
return bytes + " B";
}
int exp = (int) (Math.log(bytes) / Math.log(unit));
String pre = (si ? "kMGTPE" : "KMGTPE").charAt(exp - 1) + (si ? "" : "i");
return String.format("%.1f %sB", bytes / Math.pow(unit, exp), pre);
}
}