Memento Pattern

1.Memento 模式的关键是在不破坏封装的前提下，捕获并保存一个类的内部状态，这样就可以利用该保存的状态实施恢复操作。

2.Memento 模式结构图

3.实现

#ifndef _MEMENTO_H_ 
#define _MEMENTO_H_

#include <string> 
using namespace std;
class Memento;

class Originator 
{ 
public: 
    typedef string State;
    Originator();
    Originator(const State& sdt);
    ~Originator();
    Memento* CreateMemento();
    void SetMemento(Memento* men);
    void RestoreToMemento(Memento* mt);
    State GetState();
    void SetState(const State& sdt);
    void PrintState();
protected:
private: 
    State _sdt;
    Memento* _mt; 
};

class Memento 
{ 
public:
protected:
private: 
    //这是最关键的地方，将Originator为friend类，可以访问内部信息，但是其他类不能访问 
    friend class Originator; 
    typedef string State;
    Memento();
    Memento(const State& sdt);
    ~Memento();
    void SetState(const State& sdt);
    State GetState();

private: 
        State _sdt;
};

#endif

#include "Memento.h"
#include <iostream>

using namespace std;
typedef string State;

Originator::Originator() 
{ 
    _sdt = "";
    _mt = 0; 
}
Originator::Originator(const State& sdt)
{ 
    _sdt = sdt;
    _mt = 0; 
}
Originator::~Originator() 
{

}
Memento* Originator::CreateMemento() 
{ 
    return new Memento(_sdt); 
}
State Originator::GetState() 
{ 
    return _sdt; 
}
void Originator::SetState(const State& sdt)
{
    _sdt = sdt;
}
void Originator::PrintState() 
{ 
    cout<<this->_sdt<<"....."<<endl; 
}
void Originator::SetMemento(Memento* men)
{

}
void Originator::RestoreToMemento(Memento* mt) 
{ 
    this->_sdt = mt->GetState();
}
//class Memento
Memento::Memento() 
{

}
Memento::Memento(const State& sdt) 
{ 
    _sdt = sdt;
}
State Memento::GetState() 
{ 
    return _sdt;
}
void Memento::SetState(const State& sdt) 
{ 
    _sdt = sdt; 
}

#include "Memento.h"
#include <iostream> 
using namespace std;

int main(int argc,char* argv[]) 
{ 
    Originator* o = new Originator();
    o->SetState("old"); //备忘前状态
    o->PrintState();
    Memento* m = o->CreateMemento(); //将状态备忘
    o->SetState("new"); //修改状态
    o->PrintState();
    o->RestoreToMemento(m); //恢复修改前状态
    o->PrintState();
    return 0;
}

4.代码说明

　　Memento模式的关键就是friend class Originator;我们可以看到，Memento的接口都声明为private，而将Originator声明为Memento的友元类。我们将Originator的状态保存在Memento类中，而将Memento接口private起来，也就达到了封装的功效。
　　在Originator类中我们提供了方法让用户后悔：RestoreToMemento(Memento* mt)我们可以通过这个接口让用户后悔。