一些代码

package com.slm.main;

import com.alibaba.fastjson.JSON;

import java.util.ArrayList;
import java.util.List;

public class Application {


    public static void main(String[] args) {
        TreeNode re = re(0);
        System.out.println(JSON.toJSONString(re));
    }

    /**
     * 根节点--根节点
     *          |----节点
     *                |---子孩子
     *          |----节点
     *          |----节点
     *          |----节点
     * @param id
     * @return
     */
    public static TreeNode re(int id) {
        Org byId = DB.getById(id);
        TreeNode treeNode = TreeNode.fromOrg(byId);
        List<Org> byParentId = getByParentId(byId.getId());
        treeNode.setChildren(new ArrayList<>());
        for (Org org : byParentId) {
            TreeNode re = re(org.getId());
            treeNode.getChildren().add(re);
        }
        return treeNode;
    }

    public static List<Org> getByParentId(int parentId) {
        List<Org> ret = new ArrayList<>();
        for (Org org : DB.list) {
            if (org.getParentId().equals(parentId)) {
                ret.add(org);
            }
        }
        return ret;
    }
}
相关文章
相关标签/搜索