Java如何构建菜单树？

清疚 2023-10-10 22:17 2阅读 0赞

> 在后台系统中，菜单栏有着严格的层次结构，它是系统方便与否的重要功能，像一级菜单，二级菜单，三级菜单甚至更多层级的菜单，我们将它称为`菜单树`，菜单树往往需要后端程序员来构建好了之后再给前端程序员，前端程序员再去渲染到页面上这样一个流程，而且菜单列表是动态变化的。

**那么该如何构建一个菜单树呢？**

在数据库中，menu 表最常见的字段

*  id（主键）
 *  parent\_id（父级ID）
 *  menu\_name（菜单名称）
 *  path（路径）
 *  ...（等等）

### 思路 ###

> 1.  我们可以把菜单列表返回，获取`parent_id`字段等于`0`的节点，称为`根节点`，这样的节点代表一级菜单
> 2.  再通过`根节点`的`主键`去寻找`子菜单`，因为要有多及菜单，所以要用`递归`构建子树，直到没有子菜单为止
> 3.  最后通过构建完整的菜单树

**menu 实体类**

java

复制代码

`public class Menu { private Long id; private Long parentId; private String name; private List<Menu> childrenList; public Menu(Long id, Long parentId, String name) { this.id = id; this.parentId = parentId; this.name = name; } // getter setter... }`

**构建菜单树分为以下几个步骤：**

1.  返回菜单的根节点

java

复制代码

`private List<Menu> getRootNode() { List<Menu> rootNode = new ArrayList<>(); menuList.forEach(item -> { if (item.getParentId() == 0) { rootNode.add(item); } }); return rootNode; }`

1.  构建子菜单树

java

复制代码

`private Menu builderChildrenNode(Menu rootNode) { List<Menu> childrenList = new ArrayList<>(); menuList.forEach(item -> { if (Objects.equals(item.getParentId(), rootNode.getId())) { // 还需要遍历三级菜单以后的 Menu menu = builderChildrenNode(item); childrenList.add(menu); } }); rootNode.setChildrenList(childrenList); return rootNode; }`

1.  构建菜单树

java

复制代码

`public List<Menu> buildTree() { List<Menu> menus = getRootNode(); menus.forEach(this::builderChildrenNode); return menus; }`

1.  完整代码

java

复制代码

`public class MenuTree { private final List<Menu> menuList; public MenuTree(List<Menu> menuList) { this.menuList = menuList; } /** * 获取根结点 * * @return */ private List<Menu> getRootNode() { List<Menu> rootNode = new ArrayList<>(); menuList.forEach(item -> { if (item.getParentId() == 0) { rootNode.add(item); } }); return rootNode; } /** * 构建子树 * * @param rootNode * @return */ private Menu builderChildrenNode(Menu rootNode) { List<Menu> childrenList = new ArrayList<>(); menuList.forEach(item -> { if (Objects.equals(item.getParentId(), rootNode.getId())) { // 还需要遍历三级菜单以后的 Menu menu = builderChildrenNode(item); childrenList.add(menu); } }); rootNode.setChildrenList(childrenList); return rootNode; } /** * 构建树 * * @return */ public List<Menu> buildTree() { List<Menu> menus = getRootNode(); menus.forEach(this::builderChildrenNode); return menus; } }`

**测试**

用到了`fastjson2`依赖

xml

复制代码

`<dependency> <groupId>com.alibaba.fastjson2</groupId> <artifactId>fastjson2</artifactId> <version>2.0.28</version> </dependency>`

java

复制代码

`public class MenuTreeTest { public static void main(String[] args) { List<Menu> menuList = new ArrayList<>(); menuList.add(new Menu(1L, 0L, "一级菜单1")); menuList.add(new Menu(2L, 0L, "一级菜单2")); menuList.add(new Menu(3L, 0L, "一级菜单3")); menuList.add(new Menu(4L, 1L, "二级菜单1")); menuList.add(new Menu(5L, 1L, "二级菜单2")); menuList.add(new Menu(6L, 4L, "三级菜单")); menuList.add(new Menu(7L, 6L, "四级菜单")); MenuTree menuTree = new MenuTree(menuList); List<Menu> menus = menuTree.buildTree(); String jsonString = JSON.toJSONString(menus); System.out.println(jsonString); } }`

**结果**

json

复制代码

`[{ "childrenList": [{ "childrenList": [{ "childrenList": [{ "childrenList": [], "id": 7, "name": "四级菜单", "parentId": 6 }], "id": 6, "name": "三级菜单", "parentId": 4 }], "id": 4, "name": "二级菜单1", "parentId": 1 }, { "childrenList": [], "id": 5, "name": "二级菜单2", "parentId": 1 }], "id": 1, "name": "一级菜单1", "parentId": 0 }, { "childrenList": [], "id": 2, "name": "一级菜单2", "parentId": 0 }, { "childrenList": [], "id": 3, "name": "一级菜单3", "parentId": 0 }]`

作者：饕餮者也  
链接：https://juejin.cn/post/7236687156067172409