Java构建一级二级菜单

全栈_千凝

于 2023-07-27 10:18:16 发布

阅读量334

点赞数

文章标签： java 开发语言

本文链接：https://blog.csdn.net/weixin_46428363/article/details/131954294

版权

人狠话不多直接上代码

    /**
     * 栏目菜单结构分类树
     * @return
     */
    @Override
    public List<Category> listWithTree() {
        // 1、查询所有分类
        List<Category> categoryList = categoryMapper.selectCategoryList(null);
        // 2、构建分类id与分类对象的映射关系
        Map<Long, List<Category>> categoryMap = new HashMap<>();
        for (Category category : categoryList) {
            Long parentId = category.getParentId();
            // 将 parentId 为 null 转换为 0L，表示根分类
            if (parentId == null) {
                parentId = 0L;
            }
            categoryMap.computeIfAbsent(parentId, k -> new ArrayList<>()).add(category);
        }
        // 3、构建分类树形结构
        List<Category> rootCategories = categoryMap.get(0L);
        if (rootCategories != null) {
            for (Category rootCategory : rootCategories) {
                buildCategoryTree(rootCategory, categoryMap);
            }
        }
        return rootCategories;
    }

    /**
     * 递归构建分类树形结构
     */
    private void buildCategoryTree(Category category, Map<Long, List<Category>> categoryMap) {
        List<Category> children = categoryMap.get(category.getCategoryId());
        if (children != null) {
            for (Category child : children) {
                buildCategoryTree(child, categoryMap);
            }
        }
        category.setCategory(children);
    }