Golang如何使用组合模式构建树形结构_Golang Composite模式应用实践

package main

import "fmt"

// Component 接口：定义统一的操作
type Component interface {
    Print(indent string)
    Add(Component)
}

// File 叶节点
type File struct {
    name string
}

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

func (f *File) Add(c Component) {
    panic("不支持添加操作")
}

// Folder 组合节点
type Folder struct {
    name     string
    children []Component
}

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

func (fo *Folder) Add(c Component) {
    fo.children = append(fo.children, c)
}

func main() {
    // 创建根目录
    root := &Folder{name: "root"}

    // 创建子文件夹
    documents := &Folder{name: "Documents"}
    pictures := &Folder{name: "Pictures"}

    // 添加文件
    file1 := &File{name: "resume.doc"}
    file2 := &File{name: "photo.jpg"}

    documents.Add(file1)
    pictures.Add(file2)

    root.Add(documents)
    root.Add(pictures)

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

root
  Documents
    resume.doc
  Pictures
    photo.jpg