21.Golang设计模式之状态模式

状态模式

GitHub代码链接
状态模式(State Pattern)中，类的行为是基于它的状态改变的。

什么是状态模式

状态模式允许对象在内部状态改变时，改变它的行为。这样对象看起来好像是
修改了它的类

解决了什么问题

状态模式解决了对象的行为依赖它的状态(属性)的问题。如果代码中包含大量与对象
状态有关的条件语句时，可以使用状态模式。

优点

封装了转换规则
将所有与某个状态有关的行为放到一个类中，并且可方便的新增状态
允许状态转换逻辑和状态对象合成一体，而不是巨大的条件语句块

缺点

状态模式的使用，必然会增加系统类和对象的个数
状态模式的结构与实现都比较复杂，如果使用不当会导致系统混乱

代码实现

1.实现一个状态保存类

状态保存类用于保存当前状态。

//Context 状态保存 类
type Context struct {
    state State
}

//NewContext 实例化状态保存类
func NewContext() *Context {
    return &Context{
        state: nil,
    }
}

//SetState 设置状态保存类当前的状态
func (c *Context) SetState(s State) {
    c.state = s
}

//GetState 获取状态保存类当前的状态
func (c *Context) GetState() State {
    return c.state
}

2. 定义状态类接口

//State 状态接口
type State interface {
    DoAction(context *Context)
    ToString() string
}

3. 实现状态类

//StartState 开始状态类
type StartState struct{}

//NewStatartState 实例化开始状态类
func NewStatartState() *StartState {
    return &StartState{}
}

//DoAction 开始状态类的DoAction，实现State接口
func (start *StartState) DoAction(context *Context) {
    fmt.Println("Now is start state")
    context.state = start
}

//ToString 返回开始状态类名称
func (start *StartState) ToString() string {
    return "Start state"
}

4. 测试

context := NewContext()

startState := NewStatartState()
startState.DoAction(context)
fmt.Println(context.GetState().ToString())

stopState := NewStopState()
stopState.DoAction(context)
fmt.Println(context.GetState().ToString())