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

// demo/db/transaction.go
package 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的执行方法Exec
func (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"
)

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

// src/os/exec/exec.go
package exec

// 对应命令模式中的Command
type 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{...})
  ...
}