装饰是一种结构型设计模式,允许你通过将对象放入包含行为的特殊封装对象中来为原对象绑定新的行为
*装饰器模式会增加许多子类,过度使用会增加程序得复杂性
Component.h:
#ifndef COMPONENT_H_
#define COMPONENT_H_
#include
// 部件: 是具体部件和装饰类的共同基类, 在C++中实现成抽象基类
class DataSource {
public:
virtual void writeData(std::string data) = 0;
};
#endif // COMPONENT_H_
BaseDecorator.h:
#ifndef BASE_DECORATOR_H_
#define BASE_DECORATOR_H_
#include
#include "Component.h"
// 装饰基类和其他组件遵循相同的接口。该类的主要任务是定义所有具体装饰的封装接口。
// 封装的默认实现代码中可能会包含一个保存被封装组件的成员变量,并且负责对其进行初始化。
class DataSourceDecorator : public DataSource {
public:
explicit DataSourceDecorator(DataSource* ds) : data_source_(ds) {}
void writeData(std::string data) override {
data_source_->writeData(data);
}
protected:
DataSource* data_source_; // component
};
#endif // BASE_DECORATOR_H_
ConcreteComponent.h:
#ifndef CONCRETE_COMPONENT_H_
#define CONCRETE_COMPONENT_H_
#include
#include
#include
#include "Component.h"
// 具体组件提供操作的默认实现, 这些类在程序中可能会有几个变体
class FileDataSource : public DataSource {
public:
explicit FileDataSource(std::string file_name) : file_name_(file_name) {}
void writeData(std::string data) override {
printf("写入文件%s中: %s\n", file_name_.c_str(), data.c_str());
return;
}
private:
std::string file_name_;
};
#endif // CONCRETE_COMPONENT_H_
ConcreteDecorator.h:
#ifndef CONCRETE_DECORATOR_H_
#define CONCRETE_DECORATOR_H_
#include
#include "BaseDecorator.h"
// 加密装饰器
class EncryptionDecorator : public DataSourceDecorator {
public:
using DataSourceDecorator::DataSourceDecorator;
void writeData(std::string data) override {
// 1. 对传递数据进行加密(这里仅简单实现)
data = "已加密(" + data + ")";
// 2. 将加密后数据传递给被封装对象 writeData(写入数据)方法
data_source_->writeData(data);
return;
}
};
// 压缩装饰器
class CompressionDecorator : public DataSourceDecorator {
public:
using DataSourceDecorator::DataSourceDecorator;
void writeData(std::string data) override {
// 1. 对传递数据进行压缩(这里仅简单实现)
data = "已压缩(" + data + ")";
// 2. 将压缩后数据传递给被封装对象 writeData(写入数据)方法
data_source_->writeData(data);
return;
}
};
#endif // CONCRETE_DECORATOR_H_
main.cpp:
#include "ConcreteComponent.h"
#include "ConcreteDecorator.h"
int main() {
system("chcp 65001");
FileDataSource* source1 = new FileDataSource("stdout");
// 将明码数据写入目标文件
source1->writeData("tomocat");
// 将压缩数据写入目标文件
CompressionDecorator* source2 = new CompressionDecorator(source1);
source2->writeData("tomocat");
// 将压缩且加密数据写入目标文件
EncryptionDecorator* source3 = new EncryptionDecorator(source2);
source3->writeData("tomocat");
delete source1;
delete source2;
delete source3;
system("pause");
}