Golang树形结构管理方法详解

package main
<p>import "fmt"</p><p>// Component 接口定义统一的操作
type Component interface {
Display(int) // 显示当前节点，参数为缩进层级
GetName() string
}</p><p>// Leaf 叶节点，比如具体文件
type Leaf struct {
name string
}</p><p>func (l *Leaf) Display(level int) {
indent := ""
for i := 0; i < level; i++ {
indent += "  "
}
fmt.Printf("%s- %s\n", indent, l.name)
}</p><p>func (l *Leaf) GetName() string {
return l.name
}</p><p>// Composite 组合节点，比如目录
type Composite struct {
name     string
children []Component
}</p><p>func (c *Composite) Add(child Component) {
c.children = append(c.children, child)
}</p><p>func (c *Composite) Remove(name string) {
for i, child := range c.children {
if child.GetName() == name {
c.children = append(c.children[:i], c.children[i+1:]...)
break
}
}
}</p><p>func (c *Composite) Display(level int) {
indent := ""
for i := 0; i < level; i++ {
indent += "  "
}
fmt.Printf("%s+ %s\n", indent, c.name)
for _, child := range c.children {
child.Display(level + 1)
}
}</p><p>func (c *Composite) GetName() string {
return c.name
}
</p>

func main() {
    // 根目录
    root := &Composite{name: "root"}
<pre class="brush:php;toolbar:false"><code>// 子目录 /home
home := &amp;Composite{name: "home"}
userA := &amp;Leaf{name: "userA.txt"}
userB := &amp;Leaf{name: "userB.txt"}
home.Add(userA)
home.Add(userB)

// 子目录 /etc
etc := &amp;Composite{name: "etc"}
hosts := &amp;Leaf{name: "hosts"}
nginx := &amp;Composite{name: "nginx"}
nginx.Add(&amp;Leaf{name: "nginx.conf"})
etc.Add(hosts)
etc.Add(nginx)

// 添加到根目录
root.Add(home)
root.Add(etc)
root.Add(&amp;Leaf{name: "readme.md"})

// 显示整个结构
fmt.Println("文件系统结构：")
root.Display(0)

// 删除某个文件
fmt.Println("\n删除 userA.txt 后：")
home.Remove("userA.txt")
root.Display(0)</code>

}

文件系统结构：
+ root
  + home
    - userA.txt
    - userB.txt
  + etc
    - hosts
    + nginx
      - nginx.conf
  - readme.md
<p>删除 userA.txt 后：</p>

• root
  • home
    • userB.txt
  • etc
    • hosts
    • nginx
      • nginx.conf
  • readme.md