簡介

相對于代理模式、工廠模式等設(shè)計模式，備忘錄模式（Memento）在我們?nèi)粘ｉ_發(fā)中出鏡率并不高，除了應(yīng)用場景的限制之外，另一個原因，可能是備忘錄模式 UML 結(jié)構(gòu)的幾個概念比較晦澀難懂，難以映射到代碼實現(xiàn)中。比如 Originator（原發(fā)器）和 Caretaker（負責(zé)人），從字面上很難看出它們在模式中的職責(zé)。

但從定義來看，備忘錄模式又是簡單易懂的，GoF 對備忘錄模式的定義如下：

Without violating encapsulation, capture and externalize an object’s internal state so that the object can be restored to this state later.

也即，在不破壞封裝的前提下，捕獲一個對象的內(nèi)部狀態(tài)，并在該對象之外進行保存，以便在未來將對象恢復(fù)到原先保存的狀態(tài)。

從定義上看，備忘錄模式有幾個關(guān)鍵點：封裝、保存、恢復(fù)。

對狀態(tài)的封裝，主要是為了未來狀態(tài)修改或擴展時，不會引發(fā)霰彈式修改；保存和恢復(fù)則是備忘錄模式的主要特點，能夠?qū)Ξ?dāng)前對象的狀態(tài)進行保存，并能夠在未來某一時刻恢復(fù)出來。

現(xiàn)在，在回過頭來看備忘錄模式的 3 個角色就比較好理解了：

Memento（備忘錄）：是對狀態(tài)的封裝，可以是struct，也可以是interface。

Originator（原發(fā)器）：備忘錄的創(chuàng)建者，備忘錄里存儲的就是 Originator 的狀態(tài)。

Caretaker（負責(zé)人）：負責(zé)對備忘錄的保存和恢復(fù)，無須知道備忘錄中的實現(xiàn)細節(jié)。

UML 結(jié)構(gòu)

場景上下文

在前文【Go實現(xiàn)】實踐GoF的23種設(shè)計模式：命令模式我們提到，在簡單的分布式應(yīng)用系統(tǒng)（示例代碼工程）中，db 模塊用來存儲服務(wù)注冊信息和系統(tǒng)監(jiān)控數(shù)據(jù)。其中，服務(wù)注冊信息拆成了profiles和regions兩個表，在服務(wù)發(fā)現(xiàn)的業(yè)務(wù)邏輯中，通常需要同時操作兩個表，為了避免兩個表數(shù)據(jù)不一致的問題，db 模塊需要提供事務(wù)功能:

事務(wù)的核心功能之一是，當(dāng)其中某個語句執(zhí)行失敗時，之前已執(zhí)行成功的語句能夠回滾，前文我們已經(jīng)介紹如何基于命令模式搭建事務(wù)框架，下面我們將重點介紹，如何基于備忘錄模式實現(xiàn)失敗回滾的功能。

代碼實現(xiàn)

//demo/db/transaction.go
packagedb

//Command執(zhí)行數(shù)據(jù)庫操作的命令接口，同時也是備忘錄接口
//關(guān)鍵點1：定義Memento接口，其中Exec方法相當(dāng)于UML圖中的SetState方法，調(diào)用后會將狀態(tài)保存至Db中
typeCommandinterface{
Exec()error//Exec執(zhí)行insert、update、delete命令
Undo()//Undo回滾命令
setDb(dbDb)//SetDb設(shè)置關(guān)聯(lián)的數(shù)據(jù)庫
}

//關(guān)鍵點2：定義Originator，在本例子中，狀態(tài)都是存儲在Db對象中
typeDbinterface{...}

