学习Go语言之装饰器模式

一，首先理解装饰器模式：动态的给一个对象增加一些额外的职责，这是在软件设计原则上面，一个功能装饰另一个功能，每个功能遵循同一个接口是这个模式的特征。

二，定义对象接口和装饰抽象类

type IDecorate interface {
    Do()
}

// 装饰器。实现接口，又定义了自己的事件DecorateFun，相当于抽象类
type Decorate struct {
    // 待装饰的抽象类
    decorate IDecorate
}

func (c *Decorate) DecorateFun(i IDecorate) {
    c.decorate = i
}

func (c *Decorate) Do() {
    if c.decorate != nil {
        c.decorate.Do()
    }
}

三，具体的装饰类

// 具体A装饰
type DecorateA struct {
    Base Decorate
}

// 重写方法，隐式实现接口
func (c *DecorateA) Do() {
    fmt.Printf("执行A装饰")
    c.Base.Do()
}

// 具体B装饰
type DecorateB struct {
    Base Decorate
}

// 重写方法，隐式实现接口
func (c *DecorateB) Do() {
    fmt.Printf("执行B装饰")
    c.Base.Do()
}

// 具体C装饰
type DecorateC struct {
    Base Decorate
}

// 重写方法，隐式实现接口
func (c *DecorateC) Do() {
    fmt.Printf("执行C装饰")
    c.Base.Do()
}

四，客户端调用。此处需要注意避免贪吃蛇无限循环问题

func main() {
    a := new(DecorateA)
    b := new(DecorateB)
    c := new(DecorateC)
    b.Base.DecorateFun(a)
    c.Base.DecorateFun(b)
    c.Do()
}