Java设计模式（九）：组合模式composite

package com.iter.devbox.composite;

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

/**
 * 抽象构件
 * @author Shearer
 *
 */
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 + "进行查杀！");
	}
}
//文件夹（接口）
interface IFolder extends AbstractFile {
	void add(AbstractFile f);
	void remove(AbstractFile f);
	AbstractFile getChild(int index);
}
//文件夹（实现类）
class Folder implements IFolder {
	private String name;
	private List<AbstractFile> list = new ArrayList<AbstractFile>();
	public Folder(String name) {
		this.name = name;
	}
	public void add(AbstractFile f) {
		list.add(f);
	}
	public void remove(AbstractFile f) {
		list.remove(f);
	}
	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.iter.devbox.composite;

public class Client {
	public static void main(String[] args) {
		AbstractFile file1 = new ImageFile("cat.jpg");
		AbstractFile file2 = new TextFile("readme.txt");
		AbstractFile file3 = new VideoFile("海贼王01.mp4");
		IFolder folder = new Folder("文件夹01");
		IFolder subFolder = new Folder("文件夹02");
		folder.add(file1);
		folder.add(subFolder);
		subFolder.add(file2);
		subFolder.add(file3);
		folder.killVirus();
	}
}

运行结果：

---文件夹：文件夹01进行查杀！
---图像文件：cat.jpg进行查杀！
---文件夹：文件夹02进行查杀！
---文本文件：readme.txt进行查杀！
---视频文件：海贼王01.mp4进行查杀！