Golang Composite树形结构组合模式实践

package main

import "fmt"

// Component 定义统一接口
type Component interface {
    Print(indent string)
    GetName() string
}

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

func (f *File) Print(indent string) {
    fmt.Printf("%s- %s\n", indent, f.name)
}

func (f *File) GetName() string {
    return f.name
}

// Composite 容器节点：目录
type Directory struct {
    name      string
    children  []Component
}

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

func (d *Directory) Print(indent string) {
    fmt.Printf("%s+ %s/\n", indent, d.name)
    for _, child := range d.children {
        child.Print(indent + "  ")
    }
}

func (d *Directory) GetName() string {
    return d.name
}

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

    root.Add(src)
    root.Add(bin)

    mainFile := &File{name: "main.go"}
    utils := &Directory{name: "utils"}
    helper := &File{name: "helper.go"}

    src.Add(mainFile)
    src.Add(utils)
    utils.Add(helper)

    // 统一打印整个结构
    root.Print("")
}

+ root/
  + src/
    - main.go
    + utils/
      - helper.go
  + bin/