组合模式

组合模式（Composite Pattern）将对象组合成树形结构以表示“部分-整体”的层次结构。组合模式使得用户可以使用一致的方法操作单个对象和组合对象。

假设一个场景：一个杀毒软件叫ABKing，我们模拟它的杀毒逻辑。首先需要选定一个文件夹吧，然后这个文件夹里的每一个文件都需要被扫描。在Linux系统中，文件夹也被视为文件，只不过是特殊的文件而已，相应地，文件和文件夹其实有共同之处。

所以我们写一个接口：

package top.bigking.composite;

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

/**
 * @Author ABKing
 * @since 2020/2/18 下午9:00
 **/
public interface AbstractFile {
    void killVirus();
}

两个实现类（文件和文件夹）：

package top.bigking.composite;

/**
 * @Author ABKing
 * @since 2020/2/18 下午9:20
 **/
public class MyFile implements AbstractFile{
    private String name;

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

    public String getName() {
        return name;
    }

    @Override
    public void killVirus() {
        System.out.println("ABKing杀毒软件已成功清除 " + name + " 病毒文件！");
    }
}

package top.bigking.composite;

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

/**
 * @Author ABKing
 * @since 2020/2/18 下午9:21
 **/
public class MyFolder implements AbstractFile{
    private String name;
    private List<AbstractFile> list = new ArrayList<>();

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

    @Override
    public void killVirus() {
        System.out.println("开始清除文件夹：" + name);
        for (AbstractFile file: list) {
            file.killVirus();
        }
        System.out.println("清除文件夹 " + name + " 成功！");
    }
    public void add(AbstractFile file){
        list.add(file);
    }
    public void remove(AbstractFile file){
        list.remove(file);
    }
    public AbstractFile getChild(int index){
        return list.get(index);
    }
}

JUnit单元测试：

package top.bigking.composite;

import org.junit.Test;

/**
 * @Author ABKing
 * @since 2020/2/18 下午9:21
 **/
public class TestComposite {
    @Test
    public void testComposite(){
        MyFolder folder = new MyFolder("主文件夹");
        MyFolder folder1 = new MyFolder("毒文件夹");
        AbstractFile f1 = new MyFile("文件1");
        AbstractFile f2 = new MyFile("文件2");
        AbstractFile f3 = new MyFile("文件3");
        AbstractFile f4 = new MyFile("文件4");
        AbstractFile f5 = new MyFile("文件5");

        folder.add(f1);
        folder.add(f2);
        folder1.add(f3);
        folder1.add(f4);
        folder.add(folder1);
        folder.add(f5);

        folder.killVirus();
    }
}

运行结果：

开始清除文件夹：主文件夹
ABKing杀毒软件已成功清除 文件1 病毒文件！
ABKing杀毒软件已成功清除 文件2 病毒文件！
开始清除文件夹：毒文件夹
ABKing杀毒软件已成功清除 文件3 病毒文件！
ABKing杀毒软件已成功清除 文件4 病毒文件！
清除文件夹 毒文件夹 成功！
ABKing杀毒软件已成功清除 文件5 病毒文件！
清除文件夹 主文件夹 成功！

开发中的应用场景：

• 操作系统的资源管理器
• GUI中的容器层次图
• XML文件解析
• OA系统中，组织架构的处理
• JUnit单元测试框架：底层设计就是典型的组合模式，TestCase（叶子）、TestUnite（容器）、Test接口（抽象）