一、商品分类模块里递归算法的使用场景
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 id="selectCategoriesByParentId" resultType="top.kongk.mmall.pojo.Category">
select
<include refid="Base_Column_List" />
from mmall_category
where parent_id = #{parentId,jdbcType=INTEGER}
</select>
2. 原项目的代码与不足:
原项目中的获取所有子节点的代码是这样写的：
public ServerResponse<List<Integer>> selectCategoryAndChildrenById(Integer categoryId){
Set<Category> categorySet = Sets.newHashSet();
findChildCategory(categorySet,categoryId);
List<Integer> categoryIdList = Lists.newArrayList();
if(categoryId != null){
for(Category categoryItem : categorySet){
categoryIdList.add(categoryItem.getId());
}
}
return ServerResponse.createBySuccess(categoryIdList);
}
//递归算法,算出子节点
private Set<Category> findChildCategory(Set<Category> categorySet ,Integer categoryId){
Category category = categoryMapper.selectByPrimaryKey(categoryId);
if(category != null) {
categorySet.add(category);
}
//查找子节点,递归算法一定要有一个退出的条件
List<Category> categoryList = categoryMapper.selectCategoryChildrenByParentId(categoryId);
for(Category categoryItem : categoryList){
findChildCategory(categorySet,categoryItem.getId());
}
return categorySet;
}
原项目的代码有两个不足之处：
1.本来已经查到了 某个节点 封装好的的 category 对象，递归的时候直接传 category 即可。原代码中却只传了 category 的 id ，然后再从数据库中获取主键为该 id 的数据，再封装成原来的 category，多此一举！
2.本来前台要求的是List<Integer>，原代码中却用 Set<Category> 保存节点。最后返回前台的时候，再新 new 一个 List，遍历 Set，把Set 里的值添加到 List中。Set 是多余的！
3. 改进：
@Override
public ServerResponse getDeepCategory(Integer categoryId) {
Category category = categoryMapper.selectByPrimaryKey(categoryId);
List<Integer> categories = new LinkedList<>();
findAllChildrenCategories(categories, category);
if (CollectionUtils.isEmpty(categories)) {
return ServerResponse.createErrorWithMsg("获取商品类别失败");
} else {
return ServerResponse.createSuccess(categories);
}
}
private void findAllChildrenCategories(List<Integer> categories, Category category) {
if (category != null) {
categories.add(category.getId());
} else {
return;
}
//获取当前 category 的第一层孩子
List<Category> categoryList = categoryMapper.selectCategoriesByParentId(category.getId());
//对第一层的每一个孩子递归调用此方法
for (Category categoryItem : categoryList) {
findAllChildrenCategories(categories, categoryItem);
}
}
完。
网友评论