go语言中递归权限树的实现

模拟一个简单系统的权限设置部分，仅用于测试说明。

一、效果展示

二、数据表设计(其中pId表示父Id)

三、go语言实现

思路：以父结点Id得到所有子节点，进而递归得到所有子节点的子节点递归。貌似有点点绕，还是直接看代码吧！

声明：程序与数据库处理部分使用了GORM(https://github.com/jasperxu/gorm-cn-doc)，这里不做说明，可自行了解

(1)结构体

type Tree struct {
	Id       int     `json:"id"`
	Label    string  `json:"label"`
	Children []*Tree `json:"children"`
}

type Perm_info struct{
	Id int `gorm:"column:Id"`
	Name string `gorm:"column:name"`
	PId int `gorm:"column:pId"`
}

(2)实现部分

//获取所有权限
func GetAllPerm()(perms []model.Tree) {
	child := model.Tree{1, "所有权限", []*model.Tree{}}
	err:=getTreeNode(1, &child)
	if err != nil {
		return
	}
	perms=append(perms, child)
	return perms
}
//递归获取子节点
func getTreeNode(pId int, treeNode *model.Tree) error{
	var (
		baseGorm *db.BaseGorm
		perms    []model.Perm_info
	)
	perms, err := baseGorm.QueryPerm(pId)//根据父结点Id查询数据表，获取相应的子结点信息(使用gorm)
	if err != nil {
		return err
	}
	for i := 0; i < len(perms); i++ {
		child := model.Tree{perms[i].Id, perms[i].Name, []*model.Tree{}}
		treeNode.Children = append(treeNode.Children, &child)
		getTreeNode(perms[i].Id, &child)
	}
	return err
}

(3)打印结果

 "perms": [
        {
            "id": 1,
            "label": "所有权限",
            "children": [
                {
                    "id": 2,
                    "label": "业务管理",
                    "children": [
                        {
                            "id": 6,
                            "label": "查看",
                            "children": []
                        },
                        {
                            "id": 7,
                            "label": "新建",
                            "children": []
                        },
                        {
                            "id": 8,
                            "label": "修改",
                            "children": []
                        },
                        {
                            "id": 9,
                            "label": "删除",
                            "children": []
                        }
                    ]
                },
                {
                    "id": 3,
                    "label": "用户管理",
                    "children": [
                        {
                            "id": 10,
                            "label": "查看",
                            "children": []
                        },
                        {
                            "id": 11,
                            "label": "修改",
                            "children": []
                        },
                        {
                            "id": 12,
                            "label": "删除",
                            "children": []
                        }
                    ]
                },
                {
                    "id": 4,
                    "label": "角色管理",
                    "children": [
                        {
                            "id": 13,
                            "label": "查看",
                            "children": []
                        },
                        {
                            "id": 14,
                            "label": "新建",
                            "children": []
                        },
                        {
                            "id": 15,
                            "label": "修改",
                            "children": []
                        },
                        {
                            "id": 16,
                            "label": "删除",
                            "children": []
                        }
                    ]
                },
                {
                    "id": 5,
                    "label": "日志管理",
                    "children": [
                        {
                            "id": 17,
                            "label": "业务日志",
                            "children": []
                        },
                        {
                            "id": 18,
                            "label": "用户日志",
                            "children": []
                        },
                        {
                            "id": 19,
                            "label": "角色日志",
                            "children": []
                        }
                    ]
                }
            ]
        }
    ]