Golang树形结构管理实现方法

type Component interface {
    GetName() string
    Print(indent string)
    Size() int
}

type File struct {
    name string
    size int
}
func (f File) GetName() string { return f.name }
func (f File) Print(indent string) {
fmt.Printf("%s- %s (file, %d bytes)\n", indent, f.name, f.size)
}
func (f *File) Size() int { return f.size }

type Directory struct {
    name     string
    children []Component
}
func (d Directory) GetName() string { return d.name }
func (d Directory) Print(indent string) {
fmt.Printf("%s+ %s (dir)\n", indent, d.name)
for , c := range d.children {
c.Print(indent + "  ")
}
}
func (d *Directory) Size() int {
total := 0
for , c := range d.children {
total += c.Size()
}
return total
}

root := &Directory{name: "root"}
src := &Directory{name: "src"}
mainFile := &File{name: "main.go", size: 1240}
goMod := &File{name: "go.mod", size: 86}
src.children = append(src.children, mainFile, goMod)
root.children = append(root.children, src, &File{name: "README.md", size: 520})
root.Print("") // 统一调用，自动展开整棵树
fmt.Println("Total size:", root.Size()) // 自动累加所有叶子大小