组合模式是结构型设计模式的一种，它的核心思想是让单个对象和组合对象拥有相同的操作接口，这样调用方在使用的时候不需要关心当前处理的是单个独立元素还是包含多个子元素的组合元素，统一调用接口即可完成操作。在Go语言开发中，我们经常会遇到需要构建树形结构的场景，比如电商系统的商品分类、企业的部门组织架构、后台管理的多级菜单等，这些场景的层级关系天然适合用组合模式来实现。
组合模式的核心角色与Go语言实现思路
组合模式通常包含三个核心角色，第一个是组件接口，也就是所有节点（包括叶子节点和组合节点）都需要实现的统一接口，这个接口里定义了节点共有的方法，比如获取节点名称、添加子节点、删除子节点、遍历节点等。第二个是叶子节点，也就是树形结构中最底层的节点，它没有子节点，所以添加、删除子节点的方法对它来说要么直接返回错误，要么空实现。第三个是组合节点，也就是包含子节点的父节点，它除了实现组件接口的所有方法之外，还需要维护一个子节点列表，用来存储下属的叶子节点或者其他组合节点。
在Go语言中实现组合模式，我们不需要像Java那样使用抽象类，直接定义接口即可。因为Go的接口是隐式实现的，只要结构体实现了接口定义的所有方法，就自动属于该接口类型，这会让我们的代码更灵活。需要注意的是，组件接口里的方法要根据节点的类型做不同的实现，比如叶子节点的AddChild方法可以直接返回错误，提示不能给叶子节点添加子节点，而组合节点的AddChild方法则要把传入的子节点追加到自己的子节点列表中。
另外，Go语言的空接口interface{}虽然可以接收任意类型，但在组合模式的场景下不建议使用，因为我们会需要调用节点的统一方法，使用明确的组件接口可以让编译器帮我们做类型检查，避免运行时出现类型错误。如果后续需要给节点扩展额外的属性，比如节点的ID、节点的额外元数据，只需要在结构体里添加对应的字段，然后实现接口方法的时候把这些属性用上即可，不会影响整体的结构。
完整代码示例：实现多级商品分类树
下面我们通过一个具体的例子来实现组合模式构建树形结构，场景是电商平台的商品分类，顶级分类是电子产品，下面有手机、电脑两个子分类，手机分类下又有苹果、华为两个叶子分类，电脑分类下有联想、戴尔两个叶子分类。首先我们定义组件接口Category，里面包含获取分类名称、添加子分类、删除子分类、打印分类树的方法。
// 定义分类组件接口，所有分类节点都需要实现这个接口
type Category interface {
GetName() string
AddChild(child Category) error
RemoveChild(child Category) error
PrintTree(prefix string)
}
接下来实现叶子节点结构体LeafCategory，它只有名称属性，添加和删除子分类的方法直接返回错误，打印的时候只输出自己的名称。
// 叶子分类节点，没有子分类
type LeafCategory struct {
Name string
}
func (l *LeafCategory) GetName() string {
return l.Name
}
func (l *LeafCategory) AddChild(child Category) error {
return fmt.Errorf("叶子分类 %s 不能添加子分类", l.Name)
}
func (l *LeafCategory) RemoveChild(child Category) error {
return fmt.Errorf("叶子分类 %s 没有子分类可以删除", l.Name)
}
func (l *LeafCategory) PrintTree(prefix string) {
fmt.Println(prefix + l.Name)
}
然后实现组合节点结构体CompositeCategory，它除了名称属性之外，还有一个子分类列表，添加子分类的时候把子节点追加到列表，删除的时候遍历列表找到对应的节点移除，打印的时候先输出自己的名称，再递归调用所有子节点的打印方法，加上缩进前缀。
// 组合分类节点，包含子分类
type CompositeCategory struct {
Name string
Children []Category
}
func (c *CompositeCategory) GetName() string {
return c.Name
}
func (c *CompositeCategory) AddChild(child Category) error {
c.Children = append(c.Children, child)
return nil
}
func (c *CompositeCategory) RemoveChild(child Category) error {
for i, ch := range c.Children {
if ch.GetName() == child.GetName() {
c.Children = append(c.Children[:i], c.Children[i+1:]...)
return nil
}
}
return fmt.Errorf("未找到名称为 %s 的子分类", child.GetName())
}
func (c *CompositeCategory) PrintTree(prefix string) {
fmt.Println(prefix + c.Name)
// 递归打印所有子分类，前缀增加两个空格作为缩进
newPrefix := prefix + " "
for _, child := range c.Children {
child.PrintTree(newPrefix)
}
}
最后我们编写主函数，构建完整的商品分类树，然后调用打印方法输出整棵树的结构，同时测试添加和删除子分类的功能。
func main() {
// 创建顶级分类：电子产品
electronics := &CompositeCategory{Name: "电子产品"}
// 创建二级分类：手机、电脑
phone := &CompositeCategory{Name: "手机"}
computer := &CompositeCategory{Name: "电脑"}
// 创建叶子分类
apple := &LeafCategory{Name: "苹果"}
huawei := &LeafCategory{Name: "华为"}
lenovo := &LeafCategory{Name: "联想"}
dell := &LeafCategory{Name: "戴尔"}
// 构建树形结构
phone.AddChild(apple)
phone.AddChild(huawei)
computer.AddChild(lenovo)
computer.AddChild(dell)
electronics.AddChild(phone)
electronics.AddChild(computer)
// 打印整棵树
fmt.Println("商品分类树：")
electronics.PrintTree("")
// 测试给叶子分类添加子分类，会返回错误
err := apple.AddChild(&LeafCategory{Name: "iPhone 15"})
if err != nil {
fmt.Printf("添加子分类失败：%vn", err)
}
// 测试删除子分类
err = phone.RemoveChild(huawei)
if err != nil {
fmt.Printf("删除子分类失败：%vn", err)
} else {
fmt.Println("n删除华为分类后的手机子分类：")
phone.PrintTree("")
}
}
组合模式的优势与适用场景分析
使用组合模式构建树形结构最大的优势是调用方不需要关心节点的具体类型，不管是叶子节点还是组合节点，都可以用统一的方法处理。比如上面的例子里，我们打印整棵树的时候，只需要调用顶级节点的PrintTree方法，不需要判断每个节点是叶子还是组合，递归逻辑已经在节点的方法里实现了，这大幅减少了调用方的代码复杂度，也避免了大量的类型断言和判断逻辑。如果后续我们需要给所有节点增加一个获取分类层级的方法，只需要在组件接口里添加这个方法，然后分别给叶子节点和组合节点实现即可，不会影响已有的调用代码，扩展性非常好。
组合模式非常适合处理具有层级关系的数据结构，除了前面提到的商品分类、组织架构、多级菜单之外，还可以用在文件系统的实现上，比如把文件和文件夹都实现统一的节点接口，文件是叶子节点，文件夹是组合节点，这样遍历文件夹、统计文件大小等操作都可以统一处理。不过组合模式也有不适合的场景，如果树形结构中的叶子节点和组合节点的差异非常大，很多接口方法对叶子节点来说都没有意义，这时候强行使用组合模式会导致叶子节点需要实现很多空方法或者返回错误的方法，反而让代码变得冗余。这种情况下可以考虑把接口拆分成更细的粒度，或者只对差异小、操作统一的场景使用组合模式。
在实际的Go项目中使用组合模式的时候，还需要注意循环引用的问题，比如给一个组合节点添加它自己的父节点作为子节点，会导致递归遍历的时候进入死循环，所以在AddChild方法里可以增加一个检查逻辑，判断要添加的子节点是不是当前节点本身，或者是不是当前节点的祖先节点，避免循环引用的出现。另外，如果树形结构需要频繁修改，比如经常添加删除节点，组合节点的子节点列表可以使用sync.RWMutex做并发保护，避免多个 goroutine 同时操作子节点列表导致数据竞争的问题。