2021年12月13日

tree/MenuNode.go

-package main
+package tree
 
 import "fmt"
 
-type treeMenuNode struct {
-	name          string
-	perent, child *treeMenuNode
+type TreeMenuNode struct {
+	Name          string
+	Perent, Child *TreeMenuNode
 }
 
 /*
@@ -13,62 +13,15 @@ type treeMenuNode struct {
 
 */
 // 值接收者
-func (node treeMenuNode) print() {
-	fmt.Println(node.name)
+func (node TreeMenuNode) Print() {
+	fmt.Println(node.Name)
 }
 
 //指针接收者
-func (node *treeMenuNode) setName(name string) {
+func (node *TreeMenuNode) SetName(name string) {
 	if node == nil {
 		fmt.Println(" Setting value to nil node. Ignorde.")
 		return
 	}
-	node.name = name
-}
-
-// 遍历
-func (node *treeMenuNode) traverse() {
-	if node == nil {
-		return
-	}
-	node.perent.traverse()
-	node.print()
-	node.child.traverse()
-}
-
-func main() {
-
-	var menu treeMenuNode
-
-	menu = treeMenuNode{name: "角色管理"}
-	menu.perent = &treeMenuNode{}
-	menu.child = &treeMenuNode{"增加用户", nil, nil}
-	menu.child.child = new(treeMenuNode)
-
-	org := treeMenuNode{name: "xx研发中心"}
-	org.child = &treeMenuNode{"A组", nil, nil}
-
-	nodes := []treeMenuNode{
-		{name: "xxx系统"},
-		{name: "测试"},
-		{"系统管理", &menu, &org},
-	}
-	fmt.Println(nodes)
-
-	fmt.Println("调用自定义方法： ")
-
-	menu.print()
-	menu.child.child.setName("自定义名称")
-	menu.child.child.print()
-
-	var xMenu *treeMenuNode
-	xMenu.setName("测试参数")
-	xMenu = &org
-	xMenu.setName("测试参数2")
-	xMenu.print()
-
-	fmt.Println("  遍历菜单树")
-	menu.traverse()
-	org.traverse()
-
+	node.Name = name
 }

tree/Tranverse.go

+package tree
+
+// 遍历
+func (node *TreeMenuNode) Traverse() {
+	if node == nil {
+		return
+	}
+	node.Perent.Traverse()
+	node.Print()
+	node.Child.Traverse()
+}