组合模式的Java实现

组合模式简述

• 组合模式（Composite Pattern），又叫部分整体模式，是用于把一组相似的对象当作一个单一的对象，从而使得某些操作具有一致性，在客户端角度来看，不需要针对部分与整体的特殊性进行分类处理，这种操作的一致性是通过组合模式在内部进行实现的。
• 这里主要解决的问题是类似于树形结构的问题，在树形结构中，叶子结点就是部分，整体就是非叶子结点，整体中包含部分，可以理解为非叶子节点还会继续往下链接，直到链接到叶子结点。
• 组合模式依据树形结构来组合对象，用来表示部分以及整体层次。
• 这种类型的设计模式属于结构型模式，它创建了对象组的树形结构。
• 这种模式创建了一个包含自己对象组的类，该类提供了修改相同对象组的方式。

介绍

意图：将对象组合成树形结构以表示"部分-整体"的层次结构，组合模式使得用户对单个对象和组合对象的使用具有一致性。

主要解决：它在我们树型结构的问题中，模糊了简单元素和复杂元素的概念，客户程序可以像处理简单元素一样来处理复杂元素，从而使得客户程序与复杂元素的内部结构解耦。

何时使用： 

• 想表示对象的部分-整体层次结构（树形结构）。
• 希望用户忽略组合对象与单个对象的不同，用户将统一地使用组合结构中的所有对象。

组合模式核心：

• 抽象构建角色（component）：定义了叶子和容器构建的共同点
• 叶子构建角色（leaf）：无子节点
• 容器构建角色（composite）：有容器特征，可以包含子节点
• 树枝（容器）和叶子实现统一接口，树枝内部组合该接口

关键代码：树枝内部组合该接口，并且含有内部属性 List，里面放 Leaf，展示如下

package com.ly.composite;

/**
 * liyang 2020-07-20
 * 组合模式的组织形式，仅用于展示
 * 实际使用是叶子组件与容器组件去实现Component
 * 具体见下文的案例
 */

public interface Component {
    void operation();
}

//叶子组件
interface Leaf extends Component {

}

//容器组件
interface Composite extends Component {
    //List(Leaf)
    void add(Component c);
    void remove(Component c);
    Component getChild(int index);
}

组合模式工作流程分析：

• 组合模式为处理树形结构提供了完美的解决方案，描述了如何将容器和叶子进行递归组合，使得用户在使用时可以一致的对待容器和叶子。
• 当容器方法被调用时，经遍历整个树形结构，寻找也包含这个方法的成员，并调用执行。其中，使用了递归调用的机制对整个结构进行处理。

优点：

• 高层模块调用简单
• 节点自由增加。

缺点：在使用组合模式时，其叶子和树枝的声明都是实现类，而不是接口，违反了依赖倒置原则。

使用场景：部分、整体场景，如树形菜单，文件、文件夹的管理。

应用实例：

• 算术表达式包括操作数、操作符和另一个操作数，其中，另一个操作符也可以是操作数、操作符和另一个操作数。
• 在 JAVA AWT 和 SWING 中，对于 Button 和 Checkbox 是树叶，Container 是树枝。
• 操作系统的资源管理器
• GUI中的容器层次图
• XML、HTML文件解析
• OA系统中，组织结构的处理
• Junit单元测试框架：底层设计就是典型的组合模式，TestCase（叶子）、TestUnite（容器）、Test接口（抽象）

注意：别将组合模式与继承中的组合混淆

实现

使用组合模式模拟杀毒软件架构设计
步骤1、2：创建部分-整体需要实现的统一接口，然后创建具体的叶子组件（部分）和容器组件（整体）

package com.ly.composite;

import java.util.ArrayList;
import java.util.List;

/**
 * 使用组合模式模拟杀毒软件架构设计
 * liyang 2020-07-20
 */

public interface AbstractFile {
    void killVirus();  //杀毒
}

//叶子构件1-图片文件
class ImageFile implements AbstractFile {
    private String name;
    public ImageFile(String name) {
        this.name = name;
    }
    @Override
    public void killVirus() {
        System.out.println("图片文件杀毒：" + name + "查杀中...");
    }
}

//叶子构件2-文本文件
class TextFile implements AbstractFile {
    private String name;
    public TextFile(String name) {
        this.name = name;
    }
    @Override
    public void killVirus() {
        System.out.println("文本文件杀毒：" + name + "查杀中...");
    }
}

//叶子构件3-视频文件
class VideoFile implements AbstractFile {
    private String name;
    public VideoFile(String name) {
        this.name = name;
    }
    @Override
    public void killVirus() {
        System.out.println("视频文件杀毒：" + name + "查杀中...");
    }
}

//容器构件-文件夹
class Folder implements AbstractFile {
    private String name;
    //容器下的子节点，可能是一个容器构件，也可能是一个叶子构件
    private List<AbstractFile> list = new ArrayList<>();

    public Folder(String name) {
        this.name = name;
    }

    @Override
    public void killVirus() {
        System.out.println("文件夹杀毒：" + name + "查杀中...");
        for(AbstractFile file : list ) {
            file.killVirus();
        }
    }

    public void add(AbstractFile file) {
        list.add(file);
    }

    public void remove(AbstractFile file) {
        list.remove(file);
    }

    public AbstractFile getChild(int index) {
        if(index < list.size()) return list.get(index);
        return null;
    }
}

步骤3：客户端测试

package com.ly.composite;

/**
 * liyang 2020-07-20
 * 组合模式的客户端测试
 * 模式杀毒过程，组合模式使得对文件和文件夹的删除方式一致（从客户端角度来看，没有区别）
 */

public class Client {
    public static void main(String[] args) {
        //*************************************************************************************
        //AbstractFile中没有add方法，只有killVirus方法，因此按如下方法声明定义，需要在使用add方法时进行强转
        //AbstractFile file01 = new Folder("我的收藏");
        //((Folder) file01).add(file02);
        //*************************************************************************************

        Folder file01 = new Folder("我的收藏");
        Folder file02 = new Folder("Java文件夹");
        Folder file03 = new Folder("C++文件夹");
        file01.add(file02);
        file01.add(file03);
        //file01.killVirus();

        AbstractFile file04 = new VideoFile("Java视频");
        AbstractFile file05 = new VideoFile("C++视频");
        file02.add(file04);
        file03.add(file05);
        //file04.killVirus();

        AbstractFile file06 = new ImageFile("Java垃圾回收机制");
        AbstractFile file07 = new ImageFile("C++指针原理");
        file02.add(file06);
        file03.add(file07);

        AbstractFile file08 = new TextFile("Core Java Volume I: Fundamentals.pdf");
        AbstractFile file09 = new TextFile("C++ Primer.pdf");
        file02.add(file08);
        file03.add(file09);

        file01.killVirus();
    }
}

结果：

文件夹杀毒：我的收藏查杀中...
文件夹杀毒：Java文件夹查杀中...
视频文件杀毒：Java视频查杀中...
图片文件杀毒：Java垃圾回收机制查杀中...
文本文件杀毒：Core Java Volume I: Fundamentals.pdf查杀中...
文件夹杀毒：C++文件夹查杀中...
视频文件杀毒：C++视频查杀中...
图片文件杀毒：C++指针原理查杀中...
文本文件杀毒：C++ Primer.pdf查杀中...

Process finished with exit code 0

参考资料：

https://www.runoob.com/design-pattern/composite-pattern.html