在使用组合模式中需要注意一点也是组合模式最关键的地方：叶子对象和组合对象实现相同的接口。这就是组合模式能够将叶子节点和对象节点进行一致处理的原因。
下面以File为例，文件夹里有文件和子文件夹，不管是文件还是文件夹，他们都继承了相同的接口FIle
类图
1.创建FIle抽象类
package com.design.demo.composite;
/**
* @author: GuanBin
* @date: Created in 下午10:26 2019/8/18
*/
public abstract class File {
String name;
public File(String name) {
this.name = name;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public abstract void display();
}
2.创建Folder继承File
package com.design.demo.composite;
import java.util.ArrayList;
import java.util.List;
/**
* @author: GuanBin
* @date: Created in 下午10:30 2019/8/18
*/
public class Folder extends File {
private List<File> files;
public Folder(String name) {
super(name);
this.files = new ArrayList<>();
}
/**
* 浏览文件夹中的文件信息
*/
@Override
public void display() {
for (File file : files) {
file.display();
}
}
/**
* 向文件夹中添加文件
*
* @param file
*/
public void add(File file) {
files.add(file);
}
/**
* 从文件夹中删除文件
*
* @param file
*/
public void remove(File file) {
files.remove(file);
}
}
3.创建ImageFile继承File
package com.design.demo.composite;
/**
* @author: GuanBin
* @date: Created in 下午10:50 2019/8/18
*/
public class ImageFile extends File {
public ImageFile(String name) {
super(name);
}
@Override
public void display() {
System.out.println("这是图像文件，文件名："+super.name);
}
}
4.创建TextFile继承FIle
package com.design.demo.composite;
/**
* @author: GuanBin
* @date: Created in 下午10:50 2019/8/18
*/
public class TextFile extends File {
public TextFile(String name) {
super(name);
}
@Override
public void display() {
System.out.println("这是文本文件，文件名："+super.name);
}
}
5.创建VideoFile继承File
package com.design.demo.composite;
/**
* @author: GuanBin
* @date: Created in 下午10:50 2019/8/18
*/
public class VideoFile extends File {
public VideoFile(String name) {
super(name);
}
@Override
public void display() {
System.out.println("这是影像文件，文件名："+super.name);
}
}
6.创建Client
package com.design.demo.composite;
/**
* @author: GuanBin
* @date: Created in 下午10:53 2019/8/18
*/
public class Client {
public static void main(String[] args) {
Folder folder = new Folder("总文件夹");
folder.add(new TextFile("text.txt"));
folder.add(new ImageFile("image.jpg"));
Folder testFolder = new Folder("test文件夹");
testFolder.add(new TextFile("test1.txt"));
testFolder.add(new ImageFile("test2.jpg"));
testFolder.add(new VideoFile("test3.rmvb"));
//把test文件夹添加到总文件夹下
folder.add(testFolder);
//遍历
folder.display();
System.out.println("*****************************");
//添加
TextFile addFile = new TextFile("add.txt");
testFolder.add(addFile);
folder.display();
System.out.println("*****************************");
//删除testFolder
folder.remove(testFolder);
folder.display();
}
}
输出
这是文本文件，文件名：text.txt 这是图像文件，文件名：image.jpg 这是文本文件，文件名：test1.txt 这是图像文件，文件名：test2.jpg 这是影像文件，文件名：test3.rmvb ***************************** 这是文本文件，文件名：text.txt 这是图像文件，文件名：image.jpg 这是文本文件，文件名：test1.txt 这是图像文件，文件名：test2.jpg 这是影像文件，文件名：test3.rmvb 这是文本文件，文件名：add.txt ***************************** 这是文本文件，文件名：text.txt 这是图像文件，文件名：image.jpg Process finished with exit code 0