记一个递归封装树形结构叁有三分之一

记一个递归封装树形结构

372次阅读

最近改了个小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;
}