java实现构造无限层级树形菜单

Java实现构造无限层级树形菜单的攻略

背景

在Web开发中，经常会用到树形菜单，这种菜单常常存在多级目录结构，是一种比较典型和重要的web组件。如何实现树形菜单呢？

算法

实现树形菜单的算法其实比较简单，一般情况下采用递归的方式实现。基本思路是：先找到父节点，再找到其下面的子节点，然后再利用递归进行处理，以此达到构造完整的树形结构。

代码示例

下面给出一个简单的代码示例来说明如何构造无限层级树形菜单：

public class TreeView {
    private Long id;
    private String label;
    private List<TreeView> children;

    // getter setter 省略...

    public static List<TreeView> buildTree(List<Menu> menus) {
        // 先获取根节点
        List<TreeView> roots = new ArrayList<>();
        for (Menu menu : menus) {
            if (menu.getParentId() == null) {
                TreeView root = new TreeView();
                root.setId(menu.getId());
                root.setLabel(menu.getName());
                root.setChildren(findChildren(root, menus));
                roots.add(root);
            }
        }
        return roots;
    }

    private static List<TreeView> findChildren(TreeView parent, List<Menu> menus) {
        List<TreeView> children = new ArrayList<>();
        for (Menu menu : menus) {
            if (parent.getId().equals(menu.getParentId())) {
                TreeView child = new TreeView();
                child.setId(menu.getId());
                child.setLabel(menu.getName());
                child.setChildren(findChildren(child, menus));
                children.add(child);
            }
        }
        return children;
    }
}

以上代码中，TreeView是树形结构的对象，而Menu则是数据对象。buildTree方法用来构造整个树形结构，它的逻辑是先找到根节点，然后递归找到其下面的子节点。findChildren方法则是递归的核心实现，根据父节点进行查找子节点，如果有则把子节点加入到children中，并再次递归查找子节点的子节点。

以上代码只是基本递归实现，实际使用中还需要考虑多种情况，比如节点重复、数据异常等问题。

下面再给出一个使用示例，通过构造一个菜单列表，来构造树形结构：

public class Main {
    public static void main(String[] args) {
        List<Menu> menus = new ArrayList<>();
        menus.add(new Menu(1L, "系统管理"));
        menus.add(new Menu(2L, "用户管理", 1L));
        menus.add(new Menu(3L, "角色管理", 1L));
        menus.add(new Menu(4L, "菜单管理", 1L));
        menus.add(new Menu(5L, "添加用户", 2L));
        menus.add(new Menu(6L, "修改用户", 2L));
        menus.add(new Menu(7L, "删除用户", 2L));

        List<TreeView> tree = TreeView.buildTree(menus);
        for (TreeView t : tree) {
            System.out.println(t.getLabel());
            if (t.getChildren() != null) {
                for (TreeView child : t.getChildren()) {
                    System.out.println("--" + child.getLabel());
                    print(child.getChildren(), "--" + "--");
                }
            }
        }

    }

    private static void print(List<TreeView> children, String prefix) {
        if (children == null) {
            return;
        }
        for (TreeView child : children) {
            System.out.println(prefix + child.getLabel());
            print(child.getChildren(), prefix + "--");
        }
    }
}

以上代码中，首先定义了一些菜单数据，然后调用TreeView.buildTree方法构造出树形结构。最后通过递归输出整个树形结构。

总结

实现无限层级树形菜单的算法基本思路是使用递归，在找到父节点和子节点后，利用递归进行处理，以此达到构造完整的树形结构。递归实现的代码容易编写，但也需要注意多种细节问题，比如不合法的数据导致递归死循环等问题，需格外注意。