java 菜单树 递归_MenuTreeUtil菜单树形递归工具类

import com.aaa.controller.vm.MenuVM;

import com.aaa.entity.Menu;

import java.util.ArrayList;

import java.util.List;

/**

* 组装菜单树工具类

* Created by cws

* @author Administrator

*/

public class MenuTreeUtil {

public static MenuVM getTree(List

//判断是否有获取到数据

if (list == null || list.size() < 1) {

return null;

}

return buildTree(getRoot(list), list);

}

private static MenuVM buildTree(MenuVM pnode, List

List childs = new ArrayList<>();

for (Menu tmp : nodes) {

if (pnode.getId().equals(tmp.getParentId())) {

childs.add(new MenuVM(tmp));

}

}

//二次递归遍历子节点

if (childs.size() > 0) {

pnode.setSubMenu(childs);

for (MenuVM ptmp : childs) {

buildTree(p