Gorm 实现无限树形菜单

原文链接：https://www.zhoubotong.site/post/91.html
通常树形菜单的实现基本就是递归调用，大部分场景毕竟这种数据不多，性能倒是并不突出， 下面给个demo,有兴趣的朋友可以看看： 新建一个city表：

1. CREATE TABLE `city` (
2.   `id` int NOT NULL AUTO_INCREMENT,
3.   `pid` int NOT NULL DEFAULT '0' COMMENT '父级别id',
4.   `name` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NOT NULL DEFAULT '0' COMMENT '名称',
5.   PRIMARY KEY (`id`) USING BTREE,
6.   KEY `pid` (`pid`) USING BTREE
7. ) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;

复制代码

测试数据：

1. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (1, 0, '广东省');
2. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (2, 1, '广州市');
3. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (3, 2, '广州天河区');
4. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (4, 2, '海珠区');
5. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (5, 0, '福建省');
6. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (6, 5, '厦门市');
7. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (7, 4, '新港');
8. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (8, 4, '沙园');
9. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (9, 7, '新港一区');
10. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (10, 0, '湖北省');
11. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (11, 10, '武汉市');
12. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (12, 11, '黄陂区');
13. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (13, 11, '洪山区');
14. INSERT INTO `city` (`id`, `pid`, `name`) VALUES (14, 12, '黄陂天河街道');

复制代码

demo.go：

1. package main
3. import (
4.     "encoding/json"
5.     "fmt"
7.     "gorm.io/driver/mysql"
8.     "gorm.io/gorm"
9. )
11. var DB gorm.DB
13. func (u TreeNode) TableName() string {
14.     return "city"
15. }
17. type TreeNode struct {
18.     Id       int         `json:"id"`
19.     Pid      int         `json:"pid" gorm:"index"`
20.     Name     string      `json:"name" gorm:"size:255"`
21.     Children []*TreeNode `json:"children" gorm:"-"`
22. }
24. func init() {
25.     dsn := "root:uos123456@tcp(11213:3306)/demo?charset=utf8mb4&parseTime=True&loc=Local"
26.     db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
27.     if err != nil {
28.         panic("failed to connect database")
29.     }
30.     DB = *db
31. }
32. func main() {
34.     data := TreeList(10) // 查询武汉市下的子节点
35.     str, _ := json.Marshal(data)
36.     fmt.Printf("%s\\n", str)
38. }
40. // TreeList 递归处理子节点list
41. func TreeList(pid int) []*TreeNode {
42.     var list []*TreeNode
43.     DB.Model(TreeNode{}).Where("pid= ?", pid).Find(&list)
44.     result := make([]*TreeNode, 0)
45.     for _, v := range list {
46.         if v.Pid == pid {
47.             v.Children = TreeList(v.Id)
48.             if v.Children == nil {
49.                 v.Children = make([]*TreeNode, 0)
50.             }
51.             result = append(result, v)
52.         }
53.     }
55.     return result
57. }

复制代码

输出结果：

1. [Running] go run "/media/uos/G/web/demo-go/demo.go"
2. [{"id":11,"pid":10,"name":"武汉市","children":[{"id":12,"pid":11,"name":"黄陂区","children":[{"id":14,"pid":12,"name":"黄陂天河街道","children":[]}]},{"id":13,"pid":11,"name":"洪山区","children":[]}]}]

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！