结构型设计模式-组合（composite）模式-CSDN博客

本文链接：https://blog.csdn.net/linkedin_21843693/article/details/141431975

设计模式汇总：查看

通俗示例

想象一下，你正在管理一家公司的组织结构。这个公司有多个部门，每个部门下面又可能有多个子部门，同时还有不同级别的员工。在管理这个组织结构时，你希望能够统一地处理单个员工和整个部门，比如进行薪资计算或者通知发布。这里，组织结构就是一个树形结构，而组合模式允许你以统一的方式对待单个员工和整个部门。

通俗解释

组合模式是一种结构型设计模式，用于创建树形结构的对象模型，使得客户端可以统一地使用单个对象和组合对象。在组合模式中，我们将对象组织成树形结构，使得客户在操作单个对象和组合对象时，可以使用相同的接口。

在组合模式中，通常涉及以下角色：

组件（Component）：为所有对象定义统一的接口，可以是一个抽象类或接口，定义了叶节点和组合节点共有的操作。
叶节点（Leaf）：树形结构中的最小单位，没有子节点。
组合节点（Composite）：包含叶节点或其他组合节点的容器，定义了添加、移除和获取子节点的方法。

组合模式的优点

统一接口：客户端可以一致地处理单个对象和组合对象。
增加灵活性：可以动态地增加或删除树的节点，而不影响客户端代码。
简化客户端代码：客户端不必区分单个对象和组合对象，简化了代码的复杂性。

Python代码示例

以下是一个组合模式的简单实现，以公司组织结构为例：

# 组件
class Employee:
    def __init__(self, name, title):
        self.name = name
        self.title = title
    
    def get_details(self):
        return f"{self.title} {self.name}"

# 叶节点
class Developer(Employee):
    pass

# 组合节点
class CompositeEmployee(Employee):
    def __init__(self, name, title):
        super().__init__(name, title)
        self.subordinates = []
    
    def add_subordinate(self, employee):
        self.subordinates.append(employee)
    
    def remove_subordinate(self, employee):
        self.subordinates.remove(employee)
    
    def get_details(self):
        details = f"{self.title} {self.name}\n"
        for subordinate in self.subordinates:
            details += f"  {subordinate.get_details()}\n"
        return details

# 客户端代码
if __name__ == "__main__":
    # 创建员工
    john = Developer("John Doe", "Software Developer")
    jane = Developer("Jane Smith", "QA Engineer")
    
    # 创建部门
    dev_department = CompositeEmployee("Development", "Department")
    dev_department.add_subordinate(john)
    dev_department.add_subordinate(jane)
    
    # 打印组织结构
    print(dev_department.get_details())

在这个例子中，Employee是组件，Developer是叶节点，CompositeEmployee是组合节点。客户端代码创建了一个开发部门dev_department，并添加了两个员工john和jane作为下属。get_details方法会递归地打印出所有的下属信息，无论是单个员工还是整个部门。