设计模式-结构型-组合模式

1. 什么是组合模式？

组合模式（Composite Pattern） 是一种结构型设计模式，它允许将对象组合成树形结构来表示“部分-整体”的层次结构。组合模式使得客户端对单个对象和组合对象的使用具有一致性。换句话说，组合模式允许你将多个对象组合成一个复合对象，然后统一处理这些对象。

核心思想

组合模式通过将简单对象（叶子节点）和复合对象（包含子对象的树形结构）统一视为相同的类型，从而简化客户端对不同类型对象的处理。它常用于表示树形结构，如目录结构、组织架构等。

2. 组合模式的结构

UML 类图

以下是组合模式的 UML 类图：

角色

Component（抽象组件）
定义了所有对象（无论是叶子节点还是组合节点）共用的方法，如 operation()。可以是抽象类或接口。
Leaf（叶子节点）
叶子节点是组合树结构的最底层元素，不能再包含子对象。它实现了 operation() 方法，通常用于具体的操作。
Composite（组合节点）
组合节点包含叶子节点或其他组合节点。它也实现了 operation() 方法，并允许对子节点进行操作（如 add()、remove() 等）。通常用来组织和管理子组件。

3. 组合模式的示例

场景描述

假设我们需要构建一个文件系统的树形结构，其中包含文件夹（可以包含文件和子文件夹）和文件。文件和文件夹都有一个 display() 方法。组合模式可以帮助我们统一处理文件和文件夹。

代码实现

from abc import ABC, abstractmethod# 抽象组件类
class FileSystemComponent(ABC):@abstractmethoddef display(self):pass# 叶子节点：文件类
class File(FileSystemComponent):def __init__(self, name):self.name = namedef display(self):print(f"File: {self.name}")# 组合节点：文件夹类
class Folder(FileSystemComponent):def __init__(self, name):self.name = nameself.children = []def add(self, component: FileSystemComponent):self.children.append(component)def remove(self, component: FileSystemComponent):self.children.remove(component)def display(self):print(f"Folder: {self.name}")for child in self.children:child.display()# 测试组合模式
if __name__ == "__main__":# 创建文件和文件夹file1 = File("file1.txt")file2 = File("file2.txt")file3 = File("file3.txt")folder1 = Folder("Folder1")folder2 = Folder("Folder2")folder1.add(file1)folder1.add(file2)folder2.add(file3)# 创建根文件夹，添加子文件夹root = Folder("Root")root.add(folder1)root.add(folder2)# 显示整个文件系统结构root.display()

输出

Folder: Root

Folder: Folder1

File: file1.txt

File: file2.txt

Folder: Folder2

File: file3.txt

在上面的代码中，FileSystemComponent 是抽象组件类，File 是叶子节点，Folder 是组合节点。Folder 具有 add() 和 remove() 方法来管理其子节点，而 File 只有 display() 方法。最后，通过调用根文件夹的 display() 方法，能够递归地显示整个文件系统的结构。

4. 组合模式的优缺点

优点

简化客户端代码
客户端不需要关心对象是单个对象还是组合对象，它们可以通过相同的接口进行处理，简化了代码结构。
递归结构清晰
组合模式特别适合处理递归结构的数据，比如文件系统、组织结构等。
增加或删除元素方便
通过组合模式，添加或删除树形结构中的节点（文件或文件夹）非常方便，不影响其它节点的操作。
高扩展性
可以轻松扩展新的叶子节点或组合节点，而不需要修改现有的客户端代码。

缺点

复杂性增加
如果对象结构本身并不复杂，引入组合模式可能会让系统变得过于复杂，导致不必要的开销。
不易实现对叶子节点的具体行为
在某些情况下，叶子节点的行为可能与组合节点有所不同，这可能会导致设计上的矛盾，难以通过统一接口来处理。

5. 组合模式的应用场景

文件系统
文件夹可以包含子文件夹和文件，文件和文件夹是不同的对象类型，但是它们都实现了一个公共接口 display()，可以统一处理。
GUI 组件库
许多图形用户界面组件（如按钮、窗体、标签等）可以通过组合模式来处理。组件（如按钮）可以是叶子节点，而容器（如面板、窗体）可以是组合节点。
组织架构
企业的组织架构可以通过组合模式表示。一个部门可以包含多个员工或者子部门，所有部门和员工都可以通过统一的接口进行管理。
树形结构的表示
任何需要树形结构表示的场景，例如目录树、家族谱、层次化数据展示等，都适合使用组合模式。