如何使用Golang实现组合模式_Golang组合模式树形结构管理方法

package main

import "fmt"

// Component 接口
type FileSystemNode interface {
    Print(string) // 打印当前节点，接受缩进前缀
}

// Leaf 叶子节点：文件
type File struct {
    name string
}

func (f *File) Print(indent string) {
    fmt.Println(indent + f.name)
}

// Composite 容器节点：文件夹
type Directory struct {
    name     string
    children []FileSystemNode
}

func (d *Directory) Add(child FileSystemNode) {
    d.children = append(d.children, child)
}

func (d *Directory) Remove(child FileSystemNode) {
    for i, c := range d.children {
        if c == child {
            d.children = append(d.children[:i], d.children[i+1:]...)
            break
        }
    }
}

func (d *Directory) Print(indent string) {
    fmt.Println(indent + d.name)
    for _, child := range d.children {
        child.Print(indent + "  ")
    }
}

func main() {
    root := &Directory{name: "root"}
    src := &Directory{name: "src"}
    bin := &Directory{name: "bin"}

    mainFile := &File{name: "main.go"}
    utilFile := &File{name: "util.go"}

    src.Add(mainFile)
    src.Add(utilFile)
    root.Add(src)
    root.Add(bin)
    root.Add(&File{name: "README.md"})

    root.Print("")
}

root
  src
    main.go
    util.go
  bin
  README.md