组合模式的核心是将对象组合成树形结构以表示部分整体的层次结构，让客户端对单个对象和组合对象的使用具有一致性，在C++中实现该模式需要合理设计抽象基类和派生类。
组合模式的核心角色
在C++中实现组合模式，通常需要定义三个核心角色：
- 抽象组件（Component）：定义所有对象共有的接口，既可以代表叶子节点，也可以代表组合节点，声明公共的操作方法。
- 叶子节点（Leaf）：继承抽象组件，表示树形结构中的最底层节点，没有子节点，实现抽象组件定义的接口。
- 组合节点（Composite）：继承抽象组件，包含子组件的集合，既可以包含叶子节点，也可以包含其他组合节点，实现管理子组件的方法。
抽象组件定义
首先定义抽象组件类，声明公共的操作接口，这里以文件系统为例，定义文件和文件夹的共有操作：
#include <iostream>
#include <string>
#include <vector>
#include <memory>
// 抽象组件类
class FileSystemComponent {
public:
virtual ~FileSystemComponent() = default;
// 显示组件信息
virtual void display(int depth) = 0;
// 添加子组件，默认实现抛出异常，叶子节点不支持该操作
virtual void add(std::shared_ptr<FileSystemComponent> component) {
throw std::runtime_error("不支持添加子组件");
}
// 移除子组件，默认实现抛出异常
virtual void remove(std::shared_ptr<FileSystemComponent> component) {
throw std::runtime_error("不支持移除子组件");
}
protected:
std::string name;
FileSystemComponent(const std::string& componentName) : name(componentName) {}
};
叶子节点实现
叶子节点对应文件，没有子节点，只实现显示自身信息的方法：
// 叶子节点：文件
class File : public FileSystemComponent {
public:
File(const std::string& fileName) : FileSystemComponent(fileName) {}
void display(int depth) override {
// 根据层级添加缩进
for (int i = 0; i < depth; ++i) {
std::cout << " ";
}
std::cout << "文件: " << name << std::endl;
}
};
组合节点实现
组合节点对应文件夹，需要维护子组件的集合，实现添加、移除和显示子组件的方法：
// 组合节点：文件夹
class Folder : public FileSystemComponent {
public:
Folder(const std::string& folderName) : FileSystemComponent(folderName) {}
void display(int depth) override {
// 显示当前文件夹
for (int i = 0; i < depth; ++i) {
std::cout << " ";
}
std::cout << "文件夹: " << name << std::endl;
// 递归显示所有子组件
for (auto& component : children) {
component->display(depth + 1);
}
}
void add(std::shared_ptr<FileSystemComponent> component) override {
children.push_back(component);
}
void remove(std::shared_ptr<FileSystemComponent> component) override {
children.erase(std::remove(children.begin(), children.end(), component), children.end());
}
private:
std::vector<std::shared_ptr<FileSystemComponent>> children;
};
客户端使用示例
在客户端中，我们可以自由组合文件夹和文件，构建树形结构，并且统一调用display方法：
int main() {
try {
// 创建根文件夹
auto rootFolder = std::make_shared<Folder>("根目录");
// 创建根目录下的文件
auto file1 = std::make_shared<File>("说明文档.txt");
auto file2 = std::make_shared<File>("配置信息.ini");
rootFolder->add(file1);
rootFolder->add(file2);
// 创建子文件夹
auto subFolder = std::make_shared<Folder>("图片资源");
auto imgFile1 = std::make_shared<File>("背景图.png");
auto imgFile2 = std::make_shared<File>("图标.jpg");
subFolder->add(imgFile1);
subFolder->add(imgFile2);
// 将子文件夹添加到根目录
rootFolder->add(subFolder);
// 统一显示整个树形结构
std::cout << "文件系统结构：" << std::endl;
rootFolder->display(0);
// 尝试给文件添加子组件，会抛出异常
file1->add(subFolder);
} catch (const std::exception& e) {
std::cout << "操作异常: " << e.what() << std::endl;
}
return 0;
}
组合模式的优缺点
使用组合模式实现树形结构的优势很明显：
- 客户端不需要区分叶子节点和组合节点，统一使用抽象组件的接口，降低了使用复杂度。
- 新增叶子节点或组合节点的类型时，不需要修改现有代码，符合开闭原则，扩展性好。
- 可以很方便地构建任意层级的树形结构，适合处理有层级关系的场景。
当然也存在一定的不足：
- 抽象组件中声明了所有子类的方法，叶子节点需要实现一些自己不支持的方法，比如示例中的
add和remove，只能通过抛异常的方式处理，不够优雅。 - 如果树形结构层级过深，递归操作可能会导致栈溢出问题，需要合理控制层级深度。
适用场景
当你的C++项目中存在以下场景时，可以考虑使用组合模式：
- 需要表示对象的部分整体层次结构，比如菜单、组织架构、文件目录等。
- 希望客户端可以忽略组合对象和单个对象的差异，统一使用相同的接口操作。
- 树形结构中的节点类型相对固定，不需要频繁新增特殊类型的节点。