组合模式通过统一接口处理部分与整体,Go中定义Graphic接口,Circle和Square为叶子节点实现Draw方法,Group作为组合节点包含多个子元素并递归调用其Draw,客户端可一致操作单个或组合对象,符合开闭原则。
在Go语言中,组合模式常用于将多个对象组合成树形结构来表示“部分-整体”的层次关系,并让客户端可以统一地处理单个对象和组合对象。这种模式特别适用于需要递归处理嵌套结构的场景,比如文件系统、菜单结构或组织架构。
定义组件接口
要实现组合模式,第一步是定义一个统一的操作接口,让叶子节点(终端对象)和容器节点(组合对象)都实现这个接口。
例如:
假设我们要构建一个图形绘制系统,其中包含单独的图形(如圆形)和图形组(多个图形的集合),我们希望对它们调用 Draw() 方法时行为一致。
type Graphic interface { Draw() }
实现叶子节点
叶子节点是不能再被分解的终端对象。它直接实现接口方法。
立即学习“go语言免费学习笔记(深入)”;
type Circle struct{} func (c *Circle) Draw() { println("Drawing a circle") } type Square struct{} func (s *Square) Draw() { println("Drawing a square") }
实现组合节点
组合节点包含多个子元素(可以是叶子或其他组合),它通过遍历子元素并调用其对应方法,实现统一操作。
type Group struct { children []Graphic } func (g *Group) Add(child Graphic) { g.children = append(g.children, child) } func (g *Group) Draw() { println("Starting group draw:") for _, child := range g.children { child.Draw() } println("End of group draw") }
统一调用示例
客户端代码无需区分是单个图形还是图形组,都可以当作 Graphic 类型处理。
func main() { circle := &Circle{} square := &Square{} group1 := &Group{} group1.Add(circle) group1.Add(&Square{}) // 可以内联添加 group2 := &Group{} group2.Add(group1) group2.Add(square) // 统一调用 Draw() group2.Draw() }
输出结果:
Starting group draw: Starting group draw: Drawing a circle Drawing a square End of group draw Drawing a square End of group draw
可以看到,无论是单个图形还是嵌套的组合,调用 Draw() 都能正确递归执行,结构清晰且扩展性强。
这种设计让新增图形类型或组合方式变得容易,符合开闭原则。只要实现 Graphic 接口,就能无缝接入现有逻辑。
基本上就这些。