-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.go
75 lines (67 loc) · 1.44 KB
/
utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package tests
import (
mptt "github.com/boycs007/gorm-mptt"
"gorm.io/gorm/clause"
)
func dfs(node *Node, f func(n *Node) (int, error)) error {
if node == nil {
return nil
}
id, err := f(node)
if err != nil {
return err
}
for _, child := range node.Children {
child.ParentID = id
if err = dfs(child, f); err != nil {
return err
}
}
return nil
}
func bfs(root *Node, f func(n *Node) (int, error)) error {
if root == nil {
return nil
}
queue := []*Node{root}
for len(queue) > 0 {
node := queue[0]
queue = queue[1:]
id, err := f(node)
if err != nil {
return err
}
for _, child := range node.Children {
child.ParentID = id
queue = append(queue, child)
}
}
return nil
}
func createNode(manager mptt.TreeManager, node *Node) (int, error) {
ctNode := &CustomTree{
ModelBase: mptt.ModelBase{
ParentID: node.ParentID,
},
Name: node.Name,
}
e := manager.CreateNode(ctNode)
return ctNode.ID, e
}
func getItemByName(manager mptt.TreeManager, name string) (*CustomTree, error) {
var ret CustomTree
err := manager.GormDB().Model(&ret).Where(clause.Eq{
Column: "name",
Value: name,
}).First(&ret).Error
return &ret, err
}
func getAllNodes(manager mptt.TreeManager) (map[string]*CustomTree, error) {
var ret []*CustomTree
err := manager.GormDB().Model(new(CustomTree)).Find(&ret).Error
retMap := make(map[string]*CustomTree)
for _, item := range ret {
retMap[item.Name] = item
}
return retMap, err
}