如何用Golang实现状态模式控制复杂状态机_Golang 状态模式综合示例-Golang-PHP中文网

如何用Golang实现状态模式控制复杂状态机_Golang 状态模式综合示例

P粉602998670

发布： 2025-11-15 20:23:02

原创

752人浏览过

状态模式通过接口和组合在Golang中实现，适用于多状态、行为随状态变化的场景。以网络连接为例，定义ConnectionState接口，包含Connect、Disconnect、Send方法，由具体状态实现。上下文Connection持有当前状态，将操作委托给状态对象，并可通过SetState切换状态。三种状态——Disconnected、Connecting、Connected分别封装各自行为：初始为Disconnected，调用Connect进入Connecting，再转至Connected后可发送数据；任意状态均可Disconnect回到Disconnected。相比条件判断，状态模式结构清晰、扩展性强，新增状态无需修改现有逻辑，适合复杂状态转换与事件驱动系统。

如何用golang实现状态模式控制复杂状态机_golang 状态模式综合示例

状态模式适合处理对象内部复杂的状态转换逻辑，尤其在状态多、行为随状态变化而变化的场景下，能有效避免大量条件判断。Golang 虽无类与继承，但通过接口和组合完全可以实现状态模式。

定义状态接口与上下文

首先明确状态机的行为。假设我们要控制一个网络连接对象，它有“已断开”、“连接中”、“已连接”三种状态，不同状态下对 Connect、Disconnect、Send 的响应不同。

我们定义一个状态接口，所有具体状态需实现它：

type ConnectionState interface {
    Connect(conn *Connection)
    Disconnect(conn *Connection)
    Send(conn *Connection, data string)
}

登录后复制

上下文（即连接对象）持有当前状态，并将状态相关操作委托给当前状态对象：

立即学习“go语言免费学习笔记（深入）”；

type Connection struct {
    state ConnectionState
}

func (c *Connection) SetState(state ConnectionState) {
    c.state = state
}

func (c *Connection) Connect()    { c.state.Connect(c) }
func (c *Connection) Disconnect() { c.state.Disconnect(c) }
func (c *Connection) Send(data string) { c.state.Send(c, data) }

func (c *Connection) GetCurrentState() string {
    switch c.state.(type) {
    case *DisconnectedState:
        return "Disconnected"
    case *ConnectingState:
        return "Connecting"
    case *ConnectedState:
        return "Connected"
    default:
        return "Unknown"
    }
}

登录后复制

实现具体状态

每个状态实现接口方法，根据当前状态决定行为，并在适当时机改变上下文的状态。

已断开状态

type DisconnectedState struct{}

func (s *DisconnectedState) Connect(conn *Connection) {
    println("开始连接...")
    conn.SetState(&ConnectingState{})
}

func (s *DisconnectedState) Disconnect(conn *Connection) {
    println("已经断开，无需再断开")
}

func (s *DisconnectedState) Send(conn *Connection, data string) {
    println("无法发送：尚未连接")
}

登录后复制

连接中状态

百灵大模型

蚂蚁集团自研的多模态AI大模型系列

177

查看详情

type ConnectingState struct{}

func (s *ConnectingState) Connect(conn *Connection) {
    println("正在连接中，请勿重复连接")
}

func (s *ConnectingState) Disconnect(conn *Connection) {
    println("取消连接")
    conn.SetState(&DisconnectedState{})
}

func (s *ConnectingState) Send(conn *Connection, data string) {
    println("连接未完成，无法发送数据")
}

登录后复制

已连接状态

type ConnectedState struct{}

func (s *ConnectedState) Connect(conn *Connection) {
    println("已连接，无需再次连接")
}

func (s *ConnectedState) Disconnect(conn *Connection) {
    println("断开连接")
    conn.SetState(&DisconnectedState{})
}

func (s *ConnectedState) Send(conn *Connection, data string) {
    println("发送数据:", data)
}

登录后复制

初始化与使用状态机

创建连接时，初始状态为“已断开”：

func NewConnection() *Connection {
    return &Connection{state: &DisconnectedState{}}
}

登录后复制

实际调用示例：

func main() {
    conn := NewConnection()

    conn.Send("hello")        // 无法发送：尚未连接
    conn.Connect()            // 开始连接...
    conn.Send("hello")        // 连接未完成，无法发送数据
    conn.Disconnect()         // 取消连接
    conn.Connect()            // 开始连接...
    // 模拟进入已连接状态（实际可能由异步事件触发）
    conn.SetState(&ConnectedState{})
    conn.Send("world")        // 发送数据: world
    conn.Disconnect()         // 断开连接
}

登录后复制