Java常用写法-类目树的生成与遍历

怼烎@ 2024-03-16 09:48 5阅读 0赞

假设业务：店铺系统，每个店铺都有各种商品，商品分几种层级类目，比如：根目录是某某店铺；第一层级目录有数码产品、日用百货等等；第二层级，数码产品下面有笔记本电脑、手机等等；第三层级，笔记本电脑下面有各种品牌。。。

## 1. 对象准备 ##

### 1.1 表结构： ###

<table style="width:500px;"> 
 <tbody> 
  <tr> 
   <td>主键（目录id）</td> 
   <td>创建时间</td> 
   <td>修改时间</td> 
   <td>店铺id</td> 
   <td>父目录id</td> 
   <td>目录名</td> 
   <td>层级</td> 
   <td>是否删除</td> 
  </tr> 
  <tr> 
   <td>1</td> 
   <td>date</td> 
   <td>date</td> 
   <td>1</td> 
   <td>-1</td> 
   <td>某某店铺（因为是根目录，名字和店铺名一样即可）</td> 
   <td>0</td> 
   <td>N</td> 
  </tr> 
  <tr> 
   <td>2</td> 
   <td>date</td> 
   <td>date</td> 
   <td>1</td> 
   <td>1</td> 
   <td>数码产品</td> 
   <td>1</td> 
   <td>N</td> 
  </tr> 
  <tr> 
   <td>3</td> 
   <td>date</td> 
   <td>date</td> 
   <td>1</td> 
   <td>1</td> 
   <td>日用百货</td> 
   <td>1</td> 
   <td>N</td> 
  </tr> 
  <tr> 
   <td>4</td> 
   <td>date</td> 
   <td>date</td> 
   <td>1</td> 
   <td>2</td> 
   <td>笔记本电脑</td> 
   <td>2</td> 
   <td>N</td> 
  </tr> 
  <tr> 
   <td>5</td> 
   <td>date</td> 
   <td>date</td> 
   <td>1</td> 
   <td>4</td> 
   <td>拉里牌电脑</td> 
   <td>3</td> 
   <td>N</td> 
  </tr> 
 </tbody> 
</table>

### 1.2 实体类结构 ###

public class shopDirDO{
        Long id;// 主键，目录id
        Date createTime;// 创建时间
        Date modifiedTime;// 修改时间
        Long directoryParentId;// 父目录id
        String directoryName;// 目录名称
        Long shopId;// 商铺id
        Integer level;// 层级
        String isDeleted;// 是否删除
    }

### 1.3 目录树结构（VO）： ###

public class ShopDirVO{
        Long id;// 目录id
        Long parentDirectoryId;// 父目录id
        String directoryName;// 目录名称
        Integer level;// 层级
        List<ShopDirVO> childrenDirectories;// 子目录信息
    }

## 2. 代码部分 ##

### 2.1 生成目录树VO对象 ###

// 根据商铺id查出所有目录信息
    List<ShopDirDO> dirDOList = Mapper.selectByShopId(shopId);
    
    // 以下是创建目录树
    // 创建id->VO map
    Map<Long, ShopDirVO> dirIdVOMap = new HashMap();
    ShopDirVO dirVOTree = null;
    dirDOList.forEach(
        e->{
            // do转vo，详细省略
            ShopDirVO vo = DirConverter.DoToVO(e)
            dirIdVOMap.put(vo.getId(), vo);
        });
    // 遍历map
    for(Map.Entry<Long, ShopDirVO> entry : dirIdVOMap.entrySet()){
        ShopDirVO vo = entry.getValue();
        if(-1L == vo.getParentDirentoryId()){
            dirVOTree = vo;
            continue;
        }
        ShopDirVO parentVO = dirIdVOMap.get(vo.getParentDirentoryId());
        parentVO.getChildrenDirectories().add(vo);
    }
    
    // 得到的dirVOTree就是类目树

### 2.2 简单遍历案例1 ###

最基本的递归遍历示例，有遍历其他需求的话基本是在此基础上改动。可大致分为两部分，一部分更新想要的数据，一部分递归遍历。

// 根据目录树获取所有目录的id，dirIds变量初始调用时传空
    private void fetchDirIdsByDirVO(ShopDirVO tree, Set<Long> dirIds, int level){
        if(dirIds.contains(tree.getId())){
            return;
        }
        dirIds.add(tree.getId());
        for(ShopDirVO vo : tree.getChildrenDirectories()){
            fetchDirIdsByDirVO(vo, dirIds, level+1);
        }
    }