[慕课电商] 04.对商品分类模块递归算法的改进

一、商品分类模块里递归算法的使用场景

1.商品类别表的部分数据:

   id   parent_id  name             
------  ---------  -----------------
100001          0  家用电器     
100002          0  数码3C         
100003          0  服装箱包     
    
100006     100001  冰箱           
100007     100001  电视           
100008     100001  洗衣机        
100009     100001  空调           
100010     100001  电热水器  
  
100011     100002  电脑           
100012     100002  手机   
       
100037     100009  格力空调     
100038     100009  美的空调 

100039     100037  格力挂壁式空调
100040     100037  格力立柜式空调

id 字段是主键，parent_id 字段是该类别的父节点的 id 。现在有个需求就是给一个 id ，返回他自身以及他的所有子节点（包括子节点的子节点。。。。）

比如，查询 id 为 100001（家用电器）的所有子节点。那么结果应该是：

--家用电器
    --冰箱
    --电视
    --洗衣机
    --空调
         --格力空调
             --格力挂壁式空调
             --格力立柜式空调
         --美的空调
    --电热水器

这种需求转化为代码的话，递归调用是最方便的。

二、把需求转化为代码

1. 逻辑:

首先从数据库中获取 parent_id 为家用电器的数据，我们得到的是一个 list 集合（冰箱，电视，洗衣机，空调，电热水器）。然后对 list 集合中的每一个 item 进行遍历，在遍历的时候去从数据库中获取 parent_id 为 item 的集合，如果集合不是空的，那么再对这个集合再遍历。。。一直照此往复。。。

我们可以从这个总的过程中抽离出一个核心的过程：

   按照某个条件获取一个集合，如果集合不为空，对集合中的 每一项 再 按照原来相同的条件 获取集合。

编写 sql 语句，查询表中 parent_id 为指定 id 的所有记录：

    select
    
    from mmall_category
    where parent_id = #{parentId,jdbcType=INTEGER}
  

2. 原项目的代码与不足:

原项目中的获取所有子节点的代码是这样写的：

public ServerResponse> selectCategoryAndChildrenById(Integer categoryId){
        Set categorySet = Sets.newHashSet();
        findChildCategory(categorySet,categoryId);

        List categoryIdList = Lists.newArrayList();
        if(categoryId != null){
            for(Category categoryItem : categorySet){
                categoryIdList.add(categoryItem.getId());
            }
        }
        return ServerResponse.createBySuccess(categoryIdList);
    }

    //递归算法,算出子节点
    private Set findChildCategory(Set categorySet ,Integer categoryId){
        Category category = categoryMapper.selectByPrimaryKey(categoryId);
        if(category != null) {
            categorySet.add(category);
        }
        //查找子节点,递归算法一定要有一个退出的条件
        List categoryList = categoryMapper.selectCategoryChildrenByParentId(categoryId);
        for(Category categoryItem : categoryList){
            findChildCategory(categorySet,categoryItem.getId());
        }
        return categorySet;
    }

原项目的代码有两个不足之处：

1.本来已经查到了 某个节点 封装好的的 category 对象，递归的时候直接传 category 即可。原代码中却只传了 category 的 id ，然后再从数据库中获取主键为该 id 的数据，再封装成原来的 category，多此一举！

2.本来前台要求的是List，原代码中却用 Set 保存节点。最后返回前台的时候，再新 new 一个 List，遍历 Set，把Set 里的值添加到 List中。Set 是多余的！

3. 改进：

@Override
    public ServerResponse getDeepCategory(Integer categoryId) {
        Category category = categoryMapper.selectByPrimaryKey(categoryId);

        List categories = new LinkedList<>();

        findAllChildrenCategories(categories, category);

        if (CollectionUtils.isEmpty(categories)) {
            return ServerResponse.createErrorWithMsg("获取商品类别失败");
        } else {
            return ServerResponse.createSuccess(categories);
        }

    }

    private void findAllChildrenCategories(List categories, Category category) {
        if (category != null) {
            categories.add(category.getId());
        } else {
            return;
        }
        //获取当前 category 的第一层孩子
        List categoryList = categoryMapper.selectCategoriesByParentId(category.getId());

        //对第一层的每一个孩子递归调用此方法
        for (Category categoryItem : categoryList) {
            findAllChildrenCategories(categories, categoryItem);
        }
    }

完。