装饰器模式通过组合替代继承的方式实现功能扩展,避免了继承体系过于臃肿的问题,在C++中可以通过抽象基类和装饰器类的嵌套调用来实现动态添加功能的效果。

装饰器模式的核心结构
装饰器模式的实现通常包含四个核心角色:
- 抽象组件:定义被装饰对象和装饰器共同遵循的接口,是所有类的基类
- 具体组件:抽象组件的具体实现,是被装饰的原始对象
- 抽象装饰器:继承抽象组件,同时持有一个抽象组件的指针或引用,是所有具体装饰器的基类
- 具体装饰器:继承抽象装饰器,在调用原始对象方法的前后添加自定义的功能逻辑
C++实现装饰器模式示例
下面以给基础文本输出功能动态添加边框、颜色标记功能为例,展示完整的实现过程。
1. 定义抽象组件
首先定义文本输出的抽象基类,声明统一的输出接口:
#include <iostream>
#include <string>
// 抽象组件:文本输出接口
class TextComponent {
public:
virtual ~TextComponent() = default;
// 输出文本的纯虚函数
virtual std::string render() const = 0;
};
2. 实现具体组件
实现基础的无格式文本输出类,作为被装饰的原始对象:
// 具体组件:基础文本实现
class PlainText : public TextComponent {
private:
std::string content;
public:
explicit PlainText(const std::string& text) : content(text) {}
std::string render() const override {
return content;
}
};
3. 定义抽象装饰器
抽象装饰器继承抽象组件,同时持有抽象组件的指针,所有具体装饰器都继承这个类:
// 抽象装饰器:继承抽象组件,持有组件指针
class TextDecorator : public TextComponent {
protected:
TextComponent* component;
public:
explicit TextDecorator(TextComponent* comp) : component(comp) {}
~TextDecorator() override {
delete component;
}
std::string render() const override {
return component->render();
}
};
4. 实现具体装饰器
分别实现添加边框和添加颜色标记的两个具体装饰器:
// 具体装饰器:添加边框
class BorderDecorator : public TextDecorator {
public:
explicit BorderDecorator(TextComponent* comp) : TextDecorator(comp) {}
std::string render() const override {
// 先调用原始组件的render方法,再包裹边框
return "[" + component->render() + "]";
}
};
// 具体装饰器:添加颜色标记
class ColorDecorator : public TextDecorator {
private:
std::string color;
public:
ColorDecorator(TextComponent* comp, const std::string& clr)
: TextDecorator(comp), color(clr) {}
std::string render() const override {
// 先调用原始组件的render方法,再添加颜色标记
return "<" + color + ">" + component->render() + "</" + color + ">";
}
};
5. 功能动态组合使用
通过嵌套创建装饰器对象,实现功能的动态叠加:
int main() {
// 创建基础文本对象
TextComponent* text = new PlainText("Hello Decorator");
std::cout << "基础文本:" << text->render() << std::endl;
// 动态添加边框功能
TextComponent* borderedText = new BorderDecorator(text);
std::cout << "添加边框后:" << borderedText->render() << std::endl;
// 在边框基础上再添加颜色标记功能
TextComponent* coloredBorderedText = new ColorDecorator(borderedText, "red");
std::cout << "添加颜色后:" << coloredBorderedText->render() << std::endl;
// 也可以先加颜色再加边框,顺序可以灵活调整
TextComponent* text2 = new PlainText("Test Order");
TextComponent* coloredText2 = new ColorDecorator(text2, "blue");
TextComponent* borderedColoredText2 = new BorderDecorator(coloredText2);
std::cout << "调整顺序后:" << borderedColoredText2->render() << std::endl;
// 释放最外层装饰器的内存即可,内部组件会被递归释放
delete coloredBorderedText;
delete borderedColoredText2;
return 0;
}
装饰器模式的注意事项
在使用C++实现装饰器模式时,需要注意以下几点:
- 抽象组件接口要设计得足够通用,避免后续装饰器需要修改接口导致所有实现类都要调整
- 装饰器的嵌套层数不宜过多,否则会导致调用链过长,不利于问题排查
- 如果装饰器需要修改原始对象的状态,要注意多线程场景下的线程安全问题
- 内存管理要统一,建议由最外层的装饰器负责释放内部所有组件的内存,避免内存泄漏
适用场景
装饰器模式适合以下场景:
- 需要在运行时动态给对象添加功能,且功能可以灵活组合、撤销
- 用继承扩展功能会导致类数量爆炸,比如一个类有5个可选功能,继承需要生成2^5个子类
- 需要给一个对象添加的功能不影响其他同类对象,不需要修改原有类的代码