Memento (C++實現）

// Memento.cpp : 定義控制台應用程式的進入點。
//

#include "stdafx.h"
#include <iostream>
#include <string>

 

using namespace std;

 

typedef string STATE;

 

class Memento;

class Originator
{
public:
 Originator(){}
 Originator(const STATE& st):m_state(st)
 {
 }
 virtual ~Originator(){}

 Memento* CreateMemento();

 void SetMemento(Memento* pMemeto);

 STATE getState()
 {
  return m_state;
 }

 void setState(const STATE& st  )
 {
  m_state=st;
 }

 void printState()
 {
  cout<<m_state<<endl;
 }

private:
 STATE  m_state;

};

 

class Memento
{
private:
 friend class Originator;
 Memento(const STATE& st);
 

 void setState(const STATE& st)
 {
   m_state=st;
 }

 STATE getState();
 
 STATE  m_state;
};

 

Memento::Memento(const STATE& st)
 {
   m_state=st;
 }

STATE Memento::getState()
 {
  return m_state;
 }

void Originator::SetMemento(Memento* pMemento)
 {
  if(NULL!=pMemento)
   m_state=pMemento->getState();
 }

Memento *Originator::CreateMemento()
 {
  return new Memento(m_state);
 }

 

int _tmain(int argc, _TCHAR* argv[])
{
 
 // 建立一個原發器
 Originator* pOriginator = new Originator("old state");
 pOriginator->printState();

 // 建立一個備忘錄存放這個原發器的狀態
 Memento *pMemento = pOriginator->CreateMemento();
 
 // 更改原發器的狀態
 pOriginator->setState("new state");
 pOriginator->printState();

 // 通過備忘錄把原發器的狀態還原到之前的狀態
 pOriginator->SetMemento(pMemento);
 pOriginator->printState();

 delete pOriginator;
 delete pMemento;

 return 0;
}