memento.h

来自「常见设计模式的解析和实现,c++实现,经典的设计模式解析源码」· C头文件 代码 · 共 51 行

/********************************************************************
	created:	2006/08/09
	filename: 	Memento.h
	author:		李创
                http://www.cppblog.com/converse/

	purpose:	Memento模式的演示代码
*********************************************************************/

#ifndef MEMENTO_H
#define MEMENTO_H

#include <string>

typedef std::string State;

class Memento;

class Originator
{
public:
	Originator(const State& rState);
	Originator();
	~Originator();

	Memento*	CreateMemento();
	void		SetMemento(Memento* pMemento);
	State		GetState();
	void		SetState(const State& rState);
	void		RestoreState(Memento* pMemento);
	void		PrintState();

private:
	State		m_State;
};

// 把Memento的接口函数都设置为私有的,而Originator是它的友元,
// 这样保证了只有Originator可以对其访问
class Memento
{
private:
	friend class Originator;
	Memento(const State& rState);
	void	SetState(const State& rState);
	State	GetState();

	State	m_State;
};

#endif