首页 > itarticle > 设计模式之组合模式

设计模式之组合模式

admin 11月 13, 2020 0

Composite

属于数据结构模式

将对象组合成树形结构以表示部分-整体的层次结构使得用户对单个对象和组合对象的使用具有一致性(稳定)

class  {
public:
    virtual void process() = 0;
    virtual ~Component() { }
};

class Composite : public Component{
    string name;
    list<Component*> elements; // 存储的是指针 多态
public:
    Composite(const string &s) : name(s) {}
    void add(Component *element) {
        elements.push_back(element);
    }
    void remove(Component *element) {
        elements.remove(element);
    }
    void process() {
        // 1. process current node
        /* ... */
        // 2. process leaf nodes
        for (auto &e : elements) {
            e->process(); // 多态调用
        }
    }
};
// 叶子节点
class Leaf : public Component {
    string name;
public:
    Leaf(string s) : name(s) { }
    void process() {
        // process current node
    }
};
void Invoke(Component &c) {
    /* ... */
    c.process();
    /* ... */
}
int main() {
    Composite root("root");
    Composite treeNode1("treeNode1");
    Composite treeNode2("treeNode2");
    Composite treeNode3("treeNode3");
    Composite treeNode4("treeNode4");
    Leaf leat1("left1");
    Leaf leat2("left2");
    
    root.add(&treeNode1);
    treeNode1.add(&treeNode2);
    treeNode2.add(&leaf1);
    
    root.add(&treeNode3);
    treeNode3.add(&treeNode4);
    treeNode4.add(&leaf2);
    
    // 不管是根节点还是叶子节点 都能直接处理 这就是一致性
    Invoke(root);
    Invoke(leaf2);
    Invoke(treeNode3);
}