//TransactionDb事務(wù)實現(xiàn)，事務(wù)接口的調(diào)用順序為begin->exec->exec>...->commit
//關(guān)鍵點3：定義Caretaker，Transaction里實現(xiàn)了對語句的執(zhí)行（Do）和回滾（Undo）操作
typeTransactionstruct{
namestring
//關(guān)鍵點4：在Caretaker（Transaction）中引用Originator（Db）對象，用于后續(xù)對其狀態(tài)的保存和恢復(fù)
dbDb
//注意，這里的cmds并非備忘錄列表，真正的history在Commit方法中
cmds[]Command
}
//Begin開啟一個事務(wù)
func(t*Transaction)Begin(){
t.cmds=make([]Command,0)
}
//Exec在事務(wù)中執(zhí)行命令，先緩存到cmds隊列中，等commit時再執(zhí)行
func(t*Transaction)Exec(cmdCommand)error{
ift.cmds==nil{
returnErrTransactionNotBegin
}
cmd.setDb(t.db)
t.cmds=append(t.cmds,cmd)
returnnil
}
//Commit提交事務(wù)，執(zhí)行隊列中的命令，如果有命令失敗，則回滾后返回錯誤
func(t*Transaction)Commit()error{
//關(guān)鍵點5：定義備忘錄列表，用于保存某一時刻的系統(tǒng)狀態(tài)
history:=&cmdHistory{history:make([]Command,0,len(t.cmds))}
for_,cmd:=ranget.cmds{
//關(guān)鍵點6：執(zhí)行Do方法
iferr:=cmd.Exec();err!=nil{
//關(guān)鍵點8：當(dāng)Do方法執(zhí)行失敗時，則進行Undo操作，根據(jù)備忘錄history中的狀態(tài)進行回滾
history.rollback()
returnerr
}
//關(guān)鍵點7：如果Do方法執(zhí)行成功，則將狀態(tài)（cmd）保存在備忘錄history中
history.add(cmd)
}
returnnil
}
//cmdHistory命令執(zhí)行歷史
typecmdHistorystruct{
history[]Command
}
func(c*cmdHistory)add(cmdCommand){
c.history=append(c.history,cmd)
}

func(c*cmdHistory)rollback(){
fori:=len(c.history)-1;i>=0;i--{
c.history[i].Undo()
}
}

//InsertCmd插入命令
//關(guān)鍵點9:定義具體的備忘錄類，實現(xiàn)Memento接口
typeInsertCmdstruct{
dbDb
tableNamestring
primaryKeyinterface{}
newRecordinterface{}
}

func(i*InsertCmd)Exec()error{
returni.db.Insert(i.tableName,i.primaryKey,i.newRecord)
}
func(i*InsertCmd)Undo(){
i.db.Delete(i.tableName,i.primaryKey)
}
func(i*InsertCmd)setDb(dbDb){
i.db=db
}

//UpdateCmd更新命令
typeUpdateCmdstruct{...}
//DeleteCmd刪除命令
typeDeleteCmdstruct{...}

客戶端可以這么使用：

funcclient(){
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)
iferr:=transaction.Commit();err!=nil{
return...
}
return...
}

這里并沒有完全按照標準的備忘錄模式 UML 進行實現(xiàn)，但本質(zhì)是一樣的，總結(jié)起來有以下幾個關(guān)鍵點：

定義抽象備忘錄 Memento 接口，這里為Command接口。Command的實現(xiàn)是具體的數(shù)據(jù)庫執(zhí)行操作，并且存有對應(yīng)的回滾操作，比如InsertCmd為“插入”操作，其對應(yīng)的回滾操作為“刪除”，我們保存的狀態(tài)就是“刪除”這一回滾操作。

定義 Originator 結(jié)構(gòu)體/接口，這里為Db接口。備忘錄Command記錄的就是它的狀態(tài)。

定義 Caretaker 結(jié)構(gòu)體/接口，這里為Transaction結(jié)構(gòu)體。Transaction采用了延遲執(zhí)行的設(shè)計，當(dāng)調(diào)用Exec方法時只會將命令緩存到cmds隊列中，等到調(diào)用Commit方法時才會執(zhí)行。

在 Caretaker 中引用 Originator 對象，用于后續(xù)對其狀態(tài)的保存和恢復(fù)。這里為Transaction聚合了Db。

