package com.disign.composite;public interface AbstractFile {void killVirus();
}/*** 叶子节点角色*/
class ImageFile implements AbstractFile {private String name;public ImageFile(String name) {this.name = name;}@Overridepublic void killVirus() {System.out.println("kill  " + name + " virus ");}
}class VideoFile implements AbstractFile {private String name;public VideoFile(String name) {this.name = name;}@Overridepublic void killVirus() {System.out.println("kill  " + name + " virus ");}
}class TextFile implements AbstractFile {private String name;public TextFile(String name) {this.name = name;}@Overridepublic void killVirus() {System.out.println("kill  " + name + " virus ");}
}/*** 在组合模式中代表 容器*/
public class Folder implements  AbstractFile {private String name;public Folder(String name) {this.name = name;}public String getName() {return name;}private LinkedList<AbstractFile> list = new LinkedList<>();public void add(AbstractFile file){list.add(file);}public void remove(AbstractFile file){list.remove(file);}public AbstractFile get(int index ){return list.get(index);}@Overridepublic void killVirus() {System.out.println("kill "+name +" folder");for (AbstractFile abstractFile : list) {abstractFile.killVirus();}}
}public class Main {public static void main(String[] args) {AbstractFile f1,f2,f3,f4,f5;Folder file = new Folder("我的收藏");f1 = new ImageFile("不健康的图片.jpg");f2 = new TextFile("不健康的文本");file.add(f1);file.add(f2);Folder folder = new Folder("电影");f3 = new VideoFile("3d肉蒲团");f4 = new VideoFile("一路向西");f5 = new VideoFile("公司职工");folder.add(f3);folder.add(f4);folder.add(f5);file.add(folder);file.remove(f2);Folder file1 = (Folder)file.get(1);file.killVirus();}
}
kill 我的收藏 folder
kill  不健康的图片.jpg virus
kill 电影 folder
kill  3d肉蒲团 virus
kill  一路向西 virus
kill  公司职工 virus