组合设计模式的核心目标是将对象组合成树形结构以表示部分整体的层次结构，让客户端对单个对象和组合对象的使用具有一致性，非常适合处理需要统一操作树形结构中不同类型节点的场景。
组合设计模式的核心角色
在C++中实现组合设计模式，通常需要定义三个核心角色，每个角色的职责明确，共同支撑起树形结构的统一操作能力。
- 抽象组件（Component）：定义树形结构中所有对象的通用接口，既可以代表叶子节点，也可以代表组合节点，声明公共的操作方法。
- 叶子节点（Leaf）：继承抽象组件，代表树形结构中的最底层节点，没有子节点，实现抽象组件中定义的叶子节点专属操作。
- 组合节点（Composite）：同样继承抽象组件，代表树形结构中的容器节点，包含子组件集合，既可以包含叶子节点，也可以包含其他组合节点，实现添加、删除子组件以及遍历子组件的操作。
基础实现示例：文件目录系统
以常见的文件目录系统为例，文件是叶子节点，文件夹是组合节点，我们需要让客户端可以统一调用显示名称、获取大小的方法，不需要区分当前操作的是文件还是文件夹。
1. 定义抽象组件
抽象组件声明所有节点共有的方法，这里定义显示名称和获取大小两个通用接口，同时声明虚析构函数保证多态删除的正确性。
#include <iostream>
#include <string>
#include <vector>
#include <memory>
// 抽象组件类
class FileSystemComponent {
public:
FileSystemComponent(std::string name) : name_(name) {}
virtual ~FileSystemComponent() = default;
// 显示组件名称
virtual void display() = 0;
// 获取组件大小
virtual int getSize() = 0;
// 获取组件名称
std::string getName() const {
return name_;
}
protected:
std::string name_;
};
2. 实现叶子节点：文件类
文件类继承抽象组件，实现显示和获取大小的方法，文件没有子节点，不需要处理子组件相关的逻辑。
// 叶子节点：文件类
class File : public FileSystemComponent {
public:
File(std::string name, int size) : FileSystemComponent(name), size_(size) {}
void display() override {
std::cout << "文件: " << name_ << ", 大小: " << size_ << "KB" << std::endl;
}
int getSize() override {
return size_;
}
private:
int size_;
};
3. 实现组合节点：文件夹类
文件夹类继承抽象组件，内部维护子组件的集合，实现添加、删除子组件的方法，同时重写显示和获取大小的方法，遍历所有子组件完成对应操作。
// 组合节点：文件夹类
class Folder : public FileSystemComponent {
public:
Folder(std::string name) : FileSystemComponent(name) {}
void display() override {
std::cout << "文件夹: " << name_ << ", 包含子项:" << std::endl;
for (const auto& component : children_) {
component->display();
}
}
int getSize() override {
int totalSize = 0;
for (const auto& component : children_) {
totalSize += component->getSize();
}
return totalSize;
}
// 添加子组件
void addChild(std::shared_ptr<FileSystemComponent> component) {
children_.push_back(component);
}
// 移除子组件
void removeChild(std::shared_ptr<FileSystemComponent> component) {
for (auto it = children_.begin(); it != children_.end(); ++it) {
if (*it == component) {
children_.erase(it);
break;
}
}
}
private:
std::vector<std::shared_ptr<FileSystemComponent>> children_;
};
4. 客户端调用示例
客户端只需要持有抽象组件的指针，就可以统一操作文件和文件夹，不需要判断当前对象的类型，符合开闭原则。
int main() {
// 创建文件
auto file1 = std::make_shared<File>("文档.txt", 120);
auto file2 = std::make_shared<File>("图片.jpg", 500);
auto file3 = std::make_shared<File>("视频.mp4", 2000);
// 创建子文件夹
auto subFolder = std::make_shared<Folder>("工作资料");
subFolder->addChild(file1);
subFolder->addChild(file2);
// 创建根文件夹
auto rootFolder = std::make_shared<Folder>("根目录");
rootFolder->addChild(subFolder);
rootFolder->addChild(file3);
// 统一调用显示方法
std::cout << "目录结构:" << std::endl;
rootFolder->display();
// 统一调用获取大小方法
std::cout << "根目录总大小: " << rootFolder->getSize() << "KB" << std::endl;
return 0;
}
实现注意事项
在C++中实现组合设计模式时，有几个细节需要注意，避免后续使用出现问题。
- 抽象组件的析构函数需要声明为虚函数，否则通过基类指针删除派生类对象时会导致内存泄漏。
- 子组件的集合建议使用智能指针管理，避免手动管理内存带来的野指针、内存泄漏问题，示例中使用了
std::shared_ptr。 - 如果某些操作只有组合节点需要，比如添加、删除子组件，有两种处理方式：一种是在抽象组件中声明这些方法，叶子节点实现时抛出异常；另一种是将这些方法只放在组合节点中，客户端需要向下转型才能调用，前者更符合统一接口的初衷，后者更安全，需要根据实际场景选择。
- 遍历组合节点时如果需要控制遍历顺序，可以在组合节点中增加遍历策略的配置，比如前序遍历、后序遍历等。
适用场景
组合设计模式适合以下场景使用：
- 需要表示对象的部分整体层次结构，比如组织架构、菜单系统、UI组件树等。
- 希望客户端忽略组合对象与单个对象的差异，统一使用所有对象，简化客户端逻辑。
- 树形结构中的节点类型可能会扩展，需要保证新增节点类型时客户端代码不需要大量修改。