组合模式

GitHub代码链接
Composite Pattern 组合模式，又叫部分整体模式。用于把一组相似的对象当做一个单一的对象。

什么是组合模式

组合模式将对象组合成树形结构，以表示“部分-整体”的层次结构。使得用户对单个对象的使用具有一致性。

解决了什么问题

组合模式在树形结构的问题中，模糊了简单元素和复杂元素的概念。客户程序一个像处理简单元素一样来处理复杂元素。从而使得客户程序与复杂的元素的内部结构解耦。

优点

• 高层模块调用简单
• 节点自由增加

缺点

• 使用组合模式时，其叶子和树枝的声明都是实现类。而不是接口，违反了依赖倒置原则。

代码实现

我们使用组合模式实现一个公司的职员结构图。

1.1 实现职员类

//Employee 职员类
type Employee struct {
    Name         string
    Dept         string
    Salary       int
    Subordinates *list.List
}

//NewEmployee 实例化职员类
func NewEmployee(name, dept string, salary int) *Employee {
    sub := list.New()
    return &Employee{
        Name:         name,
        Dept:         dept,
        Salary:       salary,
        Subordinates: sub,
    }
}

//Add 添加职员的下属
func (e *Employee) Add(emp Employee) {
    e.Subordinates.PushBack(emp)
}

//Remove 删除职员的下属
func (e *Employee) Remove(emp Employee) {
    for i := e.Subordinates.Front(); i != nil; i = i.Next() {
        if reflect.DeepEqual(i.Value, emp) {
            e.Subordinates.Remove(i)
        }
    }
}

//GetSubordinates 获取职员下属列表
func (e *Employee) GetSubordinates() *list.List {
    return e.Subordinates
}

//ToString 获取职员的string信息
func (e *Employee) ToString() string {
    return "[ Name: " + e.Name + ", dept: " + e.Dept + ", Salary: " + strconv.Itoa(e.Salary) + " ]"
}

1.2 创建公司结构图

ceo := NewEmployee("lee", "ceo", 9999)

    pm := NewEmployee("wang", "technology", 1000)
    programmer1 := NewEmployee("jolin", "technology", 8000)
    programmer2 := NewEmployee("jay", "technology", 8000)

    minister := NewEmployee("zhou", "accounting", 5000)
    finance1 := NewEmployee("zzz", "accounting", 3000)
    finance2 := NewEmployee("Mary", "accounting", 2900)

    ceo.Add(*pm)
    ceo.Add(*minister)

    pm.Add(*programmer1)
    pm.Add(*programmer2)

    minister.Add(*finance1)
    minister.Add(*finance2)

    //打印所有职员
    fmt.Println(ceo.ToString())
    for i := ceo.Subordinates.Front(); i != nil; i = i.Next() {
        em := i.Value.(Employee)
        fmt.Println(em.ToString())
        for j := i.Value.(Employee).Subordinates.Front(); j != nil; j = j.Next() {
            em := j.Value.(Employee)
            fmt.Println(em.ToString())
        }
    }

有疑问加站长微信联系（非本文作者）