# (tree.go)[https://gitee.com/wgdong/ginfast/blob/master/util/tree.go]
## 功能
- 通用生成函数
- 自定义主节点
- 自定义关系依赖对象
## 测试例子
```go
package util
import (
"encoding/json"
"fmt"
"testing"
)
type Menu struct {
ID uint
PID uint
Name string
Children []Menu `json:"children,omitempty"`
}
func TestTree(t *testing.T) {
v := []Menu{{
ID: 1,
PID: 0,
Name: "0-1",
}, {
ID: 2,
PID: 1,
Name: "1-2",
}, {
ID: 3,
PID: 2,
}, {
ID: 4,
PID: 1,
Name: "1-4",
}, {
ID: 5,
PID: 2,
Name: "1-3",
}, {
ID: 6,
PID: 0,
Name: "0-6",
}}
r := TreeSlice(v, func(i int) bool { return v[i].PID == 1 }, func(i, j int) bool { return v[i].ID == v[j].PID })
b, _ := json.Marshal(r)
fmt.Printf("%s\n", b)
}
```
有疑问加站长微信联系（非本文作者）
