记一个递归封装树形结构

184次阅读
没有评论

最近改了个小bug,原有的数据字典查询,封装成树的递归写的有问题,就自己写了一个,解决这种递归的思考是:分析树形结构-->>找到父子层之间的关联关系-->>根据关系编写递归嵌套条件。代码很简单也很实用,遂总结如下

表结构

截取表结构如下,关联id 和pid(父id).其中树最外层pid为0

记一个递归封装树形结构

代码逻辑

public List<Map<String, Object>> getRootDict() {
    List<Map<String, Object>> returnList = new ArrayList<>();
    // 查询所有数据字典
    List<Map<String, Object>> dictList = dictMapper.getAllDict();
    Map<String, Object> map = new HashMap<>(16);
    map.put("name", "数据字典");
    map.put("id", null);
    map.put("pid", null);

    // 构建树形结构
    List<Map<String, Object>> rootTree = this.rebuildRootTree("0",dictList);
    map.put("child", rootTree);
    returnList.add(map);
    return returnList;
}

/**
 * 构建树结构
 * @param pid
 * @param dictList
 * @return
 */
List<Map<String, Object>> rebuildRootTree(String pid ,List<Map<String, Object>> dictList){
    List<Map<String, Object>> rootTree = new ArrayList<>();
    for(Map<String, Object> dictMap:dictList){
        if(dictMap.containsKey("pid") && Objects.equals(dictMap.get("pid").toString(),pid)) {
            Map<String, Object> newMap = new HashMap<>(16);
            newMap.put("name", dictMap.get("name"));
            newMap.put("id", dictMap.get("id"));
            newMap.put("pid", dictMap.get("pid"));
            //递归调用
            newMap.put("child",this.rebuildRootTree((dictMap.get("id")).toString(),dictList));
            rootTree.add(newMap);
        }
    }
    return rootTree;
}

若为实体,替换为对应的实体类即可;好了,就酱!日常生活的小技巧,还是总结下

viEcho
版权声明:本站原创文章,由viEcho2021-04-17发表,共计1167字。
转载提示:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(没有评论)
载入中...