10行代码搞定树形结构数据

冷不防 2024-04-18 08:38 70阅读 0赞

**我们经常会有返回树形结构数据的需求。比如这里的菜单表，第一级是目录，目录下面是菜单，菜单下面有按钮。如果我们要返回一个集合，包含目录，目录下面嵌套菜单，菜单下面嵌套按钮。使用Stream API可以很方便的解决这个问题。**

**注意：这里我们的菜单上下级之间以parent\_id来关联，parent\_id是指上一级菜单的id，顶级菜单的id为0。**

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTQwMjA1_size_16_color_FFFFFF_t_70][]

**定义包含下级菜单的对象**

继承自SysMenuEntity对象，增加了一个children属性，用于存储下级菜单。

@Data
    public class SysMenuNode extends SysMenuEntity {
    
        private List<SysMenuNode> children;
    }

**定义获取树形结构的方法**

我们先过滤出parent\_id为0的顶级菜单，然后给每个顶级菜单设置其子级菜单，covert方法的作用是从所有菜单中找出相应菜单的子级菜单

/**
         * 获取菜单树形列表
         */
        @GetMapping("/treeList")
        public List<SysMenuNode> treeList() {
            List<SysMenuEntity> sysMenuList = sysMenuService.list();
             return sysMenuList.stream().filter(e -> e.getParentId().equals(0L)).map(e -> covert(e, sysMenuList)).collect(Collectors.toList());
        }

**为每个菜单设置子级菜单**

这里我们使用filter操作来过滤出每个菜单的子级菜单，由于子级菜单下面可能还会有子级菜单，这里我们使用递归来解决。但是递归操作什么时候停止了? 这里把递归调用方法放到了map操作中去，当没有子级菜单时filter下的map操作便不会再执行，从而停止递归。

/**
         * 递归封装children属性
         * 当filter条件不满足时跳出递归
         */
    private SysMenuNode covert(SysMenuEntity sysMenu, List<SysMenuEntity> sysMenuList) {
            SysMenuNode node = new SysMenuNode();
            BeanUtils.copyProperties(sysMenu, node);
            List<SysMenuNode> children = sysMenuList.stream().filter(e -> e.getParentId().equals(sysMenu.getMenuId())).map(e -> covert(e, sysMenuList)).collect(Collectors.toList());
            node.setChildren(children);
            return node;
    }

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTQwMjA1_size_16_color_FFFFFF_t_70 1][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTQwMjA1_size_16_color_FFFFFF_t_70]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/17/c12b42419f7f4b64808cc68a614ec047.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTQwMjA1_size_16_color_FFFFFF_t_70 1]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/17/5c31a45424ce4770bc4aa65366785756.png