23种设计模式之组合模式

应用场景：

• 把部分和整体的关系用到树形结构来表示，从而使客户端可以使用统一的方式处理部分对象和整体对象。

组合模式核心：

• 1）抽象构件（component）角色：定义了叶子和容器构件的共同点。

• 2）叶子（leaf）构件角色：无子节点

• 3）容器（composite）构件角色：有容器特征，可以包含子节点。

//抽象组件角色
public interface Component {
    void operation();
}

//叶子构件角色
interface Leaf extends Component {

}

//容器构件角色
interface Composite extends Component {

    void add(Component component);

    void remove(Component component);

    Component getChild(int index);
}

使用组合模式，模拟杀毒软件的架构设计！

• 首先抽象构建搭建

//抽象构建
public interface AbstractFile {
    void killVirus();//杀毒
}

• 图片叶子构建

//图片叶子组件
public class ImageFile implements AbstractFile {
    private String name;

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

    @Override
    public void killVirus() {
        System.out.println("---图像文件：" + name + ",进行查杀！");
    }
}

• 文本叶子构建

//文本叶子节点
public class TextFile implements AbstractFile {
    private String name;

    public TextFile(String name) {
        super();
        this.name = name;
    }

    @Override
    public void killVirus() {
        System.out.println("---文本文件：" + name + ",进行查杀！");
    }
}

• 视频叶子节点

//视频文件叶子节点
public class VideoFile implements AbstractFile {
    private String name;

    public VideoFile(String name) {
        super();
        this.name = name;
    }

    @Override
    public void killVirus() {
        System.out.println("---视频文件：" + name + ",进行查杀！");
    }

}

• 容器构建搭建

//容器构件
public class Folder implements AbstractFile {
    private String name;
    //定义容器，用来存放本容器构建下的节点
    private List<AbstractFile> list = new ArrayList<>();

    public Folder(String name) {
        this.name = 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);
    }

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

测试

 AbstractFile f2, f3, f4,f5;
        Folder f1 = new Folder("我的收藏");
        f2 = new ImageFile("美女.jpg");
        f3 = new TextFile("城市.txt");
        f1.add(f2);
        f1.add(f3);
        Folder folder=new Folder("电影");
        f4=new VideoFile("战狼2.mp4");
        f5=new VideoFile("新战神.mp4");
        folder.add(f4);
        folder.add(f5);
        f1.add(folder);
        f1.killVirus();