Go语言实现GoF设计模式：备忘录模式的实践探索 - ToB企服应用市场:ToB评测及商务社交产业平台

// demo/db/transaction.go
package 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;

复制代码