设计模式简单代码之Composite模式

/**********************************************************************************************************\
* 作者：一雨田（http://blog.csdn.net/dylgsy/）。本文可随便转贴，但请保留此信息
*
* Composite模式说明：
*
* 将对象以树形结构组织起来,以达成“部分－整体” 的层次结构，使得客户端对单个对象和组合对象的使用具有一致性.
*
* Composite比较容易理解，想到Composite就应该想到树形结构图。组合体内这些对象都有共同接口,当组合体一个对象的
* 方法被调用执行时，Composite将遍历(Iterator)整个树形结构,寻找同样包含这个方法的对象并实现调用执行。
* 可以用牵一动百来形容。
*
* 所以Composite模式使用到Iterator模式，和Chain of Responsibility模式类似。
*
* Composite是个很巧妙体现智慧的模式，在实际应用中，如果碰到树形结构，我们就可以尝试是否可以使用这个模式。
\**********************************************************************************************************/

/**********************************************************************************************************\
* 任务描述：
* 学校里的一个班中，老师和学生的关系可以使用Composite模式模拟
* 老师让学生去打扫卫生！对老师来说，让一个学生去打扫卫生和让一个组的学生去打扫卫生是一样的，
* 他可不用去到小组里一个一个的去通知学生们。
\**********************************************************************************************************/

#include <iostream>
#include <vector>
using namespace std;

// 定义人的抽象类，相当于Composite模式的Component类
class CPeopleComponent
{
public:
 CPeopleComponent()
 {
 }
 virtual ~CPeopleComponent()
 {
 }
 
 virtual void DoCleaning() = 0;

 virtual bool Add(const CPeopleComponent& people)
 {
  return false;
 }

 virtual bool Remove(CPeopleComponent people)
 {
  return true;
 }
 
 virtual CPeopleComponent* GetChild(int nIndex)
 {
  return 0;
 }

protected:
 char szName[20];
};

// 定义学生组合类，相当于Composite模式的Composite类
// 这里的vector借用一下STL的vector
// 正如GOF《设计模式》所说的，如果语言没有实现继承，那
// 继承也是我们要实现的模式。
class CStudentComposite: public CPeopleComponent
{
public:
 
 virtual void DoCleaning()
 {
  vector<CPeopleComponent*>::iterator pcIter = pcVec.begin();
  for(; pcIter != pcVec.end(); pcIter++)
  {
   (*pcIter)->DoCleaning();
  }
 }

 virtual bool Add(CPeopleComponent* people)
 {
  pcVec.push_back(people);
  return true;
 }

 virtual bool Remove(CPeopleComponent* people)
 {
  pcVec.erase(&people);
  return true;
 }
 
 virtual CPeopleComponent* GetChild(int nIndex)
 {
  return pcVec[nIndex];
 }

 // 提供一个函数清除资源
 virtual void ReleaseAll()
 {
  vector<CPeopleComponent*>::iterator pcIter = pcVec.begin();
  for(; pcIter != pcVec.end(); pcIter++)
  {
   delete (*pcIter);
  }
 }

private:
 vector<CPeopleComponent*> pcVec;
};

class CStudent: public CPeopleComponent
{
public:
 CStudent(const char *pszName)
 {
  strcpy(szName, pszName);
 }
 virtual ~CStudent()
 {
 } 
 virtual void DoCleaning()
 {
  cout << "我是学生：" << szName;
  cout << " 我打扫卫生去咯" << endl;
 } 
};


// 好，我们来模拟这个场景
void main()
{
 // 1个学生，老师是这样叫他的
 CStudent *s = new CStudent("1号");
 s->DoCleaning();

 // 1组学生，老师是这样叫他们的
 char szName[20];
 CStudentComposite sGroup;
 for(int i = 2; i < 11; i++)
 {
  sprintf(szName, "%d号", i);
  CStudent *ss = new CStudent(szName);
  sGroup.Add(ss);
 }

 sGroup.DoCleaning();

 // 可以看出老师对待一个或者一组学生的对待方式是一样的
 // 这个就是Composite模式的精神所在了
 
 delete s;
 sGroup.ReleaseAll();
}

/**********************************************************************************************************************\
* 最后要说的是：
* Composite模式在实现中有一个问题就是要提供对于子节点（Leaf）的管理策略，这里使用的是STL 中的vector，可以提供其他的
* 实现方式，如数组、链表、Hash表等。
* 当然这里只是入门，要真正理解和用好Composite模式，还需要大家积极努力去探索
\**********************************************************************************************************************/