Java 实现 多级菜单

一:前言

最近老师布置了给多级菜单的作业，感觉蛮有意思的，可以提升自己的逻辑！下面我写个简易版的多级菜单，本人还是菜鸟，欢迎各位给予宝贵的建议！

二：正文

由于是给各位演示，所有我把涉及的其他条件全省略了，只做了给最简单的，以便大家能更好的理解我的思路

1,首先是数据库的设计

DROP TABLE IF EXISTS `t_category`;
CREATE TABLE `t_category` (
  `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '标识目录的唯一id',
  `parent_id` int(11) DEFAULT NULL COMMENT '存储父级目录的id，0代表一级目录',
  `name` varchar(255) COLLATE utf8_bin DEFAULT NULL COMMENT '目录的名称',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin;

2.java后端代码

① 对应数据库的实体类

public class Category {
    private int id;
    private int parentId;
    private String name;
}

② 返回给前端的实体类  //注意写一个有参的构造方法，方便我们把Category的数据注入进来

public class CategoryTwo extends Category{
    private List<CategoryTwo> categorieTwos;

    public CategoryTwo(){}

    //这个有参构造方法方便我们为属性赋值即把categoy的成员变量的值赋值到自身中
    public CategoryTwo(Category category){
        super(category.getId(),category.getParentId(),category.getName());
    }
}

③ 具体实现

//多级目录
    public void findCategoryAll(){

        //查询出所有的目录数据
        List<Category> categorys = categoryMapper.findCategoryAll(null);//这个大家自己完成


        //存放所有一级目录
        List<CategoryTwo> categoryTows = new ArrayList();

        //找出一级目录，即 parentId = 0 的目录
        for(Category category :categorys){
            if(category.getParentId() == 0){
                CategoryTwo categoryTwo = new CategoryTwo(category);
                categoryTows.add(categoryTwo);//存入一级目录集合中
            }
        }

        //利用递归，找出所有一级目录的List<CategoryTwo>,这里我把具体实现包装到 findSubCategory（）方法中了
        categoryTows = findSubCategory(categorys,categoryTows);

        //最后就完成了，把数据按需求返回给前端就可以了。
        System.out.println(categoryTows);
    }

    //找出List<CategoryTwo>中所有目录的子级目录
    public List<CategoryTwo> findSubCategory(List<Category> categorys,List<CategoryTwo> categoryTwos){

        for (CategoryTwo categoryTwo:categoryTwos) {
            //找出自身的子级目录，存放到List<CategoryTwo>中去
            List<CategoryTwo> categoryTwos2 = new ArrayList<CategoryTwo>();

            for (Category category:categorys){
                //如果自身的id = 与某个目录的parentId 相等，则这个目录是自身的子级目录
                if(categoryTwo.getId() == category.getParentId()){
                    categoryTwos2.add(new CategoryTwo(category));
                }
            }

            //所有子级目录找到后，再回调findSubCategory()方法 让子级目录找到自己的子级
            List<CategoryTwo> categoryTwos3 = findSubCategory(categorys,categoryTwos2);

            //categoryTwos3真正的List<categoryTwo>，把他添加到自身
            categoryTwo.setCategorieTwos(categoryTwos3);
        }

        return categoryTwos;//返回注入好的对象
    }

三：总结

核心就是那个回调的方法，只要把这个搞清楚了，那就很容易了