代码如下:
#include <iostream>
#include <string>
#include <memory>
#include <vector>
#include <algorithm>
class component
{
public:
component(std::string name)
{
m_name = name;
}
virtual void add(component * p) = 0;
virtual void remove(component * p) = 0;
virtual void operation(void) = 0;
protected:
std::string m_name;
};
class leaf : public component
{
public:
leaf(std::string name) : component(name) {}
void add(component * p)
{
std::cout << "leaf xxx add" << std::endl;
}
void remove(component * p)
{
std::cout << "leaf xxx remove" << std::endl;
}
void operation(void)
{
std::cout << "----" << m_name << std::endl;
}
};
class composite : public component
{
public:
composite(std::string name) : component(name) {};
~composite(void)
{
if (!m_vec.empty())
{
m_vec.clear();
}
}
void add(component * p)
{
auto it = find_if(m_vec.begin(), m_vec.end(),
[p](std::shared_ptr<component> ptr) {return p == ptr.get(); });
if (it == m_vec.end())
m_vec.push_back(std::shared_ptr<component>(p));
}
void remove(component * p)
{
auto it = find_if(m_vec.begin(), m_vec.end(),
[p](std::shared_ptr<component> ptr) {return p == ptr.get(); });
if (it == m_vec.end())
return;
m_vec.erase(it);
}
void operation(void)
{
std::cout << "operation composite" << std::endl;
for (auto it = m_vec.cbegin(); it != m_vec.cend(); it++)
{
(*it)->operation();
}
}
private:
std::vector<std::shared_ptr<component>> m_vec;
};
///////////////////////////////////////
int main()
{
std::cout << "start-up .." << std::endl;
component * p = new composite("总部");
p->add(new leaf("总部财务部门"));
p->add(new leaf("总部人力资源部门"));
component * p1 = new composite("上海分部");
p1->add(new leaf("上海分部财务部门"));
p1->add(new leaf("上海分部人力资源部门"));
p->add(p1);
p->operation();
delete p1;
delete p;
std::cout << "done .." << std::endl;
return 0;
}
运行如下: