设计模式之装饰模式 C++实现

    有时候我们需要在某个类已有的功能中添加新的功能。此时我们可以用继承的方式来给类添加功能。我们也可以用装饰模式来实现。

    装饰模式，动态的给一个对象添加功能。与继承的方式相比，它更有灵活性。

    装饰模式实现思路是。在原有对象上嵌入想要扩展的装饰器。装饰器实现装饰功能。

UML图如下；

其中，Component定义一个对象接口，可以动态的给这些对象添加功能。Decorator，装饰抽象类。继承自Component，从外类来扩展Component。但是对于Component来说，是无需知道Decoratoe的存在的。

ConcreteComponent是具体的对象。而ConcreteDecoratorA和B是具体的装饰类。负责添加功能。

实例代码如下；

首先，需要装饰的对象及其抽象父类定义如下。

#pragma once

#include <string>

class Person
{
    public:
    Person();
    virtual void show()=0;
};

class NamePerson:public Person
{
    public:
    NamePerson(std::string name)
    {
        m_name = name;
    }
    virtual void show()
    {
        printf("show %s\n", m_name);
    }

    private:
    std::string m_name;
};
           

NamePerson对应ConcreteComponent。是需要装饰的对象。

装饰类实现如下；

#pragma once

#include "Person.h"

class DecoratePerson:public Person
{
    public:
    DecoratePerson(Person* person_)
    {
        m_person = person_;
    }
    virtual void show()
    {
        m_person->show();
    }

    private:
    Person *m_person;
};

class DecoratePersonA:public DecoratePerson
{
    public:
    DecoratePersonA(Person* person_):DecoratePerson(person_)
    {
        this->m_person = person_;
    }
    ~DecoratePersonA(){};
    virtual void show()
    {
        printf("DecoratePersonA\n");
        m_person->show();
    }
    private:
    Person* m_person;
};

class DecoratePersonB:public DecoratePerson
{
    public:
    DecoratePersonB(Person* person_):DecoratePerson(person_)
    {
        this->m_person = person_;
    }
    virtual void show()
    {
        printf("DecoratePersonB\n");
        m_person->show();
    }
    private:
    Person* m_person;
};
           

DecoratePerson对应Decorator。是装饰类的抽象父类。DecoratePersonA对应ConcreteDecoratorA。具体装饰器。

然后可以在主函数中像下面这样使用。

#include "DecoratePerson.h"

int main(int argc, char* argv[])
{
    Person* onePerson = new NamePerson("zhangsan");
    Person* DecorateA = new DecoratePersonA(onePerson);
    Person* DecorateB = new DecoratePersonB(DecorateA);

    DecorateB->show();

    return 0;
}
           

当需要新增加功能的时候我们只需在添加一个ConcreteDecorator实现功能，然后在主函数中动态添加。