Golang如何使用状态模式动态切换状态_Golang State模式实现实践_技术教程

Go语言通过接口与结构体组合实现状态模式，以消除条件判断并提升可维护性。示例中订单系统定义OrderState接口及四种状态结构体，如PendingPayment、Paid等，各自实现Pay、Ship等方法，不同状态下行为各异。Order上下文持有当前状态实例，并将操作委托给状态对象处理，如支付后自动切换到已支付状态。该模式符合开闭原则，便于扩展新状态，避免冗长if-else逻辑。关键在于合理设计接口粒度与状态转换边界，确保流转合法性。

在Go语言中实现状态模式，关键在于将对象的行为随内部状态改变而改变的设计思路通过接口与结构体组合来表达。它避免了大量条件判断语句（如 if/else 或 switch），使代码更清晰、可维护性更强。下面通过一个实际例子说明如何用 Golang 实现状态模式进行动态状态切换。

状态模式核心思想

状态模式允许一个对象在其内部状态变化时改变其行为，看起来像是改变了类。在 Go 中没有类的概念，但我们可以通过接口定义行为，用不同的结构体实现该接口来表示不同状态，再由上下文对象持有当前状态实例并委托调用。

场景示例：订单状态流转

假设我们有一个订单系统，订单有“待支付”、“已支付”、“已发货”、“已完成”四种状态。每个状态下用户可执行的操作不同，比如只有“待支付”状态才能取消订单，“已支付”后才能发货。

定义状态接口：

首先定义一个状态接口，包含所有可能被触发的方法：

type OrderState interface {
    Pay(order *Order)
    Ship(order *Order)
    Complete(order *Order)
    Cancel(order *Order)
}

实现各个具体状态：

每种状态实现自己的逻辑。例如：

type PendingPayment struct{}

func (s *PendingPayment) Pay(order *Order) {
    fmt.Println("订单正在支付...")
    order.setState(&Paid{})
}

func (s *PendingPayment) Ship(order *Order) {
    fmt.Println("无法发货：订单尚未支付")
}

func (s *PendingPayment) Complete(order *Order) {
    fmt.Println("无法完成：请先支付")
}

func (s *PendingPayment) Cancel(order *Order) {
    fmt.Println("订单已取消")
    order.setState(&Canceled{})
}

type Paid struct{}

func (s *Paid) Pay(order *Order) {
    fmt.Println("订单已支付，无需重复操作")
}

func (s *Paid) Ship(order *Order) {
    fmt.Println("正在发货...")
    order.setState(&Shipped{})
}

func (s *Paid) Complete(order *Order) {
    fmt.Println("不能直接完成，请先发货")
}

func (s *Paid) Cancel(order *Order) {
    fmt.Println("订单已支付，不可取消")
}

类似地可以实现 Shipped、Completed、Canceled 等状态。

定义上下文对象（订单）：

Order 结构体持有当前状态，并将操作委托给当前状态处理：

type Order struct {
    state OrderState
}

func NewOrder() *Order {
    return &Order{state: &PendingPayment{}}
}

func (o *Order) setState(state OrderState) {
    o.state = state
}

// 委托调用
func (o *Order) Pay() { o.state.Pay(o) }
func (o *Order) Ship() { o.state.Ship(o) }
func (o *Order) Complete() { o.state.Complete(o) }
func (o *Order) Cancel() { o.state.Cancel(o) }

使用示例：

func main() {
    order := NewOrder()

    order.Pay()     // 输出：订单正在支付...
    order.Ship()    // 输出：正在发货...
    order.Complete() // 输出：无法完成：请先发货

    // 注意：此时状态已是 Shipped，应更新对应逻辑
}