设计模式--组合模式

package com.offcn.designpattern.compositepattern;

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

public class CompositepatternDemo {
    public static void main(String[] args) {
        AbstractFile a1,a2,a3,a4;
        Folder f1 = new Folder("我的收藏");
        a1 = new ImgFile("美女图片.jpg");
        a2 = new VidelFile("我的野蛮女友.avi");
        f1.add(a1);
        f1.add(a2);
        f1.killVirus();
        System.out.println("**************************");
        Folder f2 = new Folder("视频集锦");
        a3 = new VidelFile("刘老根.avi");
        a4 = new VidelFile("神雕侠侣.avi");
        f2.add(a3);
        f2.add(a4);
        f1.add(f2);
        f1.killVirus();

    }
}
//抽象构建（Component）角色
interface AbstractFile{
    void killVirus();//杀毒
}
//叶子（leaf）构件角色
class ImgFile implements AbstractFile{
    private String name;

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

    @Override
    public void killVirus() {
        System.out.println("---图像文件"+name+",杀毒中。。");
    }
}
//叶子（leaf）构件角色
class VidelFile implements AbstractFile{
    private String name;

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

    @Override
    public void killVirus() {
        System.out.println("---视频文件"+name+",杀毒中。。");
    }
}
//文件夹
class Folder implements AbstractFile{
    private String name;
    //容器（Composite）构件角色,可以存放叶子节点
    private List<AbstractFile> list = new ArrayList<AbstractFile>();

    public Folder(String name) {
        this.name = name;
    }
    public void add(AbstractFile abstractFile){
        list.add(abstractFile);
    }
    public void remove(AbstractFile abstractFile){
        list.remove(abstractFile);
    }
    public AbstractFile getChild(int index){
        return list.get(index);
    }
    @Override
    public void killVirus() {
        System.out.println("---文件夹"+name+",杀毒中。。");
        for (AbstractFile file : list) {
            file.killVirus();
        }
    }
}