JAVA设计模式之组合模式

组合模式使用的场景
- 把部分和整体的关系用树形结构来表示，从而使客户端可以使用统一的方式处理部分对象和整体对象
- 组合模式的核心：
- 抽象构建（Component）角色：定义了叶子和容器构件的共同特点
- 叶子构建角色：无子节点
- 容器（Composite）构建角色：有容器特征，可以包含节点

package com.imust.composite;

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

//抽象构建
public interface AbstractFile {

    void killVirus();//杀毒
}

class ImageFile implements AbstractFile{

    private String name;
    public ImageFile(String name){
        this.name = name;
    }
    @Override
    public void killVirus() {
        System.out.println("--图像文件"+name+"进行查杀");
    }

}

class TextFile implements AbstractFile{

    private String name;
    public TextFile(String name){
        this.name = name;
    }
    @Override
    public void killVirus() {
        System.out.println("--文本文件"+name+"进行查杀");
    }
}

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<AbstractFile>();

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

    }

}

测试类

package com.imust.composite;

public class Client {

    public static void main(String[] args) {

        AbstractFile f2,f3,f4,f5,f6;

        Folder f1 = new Folder("我的收藏");


        f2 = new ImageFile("老高的头像");
        f3 = new TextFile("hello.txt");
        f1.add(f2);
        f1.add(f3);

        f2.killVirus();
        f1.killVirus();


    }

}

原型

package com.imust.composite;

/**
 * 抽象组件
 * @author  Zachary (为上帝所心仪的人)
 * @date    2017年9月7日
 * @project shejimoshi
 * @package com.imust.composite
 */
public interface Component {

    void operation();
}
//叶子组件
interface Leaf extends Component{

}
//容器组件
interface Composite extends Component{

    void add(Component c);
    void remove(Component c);
    Component getChild(int index);
}