概要

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生成无限树

• 编辑
• 预览