Golang如何使用组合模式实现树状结构

组合模式通过统一接口处理树状结构，Go中用接口和嵌入实现；定义Component接口规范GetName和Print行为，File作为叶子节点直接输出名称，Directory作为容器持有一组Component并递归调用其方法，Add添加子节点，Print时传递层级缩进，最终构建如文件系统的树形结构，客户端可一致操作单个或组合对象。

在Go语言中，组合模式常用于处理树状结构，比如文件系统、组织架构或菜单层级。它通过统一处理单个对象和组合对象，让客户端以一致方式操作整个结构。

定义组件接口

先定义一个通用接口，表示树中的节点行为，比如计算总值、打印名称或遍历。

例如：

type Component interface {
    GetName() string
    Print(depth int)
}

这个接口规定所有节点都必须能获取名字和打印自己，depth参数控制缩进，体现层级。

立即学习“go语言免费学习笔记（深入）”；

实现叶子节点

叶子是树的末端，不能再展开。比如文件系统中的文件。

实现方式：

type File struct {
    name string
}
func (f *File) GetName() string {
return f.name
}
func (f *File) Print(depth int) {
indent := strings.Repeat("  ", depth)
fmt.Printf("%s- %s\n", indent, f.name)
}

它只代表自身，Print时输出带缩进的名字。

Fotor AI Face Generator

Fotor 平台的在线 AI 头像生成器

下载

实现容器节点（组合）

容器节点可以包含多个子节点，比如目录。

关键点是持有Component切片，并转发调用到子元素：

type Directory struct {
    name     string
    children []Component
}
func (d *Directory) Add(child Component) {
d.children = append(d.children, child)
}
func (d *Directory) GetName() string {
return d.name
}
func (d *Directory) Print(depth int) {
indent := strings.Repeat("  ", depth)
fmt.Printf("%s+ %s\n", indent, d.name)
for _, child := range d.children {
child.Print(depth + 1)
}
}

Add方法动态添加子节点，Print递归调用子节点的Print，形成树形输出。

使用示例

构建一个简单文件系统树：

root := &Directory{name: "root"}
docs := &Directory{name: "docs"}
pic := &Directory{name: "pics"}
file1 := &File{name: "readme.txt"}
file2 := &File{name: "photo.jpg"}
docs.Add(file1)
pic.Add(file2)
root.Add(docs)
root.Add(pic)
root.Print(0)

输出：

+ root
  + docs
    - readme.txt
  + pics
    - photo.jpg

结构清晰，添加新类型如压缩包或快捷方式也很容易，只需实现Component接口。

基本上就这些。组合模式在Go里靠接口和递归就能简洁实现，不需要复杂继承体系。只要把握“统一接口、递归处理”原则，树状结构很容易管理。