java递归的方式实现树形目录返回前端页面

工作中遇到的一个需求,考虑这样的程序在网上会很多想从网上直接拿,结果找了半天没找到自己想要的,干脆就自己写了,实现逻辑如下:

import cn.hutool.json.JSONUtil;
import lombok.Data;
import org.springframework.stereotype.Component;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

@Component
public class FileDirectory {

    private static Directory root = new Directory();

    private static String ROOT_LABEL = "fabcar";

    public static Directory fileList(File file, String rootName, List<Directory> child) {
        if (file.isFile() || 0 == file.listFiles().length) {
            return null;
        } else {
            // 绑定到根部录下
            if (ROOT_LABEL.equals(rootName)) {
                root.setLabel(rootName);
                root.setChildren(child);
            }

            // 遍历目录
            File[] files = file.listFiles();
            for(File f : files) {
                if (f.isFile()) {
                    // 如果是文件
                    Directory content = new Directory();
                    content.setLabel(f.getName());
                    child.add(content);
                } else if (f.isDirectory()) {
                    // 如果是目录
                    List<Directory> children = new ArrayList<>(); // 生成子目录用于存放递归目录
                    Directory content = new Directory();
                    content.setLabel(f.getName());
                    content.setChildren(children);
                    child.add(content);

                    // 递归目录
                    fileList(f, f.getName(), children);
                }
            }
        }

        return root;
    }

    public static void main(String[] args) {
        File file = new File("E:\\path\\tmp\\src\\fabcar");
        List<Directory> childList = new ArrayList<>();
        Directory directory = fileList(file, ROOT_LABEL, childList);

        System.out.println(JSONUtil.toJsonStr(directory));
    }

}
@Data
class Directory {

    // 目录标签
    private String label;

    // 子目录
    private List<Directory> children;

}

效果如下:
java递归的方式实现树形目录返回前端页面_第1张图片

你可能感兴趣的:(java,java,递归算法,树结构)