前言
在设计电商系统订单模块时，订单会涉及各种状态以及状态与状态之间的流转，
可扩展性、可维护性是我们需要关注的重点！本文分享一下我的技术方案。
如上图，使用 golang 实现上图的订单流转，同时当后续增加订单状态或订单事件时，可以进行快速完成。
目的
关于订单状态的处理，使用统一入口，提高程序的 可扩展性 和 可维护性。
逻辑分析
订单状态包括：默认、已预订、已确认、已锁定。
订单事件包括：创建订单、确认订单、修改订单、支付订单。
通过上图我们还知道了状态与事件之间的关系，比如只有 已确认 的订单才可以进行 修改订单。
需要考虑如下问题：
- 当订单状态增加时，如何尽可能少的改动或改动对历史影响不大？
- 如果在同一入口调用，每个事件的处理方法需要的入参都有所不同，如何处理？
- 当某个事件完成后，有可能会进行发短信或客户端 Push 的操作，如何处理？
- 有可能某个事件，在不同平台(C端、商家后台、管理平台)的处理逻辑也有些不同，如何处理？
如何设计代码能够解决以上问题？
下面是我的一种代码实现，供大家参考，实现了在 创建订单 时，进行传入参数和完成后给用户发送短信，其他事件的操作，同理就可以实现。
代码实现
定义状态
// 定义订单状态
const (
StatusDefault = State(0)
StatusReserved = State(10)
StatusConfirmed = State(20)
StatusLocked = State(30)
)
// statusText 定义订单状态文案
var statusText = map[State]string{
StatusDefault: "默认",
StatusReserved: "已预订",
StatusConfirmed: "已确认",
StatusLocked: "已锁定",
}
// statusEvent 定义订单状态对应的可操作事件
var statusEvent = map[State][]Event{
StatusDefault: {EventCreate},
StatusReserved: {EventConfirm},
StatusConfirmed: {EventModify, EventPay},
}
func StatusText(status State) string {
return statusText[status]
}
当有新订单状态的增加时，在此文件中增加相应状态即可，同时维护好订单状态与订单事件之间的关系。
定义事件
// 定义订单事件
const (
EventCreate = Event("创建订单")
EventConfirm = Event("确定订单")
EventModify = Event("修改订单")
EventPay = Event("支付订单")
)
// 定义订单事件对应的处理方法
var eventHandler = map[Event]Handler{
EventCreate: handlerCreate,
EventConfirm: handlerConfirm,
EventModify: handlerModify,
EventPay: handlerPay,
}
当有新订单事件的增加时，在此文件中增加相应事件即可，同时维护好订单事件与事件实现方法之间的关系。
定义事件的处理方法
var (
// handlerCreate 创建订单
handlerCreate = Handler(func(opt *Opt) (State, error) {
message := fmt.Sprintf("正在处理创建订单逻辑，订单ID(%d), 订单名称(%s) ... 处理完毕！", opt.OrderId, opt.OrderName)
fmt.Println(message)
if opt.HandlerSendSMS != nil {
_ = opt.HandlerSendSMS("18888888888", "恭喜你预定成功了！")
}
return StatusReserved, nil
})
// handlerConfirm 确认订单
handlerConfirm = Handler(func(opt *Opt) (State, error) {
return StatusConfirmed, nil
})
// handlerModify 修改订单
handlerModify = Handler(func(opt *Opt) (State, error) {
return StatusReserved, nil
})
// handlerPay 支付订单
handlerPay = Handler(func(opt *Opt) (State, error) {
return StatusLocked, nil
})
)
在此文件中维护具体的事件处理方法，如果逻辑比较复杂可以考虑拆分文件处理。
核心代码
type State int // 状态
type Event string // 事件
type Handler func(opt *Opt) (State, error) // 处理方法，并返回新的状态
// FSM 有限状态机
type FSM struct {
mu sync.Mutex // 排他锁
state State // 当前状态
handlers map[State]map[Event]Handler // 当前状态可触发的有限个事件
}
// 获取当前状态
func (f *FSM) getState() State {
return f.state
}
// 设置当前状态
func (f *FSM) setState(newState State) {
f.state = newState
}
// addHandlers 添加事件和处理方法
func (f *FSM) addHandlers() (*FSM, error) {
...
return f, nil
}
// Call 事件处理
func (f *FSM) Call(event Event, opts ...Option) (State, error) {
f.mu.Lock()
defer f.mu.Unlock()
...
return f.getState(), nil
}
// NewFSM 实例化 FSM
func NewFSM(initState State) (fsm *FSM, err error) {
fsm = new(FSM)
fsm.state = initState
fsm.handlers = make(map[State]map[Event]Handler)
fsm, err = fsm.addHandlers()
if err != nil {
return
}
return
}
对订单状态的操作，只需要使用 Call 方法即可！
关于方法 addHandlers 和 Call 的代码就不贴了，在文章后面我提供了源码地址，供大家下载。
调用方式
例如当前状态为 默认状态，依次进行如下操作：
创建订单，状态变为已预订；修改订单，不可操作（已预订状态不可修改）；确定订单，状态变为已确认；修改订单，状态变为已预订；确定订单，状态变为已确认；支付订单，状态变为已锁定；
// 通过订单ID 或 其他信息查询到订单状态
orderStatus := order.StatusDefault
orderMachine, err := order.NewFSM(orderStatus)
if err != nil {
fmt.Println(err.Error())
return
}
// 创建订单，订单创建成功后再给用户发送短信
if _, err = orderMachine.Call(order.EventCreate,
order.WithOrderId(1),
order.WithOrderName("测试订单"),
order.WithHandlerSendSMS(sendSMS),
); err != nil {
fmt.Println(err.Error())
}
// 修改订单
if _, err = orderMachine.Call(order.EventModify); err != nil {
fmt.Println(err.Error())
}
// 确认订单
if _, err = orderMachine.Call(order.EventConfirm); err != nil {
fmt.Println(err.Error())
}
// 修改订单
if _, err = orderMachine.Call(order.EventModify); err != nil {
fmt.Println(err.Error())
}
// 确认订单
if _, err = orderMachine.Call(order.EventConfirm); err != nil {
fmt.Println(err.Error())
}
// 支付订单
if _, err = orderMachine.Call(order.EventPay); err != nil {
fmt.Println(err.Error())
}
输出：
正在处理创建订单逻辑，订单ID(1), 订单名称(测试订单) ... 处理完毕！
发送短信，给(18888888888)发送了(恭喜你预定成功了！)
操作[创建订单]，状态从 [默认] 变成 [已预订]
[警告] 状态(已预订)不允许操作(修改订单)
操作[确定订单]，状态从 [已预订] 变成 [已确认]
操作[修改订单]，状态从 [已确认] 变成 [已预订]
操作[确定订单]，状态从 [已预订] 变成 [已确认]
操作[支付订单]，状态从 [已确认] 变成 [已锁定]
小结
以上就是我的技术方案，希望能对你有所帮助，感兴趣的可以再进行封装，上述代码已提交到 github go-fsm-order，供下载使用。
作者：新亮笔记（关注公众号，可申请添加微信好友）
出处：https://www.cnblogs.com/xinliangcoder
本文版权归作者和博客园共有，欢迎转载，但未经作者同意必须保留此段声明，且在文章页面明显位置给出原文连接，否则保留追究法律责任的权利。