j将目录以树形输出

我不是女神ヾ 2024-04-01 17:43 60阅读 0赞

public static List<Object> getFromDir(String dir){
        
    
            File file = new File(dir);
            File[] files = file.listFiles();
            if (files == null){
        
                return null;
            }
            List<Object> nameList = new ArrayList<>();
            for (File f : files){
        
                if (f.isFile()){
        
                    nameList.add(f.getName());
                }
                if (f.isDirectory()){
        
                    File[] filesFromDir = f.listFiles();
                    Map<String, List<String>> map = new HashMap<>();
                    if (filesFromDir != null){
        
                        List<String> dirNames = new ArrayList<>();
                        for (File fileFromDir : filesFromDir){
        
                            dirNames.add(fileFromDir.getName());
                        }
                        map.put(dir + "/" +f.getName() + "/", dirNames);
                        nameList.add(map);
                    }
                }
            }
            return nameList;
        }
    
        /**
         * 生成输出格式
         * @param name 输出的文件名或目录名
         * @param level 输出的文件名或者目录名所在的层次
         * @return 输出的字符串
         */
        public static String createPrintStr(String name, int level) {
        
            // 输出的前缀
            StringBuilder printStr = new StringBuilder();
            // 按层次进行缩进
            for (int i = 0; i < level; i ++) {
        
                printStr.append("  ");
            }
            printStr.append("   - ").append(name);
            return printStr.toString();
        }
    
        /**
         * 输出初始给定的目录
         * @param dirPath 给定的目录
         */
        public static void printDir(String dirPath){
        
            // 将给定的目录进行分割
            String[] dirNameList = dirPath.split("\\\\");
            // 设定文件level的base
            fileLevel = dirNameList.length;
            // 按格式输出
            for (int i = 0; i < dirNameList.length; i ++) {
        
                System.out.println(createPrintStr(dirNameList[i], i));
            }
        }
    
        /**
         * 输出给定目录下的文件，包括子目录中的文件
         * @param dirPath 给定的目录
         */
        public static boolean readFile(String dirPath, StringBuilder sb) {
        
            // 建立当前目录中文件的File对象
            File file = new File(dirPath);
            // 取得代表目录中所有文件的File对象数组
            File[] list = file.listFiles();
            if (list == null){
        
                return false;
            }
            // 遍历file数组
            for (File value : list) {
        
                if (value.isDirectory()) {
        
                    sb.append(createPrintStr(value.getName(), fileLevel)).append("\n");
                    fileLevel++;
                    // 递归子目录
                    readFile(value.getPath(), sb);
                    fileLevel--;
                } else {
        
                    sb.append(createPrintStr(value.getName(), fileLevel)).append("\n");
                }
            }
            return true;
        }
    
        public static void main(String[] args) {
        
            List<String> reportDir = new ArrayList<>();
            reportDir.add(xxDir1);
            reportDir.add(xxDir2);
            reportDir.add(xxDir3);
            reportDir.add(xxDir4);
    
            for (String dir : reportDir) {
        
                StringBuilder sb = new StringBuilder();
                readFile(dir, sb);
                System.out.println(dir + "\n" + sb);
            }
        }