组合模式通过接口和结构体嵌入实现对象树,统一处理个体与容器。定义Component接口包含Draw、Add、Remove和GetChildren方法;叶子节点如Circle实现接口但不管理子元素;容器节点Group持有一组Component并递归调用其方法;利用嵌入可复用逻辑,通过接口参数提升灵活性,支持动态修改结构,适用于文件系统、ui组件等场景。

在golang中实现组合模式,核心是通过结构体嵌入(匿名字段)和接口来构建可扩展的对象树结构。这种设计让客户端可以统一处理单个对象和组合对象,特别适合需要递归操作的场景,比如文件系统、UI组件、菜单结构等。
定义统一接口
要实现组合模式,第一步是定义一个公共接口,描述个体与容器共有的行为。例如,在处理图形对象时:
type Component interface { Draw() Add(Component) Remove(Component) GetChildren() []Component }
这个接口适用于所有图形元素,无论是单个形状还是包含多个子元素的图层。
实现叶子节点
叶子节点不包含子元素,因此它的 Add、Remove 和 GetChildren 方法可以为空或返回默认值:
立即学习“go语言免费学习笔记(深入)”;
type Circle struct{} func (c *Circle) Draw() { println("Drawing a circle") } func (c *Circle) Add(comp Component) {} func (c *Circle) Remove(comp Component) {} func (c *Circle) GetChildren() []Component { return nil }
这样,叶子节点只关注自身行为,不影响整体结构。
实现容器节点
容器节点持有子组件列表,并将操作委托给它们。利用Go的结构体嵌入机制,可以复用部分逻辑:
type Group struct { children []Component } func (g *Group) Draw() { for _, child := range g.children { child.Draw() } } func (g *Group) Add(comp Component) { g.children = append(g.children, comp) } func (g *Group) Remove(comp Component) { for i, c := range g.children { if c == comp { g.children = append(g.children[:i], g.children[i+1:]...) break } } } func (g *Group) GetChildren() []Component { return g.children } </font>
通过嵌套调用 Draw,整个结构能自动递归渲染。
便于扩展的关键技巧
- 使用接口而非具体类型:函数参数和字段尽量使用 Component 接口,提升灵活性。
- 嵌入简化代码:可定义一个基础结构体实现通用方法,供多个容器复用。
- 运行时安全判断:用类型断言或反射检查是否支持添加操作,避免误操作叶子节点。
- 支持动态修改结构:允许在运行时增删节点,适应配置化或可视化编辑需求。


