go通過數組（切片）構建菜單樹結構

有這樣的一組節點，每個節點包含自己的Id，還有父Id （Parent Id），包含children指針數組，但是children是空，需要根據id和parentId把cihldren填充上。

實現瞭如下的方法

type TreeNode interface {
    ID() int
    ParentID() int
    AppendChildren(interface{})
}
func BuildTree(array []TreeNode) TreeNode {
    maxLen := len(array)
    var rootNode TreeNode = nil
	///<找出根節點,根節點的特點，沒有父節點
    for i := 0; i < maxLen; i++ {
        ///< 統計每個節點的父節點出現的次數，父節點出現0次就是根節點
        count := 0
        for j := 0; j < maxLen; j++ {
            ///< 如果有節點的ID == i的parentID 那麼j就是父節點
            if array[j].ID() == array[i].ParentID() {
                count++
                array[j].AppendChildren(array[i])
            }
        }
        if count == 0 {
            rootNode = array[i]
        }
    }
    return rootNode
}

以下是測試使用的方法

type DataNode struct {
    Id       int         `json:"id"`
    ParentId int         `json:"parentId"`
    Children []*DataNode `json:"children"`
}

func (d *DataNode) ID() int {
    return d.Id
}

func (d *DataNode) ParentID() int {
    return d.ParentId
}

func (d *DataNode) AppendChildren(node interface{}) {
    d.Children = append(d.Children, node.(*DataNode))
}
func TestBuildTree(t *testing.T) {
    dataArr := []DataNode{
        DataNode{
            Id:       1,
            ParentId: 0,
        },
        DataNode{
            Id:       2,
            ParentId: 1,
        },
        DataNode{
            Id:       3,
            ParentId: 1,
        },
        DataNode{
            Id:       4,
            ParentId: 1,
        },
        DataNode{
            Id:       5,
            ParentId: 2,
        },
        DataNode{
            Id:       6,
            ParentId: 2,
        },
        DataNode{
            Id:       7,
            ParentId: 3,
        },
        DataNode{
            Id:       8,
            ParentId: 3,
        },
        DataNode{
            Id:       9,
            ParentId: 3,
        },
    }
    nodeArray := make([]TreeNode, len(dataArr))
    for i := 0; i < len(dataArr); i++ {
        nodeArray[i] = &dataArr[i]
    }
    rootNode := BuildTree(nodeArray)
    rootNodeByte, err := json.Marshal(rootNode)
    retStr := string(rootNodeByte)
    if err != nil {
        t.Fail()
    } else {
        if retStr == `{"id":1,"parentId":0,"children":[{"id":2,"parentId":1,"children":[{"id":5,"parentId":2,"children":null},{"id":6,"parentId":2,"children":null}]},{"id":3,"parentId":1,"children":[{"id":7,"parentId":3,"children":null},{"id":8,"parentId":3,"children":null},{"id":9,"parentId":3,"children":null}]},{"id":4,"parentId":1,"children":null}]}` {
            t.Log("pass")
        } else {
            t.Fail()
        }
    }

}