有时候我们需要在某个类已有的功能中添加新的功能。此时我们可以用继承的方式来给类添加功能。我们也可以用装饰模式来实现。
装饰模式,动态的给一个对象添加功能。与继承的方式相比,它更有灵活性。
装饰模式实现思路是。在原有对象上嵌入想要扩展的装饰器。装饰器实现装饰功能。
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实现功能,然后在主函数中动态添加。