在 Caretaker 中定義備忘錄列表，用于保存某一時刻的系統(tǒng)狀態(tài)。這里為在Transaction.Commit方法中定義了cmdHistory對象，保存一直執(zhí)行成功的Command。

執(zhí)行 Caretaker 具體的業(yè)務(wù)邏輯，這里為在Transaction.Commit中調(diào)用Command.Exec方法，執(zhí)行具體的數(shù)據(jù)庫操作命令。

業(yè)務(wù)邏輯執(zhí)行成功后，保存當(dāng)前的狀態(tài)。這里為調(diào)用cmdHistory.add方法將Command保存起來。

如果業(yè)務(wù)邏輯執(zhí)行失敗，則恢復(fù)到原來的狀態(tài)。這里為調(diào)用cmdHistory.rollback方法，反向執(zhí)行已執(zhí)行成功的Command的Undo方法進行狀態(tài)恢復(fù)。

根據(jù)具體的業(yè)務(wù)需要，定義具體的備忘錄，這里定義了InsertCmd、UpdateCmd和DeleteCmd。

擴展

MySQL 的 undo log 機制

MySQL 的undo log（回滾日志）機制本質(zhì)上用的就是備忘錄模式的思想，前文中Transaction回滾機制實現(xiàn)的方法參考的就是 undo log 機制。

undo log 原理是，在提交事務(wù)之前，會把該事務(wù)對應(yīng)的回滾操作（狀態(tài)）先保存到 undo log 中，然后再提交事務(wù)，當(dāng)出錯的時候 MySQL 就可以利用 undo log 來回滾事務(wù)，即恢復(fù)原先的記錄值。

比如，執(zhí)行一條插入語句：

insertintoregion(id,name)values(1,"beijing");

那么，寫入到 undo log 中對應(yīng)的回滾語句為：

deletefromregionwhereid=1;

當(dāng)執(zhí)行一條語句失敗，需要回滾時，MySQL 就會從讀取對應(yīng)的回滾語句來執(zhí)行，從而將數(shù)據(jù)恢復(fù)至事務(wù)提交之前的狀態(tài)。undo log 是 MySQL 實現(xiàn)事務(wù)回滾和多版本控制（MVCC）的根基。

典型應(yīng)用場景

事務(wù)回滾。事務(wù)回滾的一種常見實現(xiàn)方法是 undo log，其本質(zhì)上用的就是備忘錄模式。

系統(tǒng)快照（Snapshot）。多版本控制的用法，保存某一時刻的系統(tǒng)狀態(tài)快照，以便在將來能夠恢復(fù)。

撤銷功能。比如 Microsoft Offices 這類的文檔編輯軟件的撤銷功能。

優(yōu)缺點

優(yōu)點

提供了一種狀態(tài)恢復(fù)的機制，讓系統(tǒng)能夠方便地回到某個特定狀態(tài)下。

實現(xiàn)了對狀態(tài)的封裝，能夠在不破壞封裝的前提下實現(xiàn)狀態(tài)的保存和恢復(fù)。

缺點

資源消耗大。系統(tǒng)狀態(tài)的保存意味著存儲空間的消耗，本質(zhì)上是空間換時間的策略。undo log 是一種折中方案，保存的狀態(tài)并非某一時刻數(shù)據(jù)庫的所有數(shù)據(jù)，而是一條反操作的 SQL 語句，存儲空間大大減少。

并發(fā)安全。在多線程場景，實現(xiàn)備忘錄模式時，要注意在保證狀態(tài)的不變性，否則可能會有并發(fā)安全問題。

與其他模式的關(guān)聯(lián)

在實現(xiàn) Undo/Redo 操作時，你通常需要同時使用備忘錄模式與命令模式。

另外，當(dāng)你需要遍歷備忘錄對象中的成員時，通常會使用迭代器模式，以防破壞對象的封裝。

文章配圖

可以在用Keynote畫出手繪風(fēng)格的配圖中找到文章的繪圖方法。

審核編輯：劉清