装饰器

Hexiaopi大约 3 分钟

什么是装饰器模式

装饰器模式(Decorator Pattern)：它可以在不改变原有对象的结构和功能的情况下，动态地给对象添加新的功能或者增强原有功能。装饰器设计模式的核心思想是使用一个装饰器对象来包装原有对象，从而实现对原有对象的功能扩展或修改。

包含哪些角色

Component: 组件接口
所有被装饰组件及装饰器对应的接口标准
ConcreteComponent: 具体组件
需要被装饰的组件，实现组件接口标准
Decorator: 装饰器
持有一个组件（Component）对象的实例，并定义一个与组件接口一致的接口。
ConcreteDecorator: 装饰器实现
继承自装饰器抽象类的具体子类装饰器，可以有多种实现，在被装饰组件对象的基础上为其添加新的特性。

代码示例

package decorator

import "fmt"

type Component interface {
	execute()
}

type ConcreteComponent struct{}

func (c ConcreteComponent) execute() {
	fmt.Println("concrete component execute")
}

type Decorator struct {
	component Component
}

func (d *Decorator) SetComponent(c Component) {
	d.component = c
}

func (d Decorator) execute() {
	d.component.execute()
}

type ConcreteDecortor1 struct {
	Decorator
}

func (c ConcreteDecortor1) addedFunction() {
	fmt.Println("concrete decortor1 add function")
}

func (c ConcreteDecortor1) execute() {
	c.addedFunction()
	c.Decorator.execute()
}

type ConcreteDecortor2 struct {
	Decorator
}

func (c ConcreteDecortor2) addedFunction() {
	fmt.Println("concrete decortor2 add function")
}

func (c ConcreteDecortor2) execute() {
	c.addedFunction()
	c.Decorator.execute()
}

使用示例

func ExampleComponent() {
	c := ConcreteComponent{}
	c.execute()
	//c1装饰c
	var c1 ConcreteDecortor1
	c1.SetComponent(c)
	c1.execute()
	//c2装饰c
	var c2 ConcreteDecortor2
	c2.SetComponent(c)
	c2.execute()
	//c2装饰c1
	var c21 ConcreteDecortor2
	c21.SetComponent(c1)
	c21.execute()
	// Output:
	// concrete component execute
	// concrete decortor1 add function
	// concrete component execute
	// concrete decortor2 add function
	// concrete component execute
	// concrete decortor2 add function
	// concrete decortor1 add function
	// concrete component execute
}