第19章组合模式

一概念

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

二 UML图

Component 组合中的对象声明接口，在适当情况下，实现所有类共有接口的默认行为。声明一个接口用于访问和管理Component的子部件
Leaf 在组合中表示叶节点对象，叶节点没有子节点。
Composite 定义有枝节点行为，用来存储子部件，在Component接口中实现与子部件有关的操作，比如增加Add和删除Remove

三何时使用组合模式

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

四 C++代码实现

#include "pch.h"
#include <iostream>
#include <string>
#include <list>
using namespace std;

//公司类 抽象类或接口
class Company
{
public:
    Company() {}
    Company(string str)
        :name(str) {}
    virtual void Add(Company* c) = 0;  //增加
    virtual void Remove(Company* c) = 0;  //移除
    virtual void Display(int depth) = 0;  //显示
    virtual void LineOfDuty() = 0;  //履行职责  
    inline bool operator==(const Company& company) const
    {
        return this->name == company.name;
    }
protected:
    string name;
};

class ConcreteCompany : public Company
{
public:
    ConcreteCompany(string name) : Company(name)
    {
        children = new list<Company*>;
    }
    void Add(Company* c) override
    {
        children->push_back(c);
    }
    void Remove(Company* c) override  //删除子节点
    {

        for (list<Company*>::iterator it = children->begin(); it != children->end(); ++it)
        {
            if (*(*it) == *c)
            {
                children->erase(it);
                break;
            }
        }
    }
    void Display(int depth) override  //显示
    {
        for (int ix = 0; ix < depth; ++ix)
        {
            cout << "-";
        }
        cout << this->name << endl;
        for (list<Company*>::iterator it = children->begin(); it != children->end(); ++it)
        {
            (*it)->Display(depth + 2);
        }
    }
    virtual void LineOfDuty() override  //履行职责
    {
        for (list<Company*>::iterator it = children->begin(); it != children->end(); ++it)
        {
            (*it)->LineOfDuty();
        }
    }
private:
    list<Company*>* children;
};

//人力资源部 树叶节点
class HRDepartment : public Company
{
public:
    HRDepartment(string str) : Company(str) {}

    void Add(Company* c) override  //增加
    {

    }
    void Remove(Company* c) override  //移除
    {

    }
    void Display(int depth) override  //显示
    {
        for (int ix = 0; ix < depth; ++ix)
        {
            cout << "-";
        }
        cout << name << endl;
    }
    void LineOfDuty() override  //履行职责  
    {
        cout << this->name << " 员工招聘培训管理" << endl;
    }
};

class FinanceDepartment : public Company
{
public:
    FinanceDepartment(string str) : Company(str) {}
    void Add(Company* c) override  //增加
    {

    }
    void Remove(Company* c) override  //移除
    {

    }
    void Display(int depth) override  //显示
    {
        for (int ix = 0; ix < depth; ++ix)
        {
            cout << "-";
        }
        cout << name << endl;
    }
    void LineOfDuty() override  //履行职责  
    {
        cout << this->name << " 公司财务收支管理" << endl;
    }
};

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("华东分公司财务部"));
    root->Add(comp);

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

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

    cout << "\n结构图" << endl;
    root->Display(1);

    cout << "\n职责：" << endl;
    root->LineOfDuty();

    return 0;
}

猜你喜欢