组合模式

1 核心思想

组合模式(Composite)，将对象组合成树形结构以表示“部分-整体”的层次结构。组合模式使得用户对单个对象和组合对象的使用具有一致性。

2 什么时候用组合模式及注意事项

当需求中需要体现部分和整体的层次结构，以及希望用户可以忽略组合对象和个体对象的不同，统一地使用组合结构中的所有对象时，就应该考虑使用组合模式。

组合模式就是实现了一个大而全的基类，这个基类包含个体对象和组合对象的所有可能的操作，也可以为这些操作提供默认的实现，也可以不提供实现，但必须保留接口。个体对象和组合对象都继承自基类，个体对象实现/重载自己需要的操作。组合对象也实现/重载自己需要的操作。

组合模式实现的时候需要平衡透明性和安全性，所谓透明性是指不需要知道一个对象到底是个体对象还是组合对象，一视同仁的处理它们。这样就要求基类提供全部的操作实现，那么对于叶子节点不支持的操作就调用基类的实现。这样代码调用过程中不需要进行具体对象是叶子节点还是组合节点的判断。但这样也带来了一个问题，对于一个叶子节点，很多没有意义的操作也被调用了。那么我们可以采用安全方式，在基类中不实现叶子节点不支持的操作，只在组合节点中实现这些操作。这样做不符合透明性，子节点和组合节点不具备相同的接口，因此客户端需要对正在处理的节点进行类型判断，会带来一些处理上的不便。在使用组合模式时，需要根据实际需求平衡透明性和安全性。

3 代码举例

采用透明方式来实现组合模式，公司总部管理财务部、人事部和分公司，分公司同样设立财务部和人事部。Company实现所有的操作，各基类根据自己的需要进行重载。

#include <iostream>
#include <list>
#include <string>

class Company
{
public:
	virtual void Add(Company* pCompany) = 0;
	virtual void Remove(Company* pCompany) = 0;
	virtual void Dispaly(int nDepth) = 0;
	virtual void LineOfDuty() = 0;
};

class HRDepartment :public Company
{
public:
	HRDepartment(std::string strName)
	{
		m_strName = strName;
	}

	//叶子节点不需要实现Add操作
	void Add(Company* pCompany){}

	//叶子节点不需要实现Remove操作
	void Remove(Company* pCompany) {}

	void Dispaly(int nDepth)
	{
		std::cout << std::string(nDepth, '-') << m_strName << std::endl;
	}

	void LineOfDuty()
	{
		std::cout << m_strName << "  负责人力资源相关工作!" << std::endl;
	}

private:
	std::string m_strName;
};

class FinanceDepartment :public Company
{
public:
	FinanceDepartment(std::string strName)
	{
		m_strName = strName;
	}

	//叶子节点不需要实现Add操作
	void Add(Company* pCompany) {}

	//叶子节点不需要实现Remove操作
	void Remove(Company* pCompany) {}

	void Dispaly(int nDepth)
	{
		std::cout << std::string(nDepth, '-') << m_strName << std::endl;
	}

	void LineOfDuty()
	{
		std::cout << m_strName << "  负责财务管理相关工作!" << std::endl;
	}

private:
	std::string m_strName;
};

class ConcreteCompany :public Company
{
public:
	ConcreteCompany(std::string strName)
	{
		m_strName = strName;
	}

	//组合节点需要实现Add操作
	void Add(Company* pCompany) 
	{
		m_listCompanies.push_back(pCompany);
	}

	//组合节点需要实现Remove操作
	void Remove(Company* pCompany) 
	{
		std::list<Company*>::iterator iter = m_listCompanies.begin();
		while (iter != m_listCompanies.end())
		{
			if (*iter == pCompany)
			{
				iter = m_listCompanies.erase(iter);
				break;
			}

			iter++;
		}
	}

	void Dispaly(int nDepth)
	{
		std::cout << std::string(nDepth, '-') << m_strName << std::endl;

		std::list<Company*>::iterator iter = m_listCompanies.begin();
		while (iter != m_listCompanies.end())
		{
			(*iter)->Dispaly(nDepth + 2);
			iter++;
		}
	}

	void LineOfDuty()
	{
		std::list<Company*>::iterator iter = m_listCompanies.begin();
		while (iter != m_listCompanies.end())
		{
			(*iter)->LineOfDuty();
			iter++;
		}
	}

private:
	std::string m_strName;
	std::list<Company*> m_listCompanies;
};

int main()
{
	Company* root = new ConcreteCompany("北京总部");
	root->Add(new HRDepartment("总部人力资源部"));
	root->Add(new FinanceDepartment("总部财务部"));

	ConcreteCompany* comp = new ConcreteCompany("华东分公司");
	comp->Add(new HRDepartment("华东分公司人力资源部"));
	comp->Add(new FinanceDepartment("华东分公司财务部"));

	ConcreteCompany* comp1 = new ConcreteCompany("杭州办事处");
	comp1->Add(new HRDepartment("杭州办事处人力资源部"));
	comp1->Add(new FinanceDepartment("杭州办事处财务部"));

	comp->Add(comp1);

	root->Add(comp);

	std::cout << "组织架构图:" << std::endl;
	root->Dispaly(1);

	std::cout << std::endl << std::endl << "职责:" << std::endl;
	root->LineOfDuty();

	return 0;
}