【Go 实现】实践 GoF 的 23 种设计模式：备忘录模式

// demo/db/transaction.gopackage db
// Command 执行数据库操作的命令接口，同时也是备忘录接口// 关键点1：定义Memento接口，其中Exec方法相当于UML图中的SetState方法，调用后会将状态保存至Db中type Command interface {    Exec() error // Exec 执行insert、update、delete命令    Undo() // Undo 回滚命令    setDb(db Db) // SetDb 设置关联的数据库}
// 关键点2：定义Originator，在本例子中，状态都是存储在Db对象中type Db interface {...}
// Transaction Db事务实现，事务接口的调用顺序为begin -> exec -> exec > ... -> commit// 关键点3：定义Caretaker，Transaction里实现了对语句的执行（Do）和回滚（Undo）操作type Transaction struct {    name string    // 关键点4：在Caretaker（Transaction）中引用Originator（Db）对象，用于后续对其状态的保存和恢复    db   Db    // 注意，这里的cmds并非备忘录列表，真正的history在Commit方法中    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 提交事务，执行队列中的命令，如果有命令失败，则回滚后返回错误func (t *Transaction) Commit() error {    // 关键点5：定义备忘录列表，用于保存某一时刻的系统状态    history := &cmdHistory{history: make([]Command, 0, len(t.cmds))}    for _, cmd := range t.cmds {        // 关键点6：执行Do方法        if err := cmd.Exec(); err != nil {            // 关键点8：当Do方法执行失败时，则进行Undo操作，根据备忘录history中的状态进行回滚            history.rollback()            return err        }        // 关键点7：如果Do方法执行成功，则将状态（cmd）保存在备忘录history中        history.add(cmd)    }    return nil}// cmdHistory 命令执行历史type cmdHistory struct {    history []Command}func (c *cmdHistory) add(cmd Command) {    c.history = append(c.history, cmd)}
func (c *cmdHistory) rollback() {    for i := len(c.history) - 1; i >= 0; i-- {        c.history[i].Undo()    }}
// InsertCmd 插入命令// 关键点9: 定义具体的备忘录类，实现Memento接口type InsertCmd struct {    db         Db    tableName  string    primaryKey interface{}    newRecord  interface{}}
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 ...}

insert into region(id, name) values (1, "beijing");

delete from region where id = 1;