【Go 实现】实践 GoF 的 23 种设计模式：命令模式

// demo/db/transaction.gopackage db
// Command 执行数据库操作的命令接口// 关键点1: 定义命令抽象接口type Command interface {  // 关键点2: 命令抽象接口中声明执行命令的方法    Exec() error // Exec 执行insert、update、delete命令  // 关键点3: 如果有撤销功能，则需要定义Undo方法    Undo() // Undo 回滚命令    setDb(db Db) // SetDb 设置关联的数据库}
// Transaction Db事务实现，事务接口的调用顺序为begin -> exec -> exec > ... -> commit// 关键点4: 定义Invoker对象type Transaction struct {    name string    db   Db  // 关键点5: Invoker对象持有Command的引用    cmds []Command}// Begin 开启一个事务func (t *Transaction) Begin() {    t.cmds = make([]Command, 0)}// Exec 在事务中执行命令，先缓存到cmds队列中，等commit时再执行func (t *Transaction) Exec(cmd Command) error {    if t.cmds == nil {        return ErrTransactionNotBegin    }    cmd.setDb(t.db)    t.cmds = append(t.cmds, cmd)    return nil}// Commit 提交事务，执行队列中的命令，如果有命令失败，则回滚后返回错误// 关键点6: 为Invoker对象定义Call方法，在方法内调用Command的执行方法Execfunc (t *Transaction) Commit() error {    history := &cmdHistory{history: make([]Command, 0, len(t.cmds))}    for _, cmd := range t.cmds {        if err := cmd.Exec(); err != nil {            history.rollback()            return err        }        history.add(cmd)    }    return nil}// cmdHistory 命令执行历史type cmdHistory struct {    history []Command}func (c *cmdHistory) add(cmd Command) {    c.history = append(c.history, cmd)}// 关键点7: 在回滚方法中，调用已执行命令的Undo方法func (c *cmdHistory) rollback() {    for i := len(c.history) - 1; i >= 0; i-- {        c.history[i].Undo()    }}
// InsertCmd 插入命令// 关键点8: 定义具体的命令类，实现Command接口type InsertCmd struct {  // 关键点9: 命令通常持有接收者的引用，以便在执行方法中与接收者交互    db         Db    tableName  string    primaryKey interface{}    newRecord  interface{}}// 关键点10: 命令对象执行方法中，调用Receiver的Action方法，这里的Receiver为db对象，Action方法为Insert方法func (i *InsertCmd) Exec() error {    return i.db.Insert(i.tableName, i.primaryKey, i.newRecord)}func (i *InsertCmd) Undo() {    i.db.Delete(i.tableName, i.primaryKey)}func (i *InsertCmd) setDb(db Db) {    i.db = db}
// UpdateCmd 更新命令type UpdateCmd struct {...}// DeleteCmd 删除命令type DeleteCmd struct {...}

func client() {    transaction := db.CreateTransaction("register" + profile.Id)    transaction.Begin()    rcmd := db.NewUpdateCmd(regionTable).WithPrimaryKey(profile.Region.Id).WithRecord(profile.Region)    transaction.Exec(rcmd)    pcmd := db.NewUpdateCmd(profileTable).WithPrimaryKey(profile.Id).WithRecord(profile.ToTableRecord())    transaction.Exec(pcmd)    if err := transaction.Commit(); err != nil {        return ...     }  return ...}

package main
import (  "os/exec")
// 对应命令模式中的Invokerfunc main() {  cmd := exec.Command("sleep", "1")  err := cmd.Run()}

// src/os/exec/exec.gopackage exec
// 对应命令模式中的Commandtype Cmd struct {  ...  // 对应命令模式中的Receiver    Process *os.Process  ...}
// 对应命令模式中Command的执行方法func (c *Cmd) Run() error {    if err := c.Start(); err != nil {        return err    }    return c.Wait()}
func (c *Cmd) Start() error {  ...  // Command与Receiver的交互  c.Process, err = os.StartProcess(c.Path, c.argv(), &os.ProcAttr{...})  ...}