在处理菜单配置、公司部门层级、文件目录这类具有层级关系的树形结构数据时，传统方式需要区分叶子节点和容器节点，处理逻辑会相对复杂。Golang的组合模式通过抽象统一的接口，让单个对象和组合对象可以被一致对待，大幅简化树形结构的操作逻辑。
组合模式的核心概念
组合模式主要包含三个角色：
- 抽象组件：定义树形结构中所有节点共有的方法，比如获取名称、添加子节点、遍历节点等。
- 叶子节点：树形结构的末端节点，没有子节点，实现抽象组件的方法，部分操作（比如添加子节点）可以直接返回错误或者空实现。
- 组合节点：包含子节点的容器节点，同样实现抽象组件的方法，同时维护子节点列表，负责子节点的增删和遍历。
Golang实现组合模式处理树形结构
下面以公司部门层级为例，展示Golang中组合模式的实现过程。公司的部门结构就是典型的树形结构，最末端的部门是叶子节点，包含子部门的是组合节点，所有部门都有名称和获取人数的方法。
1. 定义抽象组件接口
首先定义Department接口，作为所有部门节点的抽象组件，包含共有的方法：
package main
import "fmt"
// 抽象组件接口，定义所有部门节点的共有方法
type Department interface {
// 获取部门名称
GetName() string
// 获取部门人数
GetCount() int
// 添加子部门，叶子节点可以不实现这个方法
AddChild(dept Department)
// 遍历部门，打印层级结构
Traverse(level int)
}
2. 实现叶子节点
叶子节点是最末端的部门，没有子部门，添加子部门的方法可以直接空实现：
// 叶子节点，最末端的部门，没有子部门
type LeafDepartment struct {
Name string // 部门名称
Count int // 部门人数
}
// 实现GetName方法
func (l *LeafDepartment) GetName() string {
return l.Name
}
// 实现GetCount方法
func (l *LeafDepartment) GetCount() int {
return l.Count
}
// 叶子节点没有子部门，添加子部门的方法空实现
func (l *LeafDepartment) AddChild(dept Department) {
// 叶子节点不支持添加子部门，也可以返回错误
}
// 实现Traverse方法，打印当前部门信息
func (l *LeafDepartment) Traverse(level int) {
// 根据层级缩进，展示层级关系
for i := 0; i < level; i++ {
fmt.Print(" ")
}
fmt.Printf("部门：%s，人数：%dn", l.Name, l.Count)
}
3. 实现组合节点
组合节点包含子部门列表，需要实现子部门的增删和遍历逻辑：
// 组合节点，包含子部门的部门
type CompositeDepartment struct {
Name string // 部门名称
Children []Department // 子部门列表
}
// 实现GetName方法
func (c *CompositeDepartment) GetName() string {
return c.Name
}
// 实现GetCount方法，统计当前部门及所有子部门的总人数
func (c *CompositeDepartment) GetCount() int {
total := 0
for _, child := range c.Children {
total += child.GetCount()
}
return total
}
// 实现AddChild方法，添加子部门
func (c *CompositeDepartment) AddChild(dept Department) {
c.Children = append(c.Children, dept)
}
// 实现Traverse方法，先打印当前部门信息，再遍历所有子部门
func (c *CompositeDepartment) Traverse(level int) {
// 根据层级缩进，展示层级关系
for i := 0; i < level; i++ {
fmt.Print(" ")
}
fmt.Printf("部门：%s，总人数：%dn", c.Name, c.GetCount())
// 遍历所有子部门，层级加1
for _, child := range c.Children {
child.Traverse(level + 1)
}
}
4. 组装树形结构并测试
接下来组装一个公司的部门树形结构，测试组合模式的效果：
func main() {
// 创建叶子节点部门
leaf1 := &LeafDepartment{Name: "研发一组", Count: 10}
leaf2 := &LeafDepartment{Name: "研发二组", Count: 12}
leaf3 := &LeafDepartment{Name: "测试一组", Count: 8}
leaf4 := &LeafDepartment{Name: "测试二组", Count: 7}
leaf5 := &LeafDepartment{Name: "产品组", Count: 5}
// 创建组合节点部门
rdDept := &CompositeDepartment{Name: "研发部"}
qaDept := &CompositeDepartment{Name: "测试部"}
productDept := &CompositeDepartment{Name: "产品部"}
rootDept := &CompositeDepartment{Name: "总公司"}
// 组装部门层级
rdDept.AddChild(leaf1)
rdDept.AddChild(leaf2)
qaDept.AddChild(leaf3)
qaDept.AddChild(leaf4)
productDept.AddChild(leaf5)
rootDept.AddChild(rdDept)
rootDept.AddChild(qaDept)
rootDept.AddChild(productDept)
// 遍历整个部门树
fmt.Println("公司部门层级结构：")
rootDept.Traverse(0)
// 获取研发部总人数
fmt.Printf("n研发部总人数：%dn", rdDept.GetCount())
}
运行上述代码，会先打印出完整的部门层级结构，然后输出研发部的总人数。可以看到，无论是叶子节点还是组合节点，都可以通过统一的Department接口进行操作，不需要区分节点类型，极大简化了树形结构的处理逻辑。
组合模式的适用场景
在Golang中，组合模式适合以下场景：
- 需要处理树形结构数据，比如菜单、组织架构、文件目录等。
- 希望统一对待单个对象和组合对象，不需要在代码中判断节点类型。
- 树形结构的层级可能动态变化，需要灵活增删节点。
组合模式的优势在于简化客户端代码，让树形结构的操作逻辑更清晰，同时符合开闭原则，后续新增节点类型只需要实现抽象组件接口即可，不需要修改原有逻辑。