golang生成无限树

百里江山 · · 260 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

概要

  1. 定义无限结构体.使用指针方式 List []*TreeNode,见1处代码
  2. 构建parent_id为key的map结构.
  3. 构建树的结构.

源代码

package main

import (
    "encoding/json"
    "fmt"
)
// 1处
type TreeNode struct {
    Id   int             `json:"id"`
    PID  int             `json:"pid"`
    Name string          `json:"name"`
    List []*TreeNode `json:"list,omitempty"` // omitempty 表示如果空的话,则忽略.
}
func main() {
    test := `[{"id":1,"pid":0,"name":"a"},{"id":2,"pid":1,"name":"b"},{"id":3,"pid":1,"name":"c"},{"id":3,"pid":2,"name":"d"}]`
    var list []*TreeNode // 假设已经赋值进去,从 数据库取出的数据
    if err := json.Unmarshal([]byte(test), &list); err != nil {
        return
    }
    // a.构建map结构的数据.
    data := buildData(list)
    // b.构建树结构
    result := buildTree(0, data)
    temp, _ := json.Marshal(result)
    fmt.Println(string(temp))
}

// 构建parent_id为key的map结构.
func buildData(list []*TreeNode) map[int]map[int]*TreeNode {
    data := make(map[int]map[int]*TreeNode)
    for _, v := range list {
        id := v.Id // 主ID
        pid := v.PID // 父ID
        if _, ok := data[pid]; !ok {// 如果不存在则创建一个新节点
            data[pid] = make(map[int]*TreeNode)
        }
        data[pid][id] = v
    }
    return data
}
// 构建树的结构.
// a. 判断parent_id是否存在.
// b. 如果parent_id存在继续递归.至到data没有找到parent_id节点的数据.
func buildTree(parentId int, data map[int]map[int]*TreeNode) []*TreeNode {
    node := make([]*TreeNode, 0)
    for id, item := range data[parentId] {
        if data[id] != nil {
            item.List = buildTree(id, data)
        }
        node = append(node, item)
    }
    return node
}

参考

  1. https://www.zhangwenbing.com/blog/golang/15685

有疑问加站长微信联系

本文来自:简书

感谢作者:百里江山

查看原文:golang生成无限树

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:1006366459

260 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传