Java 返回树结构对象给前端

今天拿到一个需求 ，说是得返回树结构格式的对象。一开始感觉挺简单的，然后越看越迷，越看越糊涂。最终还是通过参照网友给的代码写出来的(当时看谁的也找不到了，就不@了吧，有侵权我就删，哈哈哈哈！！！)。方法大概就是自己id与上级id相匹配，然后递归。就这样，也挺简单的。浪费了一上午时间，加油吧 ，打工人！！！

---

/**
 * 查询树
 *
 * @return {@link List}<{@link OrgTreeNode}>
 */
public List<OrgTreeNode> queryTree() {
    LambdaQueryWrapper<SysOrgInfo> wrapper = Wrappers.lambdaQuery();
    //根据查询所有条目
    List<SysOrgInfo> sysOrgInfos = mapper.selectList(wrapper);
    menu = BeanUtil.mapAsList(sysOrgInfos, OrgTreeNode.class);
    List<OrgTreeNode> treeMenus = new ArrayList<OrgTreeNode>();
    for (OrgTreeNode menuNode : getRootNode()) {
        buildChilTree(menuNode);
        treeMenus.add(menuNode);
    }
    return treeMenus;
}

/**
 * 构建子树
 *
 * @param pNode p节点
 * @return {@link OrgTreeNode}
 */
private OrgTreeNode buildChilTree(OrgTreeNode pNode) {
    logger.info("进入pNode==>{}", pNode);
    List<OrgTreeNode> chilMenus = new ArrayList<OrgTreeNode>();
    for (OrgTreeNode menuNode : menu) {
        if (menuNode.getParentId().equals(pNode.getId())) {
            chilMenus.add(buildChilTree(menuNode));
        }
    }
    pNode.setChildren(chilMenus);
    logger.info("pNode==>{}", pNode);
    return pNode;
}

/**
 * 得到根节点
 *
 * @return {@link List}<{@link OrgTreeNode}>
 */
private List<OrgTreeNode> getRootNode() {
    logger.info("getRootNode===>{}", menu);
    List<OrgTreeNode> rootMenuLists = new ArrayList<OrgTreeNode>();
    for (OrgTreeNode menuNode : menu) {
        if (menuNode.getParentId() == 0) {
            rootMenuLists.add(menuNode);
        }
    }
    return rootMenuLists;
}