组合模式通过统一接口处理单个对象和对象集合,适用于树形结构如文件系统。定义Component接口包含Add、Remove、GetChildren、GetName和Print方法,实现叶子节点Leaf和容器节点Composite,两者均实现该接口。Leaf的Add、Remove等操作为空,Print输出自身;Composite维护子组件切片,Add添加子节点,Remove删除指定子节点,GetChildren返回子节点列表,Print递归打印子节点并缩进表示层级。构建时可创建根节点root,添加dept1(含开发组、测试组)和dept2作为子节点,调用root.Print("")输出层次结构,容器用"+"标识,叶子用"-"标识。Go接口机制使组合模式简洁,无需继承,扩展性强,可灵活添加查找或遍历功能,关键在于保持接口一致,使客户端无需区分叶与复合对象。

在Go语言中实现组合模式的树形结构,关键在于统一处理单个对象和对象集合。这种模式特别适合表示具有层级关系的数据,比如文件系统、组织架构或菜单树。
定义统一的组件接口
组合模式的核心是让叶子节点和容器节点对外暴露相同的接口。先定义一个Component接口,声明共用的方法:
type Component interface {
Add(child Component)
Remove(child Component)
GetChildren() []Component
GetName() string
Print(indent string)
}
这个接口涵盖了树形结构的基本操作:增删子节点、获取子节点、名称访问和打印展示。
实现叶子节点和容器节点
叶子节点不包含子节点,而容器节点可以持有多个子组件。两者分别实现同一接口:
立即学习“go语言免费学习笔记(深入)”;
// 叶子节点
type Leaf struct {
name string
}
func (l Leaf) Add(child Component) {}
func (l Leaf) Remove(child Component) {}
func (l Leaf) GetChildren() []Component { return nil }
func (l Leaf) GetName() string { return l.name }
func (l *Leaf) Print(indent string) {
fmt.Println(indent + "- " + l.GetName())
}
// 容器节点
type Composite struct {
name string
children []Component
}
func (c *Composite) Add(child Component) {
c.children = append(c.children, child)
}
func (c *Composite) Remove(child Component) {
for i, ch := range c.children {
if ch == child {
c.children = append(c.children[:i], c.children[i+1:]...)
break
}
}
}
func (c *Composite) GetChildren() []Component {
return c.children
}
func (c *Composite) GetName() string {
return c.name
}
func (c *Composite) Print(indent string) {
fmt.Println(indent + "+ " + c.GetName())
for _, child := range c.children {
child.Print(indent + " ")
}
}
注意Print方法的递归调用,它让整个结构能按层级输出,体现树形特征。
构建和使用树形结构
通过组合不同类型的节点,可构造出任意深度的树。例如模拟一个部门结构:
root := &Composite{name: "公司"}
dept1 := &Composite{name: "技术部"}
dept2 := &Composite{name: "销售部"}
dev := &Leaf{name: "开发组"}
qa := &Leaf{name: "测试组"}
dept1.Add(dev)
dept1.Add(qa)
root.Add(dept1)
root.Add(dept2)
root.Print("")
输出会清晰展示层级关系,容器节点用"+"标记,叶子用"-"标记,缩进反映深度。
基本上就这些。Go的接口机制让组合模式实现简洁自然,不需要复杂的继承体系。只要把握好接口一致性,就能灵活管理各种树形数据。实际项目中可根据需要扩展属性或方法,比如加入路径查找、遍历钩子等。关键是保持接口统一,让调用方无需关心当前操作的是单个元素还是复合结构。










