设计模式：组合模式解析与Go语言实现

运维开发王义杰

发布于 2023-11-20 15:21:11

1580

发布于 2023-11-20 15:21:11

文章被收录于专栏：运维开发王义杰运维开发王义杰

1. 引言

组合模式（Composite Pattern）是一种结构型设计模式，用于以树形结构来组织部分-整体的层次结构。这种模式创建了一个包含自己对象组的类，并允许客户端统一对待个别对象和组合对象。

2. 组合模式的结构

组合模式通常包括以下几个组成部分：

组件（Component）：为组合中的对象声明接口，在适当情况下，实现所有类共有接口的默认行为。
叶子（Leaf）：在组合中表示叶节点对象，叶子节点没有子节点。
复合组件（Composite）：定义有子部件的那些部件的行为，存储子部件，实现与子部件有关的操作。

3. Go语言实现示例

以下是使用Go语言实现组合模式的示例：

package main

import (
  "fmt"
  "strings"
)

// 组件接口
type Component interface {
  Add(Component)
  Remove(Component)
  Display(depth int)
}

// 叶子构件
type Leaf struct {
  name string
}

func NewLeaf(name string) *Leaf {
  return &Leaf{name: name}
}

func (l *Leaf) Add(c Component) {
  fmt.Println("Cannot add to a leaf")
}

func (l *Leaf) Remove(c Component) {
  fmt.Println("Cannot remove from a leaf")
}

func (l *Leaf) Display(depth int) {
  fmt.Println(strings.Repeat("-", depth), l.name)
}

// 复合构件
type Composite struct {
  name       string
  children []Component
}

func NewComposite(name string) *Composite {
  return &Composite{name: name}
}

func (c *Composite) Add(child Component) {
  c.children = append(c.children, child)
}

func (c *Composite) Remove(child Component) {
  // 移除操作
}

func (c *Composite) Display(depth int) {
  fmt.Println(strings.Repeat("-", depth), c.name)
  for _, component := range c.children {
    component.Display(depth + 2)
  }
}

func main() {
  root := NewComposite("root")
  root.Add(NewLeaf("Leaf A"))
  root.Add(NewLeaf("Leaf B"))

  comp := NewComposite("Composite X")
  comp.Add(NewLeaf("Leaf XA"))
  comp.Add(NewLeaf("Leaf XB"))

  root.Add(comp)
  root.Display(1)
}