如何使用Golang实现状态模式_对象行为随状态变化而改变

状态模式通过接口+结构体组合实现状态可替换与行为解耦,订单等上下文委托操作给当前状态对象,状态自行决定合法行为与切换,共用逻辑可嵌入baseState,测试需覆盖非法操作拦截与正确流转。

状态模式的核心是把对象的行为委托给当前状态对象,让状态变化自然带动行为变化。Golang 没有类继承,但通过接口 + 结构体组合 + 方法赋值,能干净地实现这一模式,关键在于“状态可替换”和“行为解耦”。

定义状态接口与具体状态类型

先抽象出统一的状态行为接口,每个具体状态实现它:

例如一个订单(Order)可能处于“新建”、“已支付”、“已发货”、“已完成”等状态,不同状态下允许的操作不同:

  • State 接口定义通用方法,如 Pay()Ship()Complete()GetStatus()
  • 每个具体状态(CreatedStatePaidStateShippedState)实现该接口,内部只做合法操作或返回错误提示
  • 避免在状态实现里直接修改上下文(如 Order),而是通过回调或传参方式通知上下文切换状态

构建上下文并支持状态切换

上下文(如 Order)持有当前状态,并提供委托入口:

  • 用指针字段 state State 存储当前状态,初始设为 &CreatedState{}
  • 所有外部操作(如 order.Pay())不写业务逻辑,只调用 o.state.Pay(o) —— 把自身传给状态处理
  • 状态实现中,若需切换,直接对上下文的 state 字段赋新状态指针,例如:o.state = &PaidState{}
  • 注意:状态切换必须由状态自己决定,上下文不判断条件,只提供“被调用”的能力

避免状态爆炸与共享逻辑

多个状态有共用行为时,不要重复写,可用嵌入或工具函数:

  • 定义 baseState 结构体,封装日志、校验、事件通知等通用逻辑,各状态结构体匿名嵌入它
  • 将状态转换规则集中管理,比如写个 TransitionRule(from, to State) bool 函数,用于运行时校验或调试
  • 如果状态间差异仅是返回值(如字符串描述),可考虑用 map + 函数闭包代替多个结构体,适合简单场景

测试状态流转是否符合预期

状态模式的价值体现在流程可控,测试要覆盖“非法操作被拒绝”和“合法操作触发正确流转”:

  • 为每个状态单独写单元测试,验证其方法在当前状态下是否执行/报错
  • 写集成测试:新建 Order → 调 Pay → 检查状态变为 Paid → 再调 Ship → 检查变为 Shipped → 尝试重复 Pay 应失败
  • 利用 Go 的 interface{} 和 mock 状态对象,可快速验证上下文是否正确调用了状态方法