Golang状态模式如何替代大量if else_状态管理思路解析_技术教程

Go中状态模式减少if else的关键是用map[State]func(*Context) error查表执行，将状态判断下沉到注册表，避免业务逻辑中重复条件分支。

状态模式在 Go 中为什么能减少 if else

因为 Go 没有继承和虚函数，直接照搬传统 OOP 的状态模式（比如 Java 里让每个状态实现 State 接口）容易写成一堆空接口+类型断言，反而更难维护。真正有效的替代方式是：用 func 类型封装状态行为，配合结构体字段存储当前状态标识，把“判断逻辑”从调用方下沉到状态注册表里。

用 map[string]func() 替代 if else 的实际写法

核心不是抽象出多少接口，而是让状态流转变成查表+执行。每个状态对应一个闭包或普通函数，避免在业务逻辑里反复写 if s.state == "pending" { ... } else if s.state == "done" { ... }。

状态名用 string 或自定义 enum 类型（如 type State string），便于日志、调试和序列化
行为函数签名统一为 func(*Context) error，保证可组合性和错误传播
注册表用 map[State]func(*Context) error，初始化时一次性注入，运行时只做一次查表
禁止在状态函数内部再写大段条件分支——那只是把 if else 搬到了另一个地方

type State string
const (
	Pending State = "pending"
	Running State = "running"
	Done    State = "done"
)

type Context struct {
	state State
	data  map[string]interface{}
}

type StateHandler func(*Context) error

var stateHandlers = map[State]StateHandler{
	Pending: func(c *Context) error {
		c.data["started_at"] = time.Now()
		c.state = Running
		return nil
	},
	Running: func(c *Context) error {
		if len(c.data["result"].(string)) > 0 {
			c.state = Done
		}
		return nil
	},
	Done: func(c *Context) error {
		return errors.New("already done")
	},
}

func (c *Context) Handle() error {
	handler, ok := stateHandlers[c.state]
	if !ok {
		return fmt.Errorf("no handler for state %q", c.state)
	}
	return handler(c)
}