今天拿到一个需求 ,说是得返回树结构格式的对象。一开始感觉挺简单的,然后越看越迷,越看越糊涂。最终还是通过参照网友给的代码写出来的(当时看谁的也找不到了,就不@了吧,有侵权我就删,哈哈哈哈!!!)。方法大概就是自己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;
}