1. import java.io.File;
  2. import java.util.HashMap;
  3. import java.util.Map;
  5. class DirectoryAggregator {
  7. /**
  8. * Aggregates values (file sizes) of directories for sandbox usage.
  9. * Handles edge cases like non-existent directories and permission issues.
  10. *
  11. * @param rootDir The root directory to aggregate from.
  12. * @return A map where keys are directory paths and values are their total sizes in bytes.
  13. * Returns an empty map if the rootDir is null or an invalid path.
  14. */
  15. public static Map<String, Long> aggregateDirectorySizes(String rootDir) {
  16. Map<String, Long> directorySizes = new HashMap<>();
  18. if (rootDir == null || rootDir.isEmpty()) {
  19. return directorySizes; // Handle null or empty rootDir
  20. }
  22. File root = new File(rootDir);
  24. if (!root.exists()) {
  25. System.err.println("Error: Directory does not exist: " + rootDir);
  26. return directorySizes; // Handle non-existent directory
  27. }
  29. if (!root.isDirectory()) {
  30. System.err.println("Error: Not a directory: " + rootDir);
  31. return directorySizes; // Handle files instead of directories
  32. }
  34. File[] files = root.listFiles();
  36. if (files == null) {
  37. System.err.println("Error: Could not list files in directory: " + rootDir);
  38. return directorySizes; // Handle permission issues or other listing errors
  39. }
  41. for (File file : files) {
  42. if (file.isDirectory()) {
  43. // Recursively aggregate sizes of subdirectories
  44. Map<String, Long> subDirSizes = aggregateDirectorySizes(file.getAbsolutePath());
  45. for (Map.Entry<String, Long> entry : subDirSizes.entrySet()) {
  46. directorySizes.put(entry.getKey(), directorySizes.getOrDefault(entry.getKey(), 0L) + entry.getValue());
  47. }
  48. } else if (file.isFile()) {
  49. directorySizes.put(file.getAbsolutePath(), file.length());
  50. }
  51. }
  53. return directorySizes;
  54. }
  56. public static void main(String[] args) {
  57. // Example usage:
  58. String sandboxDir = "test_sandbox"; // Replace with your desired directory
  60. // Create a test directory structure if it doesn't exist
  61. File sandbox = new File(sandboxDir);
  62. if (!sandbox.exists()) {
  63. sandbox.mkdir();
  64. File subdir1 = new File(sandboxDir, "subdir1");
  65. subdir1.mkdir();
  66. File file1 = new File(sandboxDir, "file1.txt");
  67. file1.createNewFile();
  68. File file2 = new File(sandboxDir, "subdir1", "file2.txt");
  69. file2.createNewFile();
  70. }
  72. Map<String, Long> sizes = aggregateDirectorySizes(sandboxDir);
  74. for (Map.Entry<String, Long> entry : sizes.entrySet()) {
  75. System.out.println("Directory: " + entry.getKey() + ", Size: " + entry.getValue() + " bytes");
  76. }
  78. }
  79. }

Add your comment