树形结构魔法——组合模式

最新推荐文章于 2024-07-11 11:35:25 发布

码匠er

最新推荐文章于 2024-07-11 11:35:25 发布

阅读量779

点赞数 20

分类专栏：设计模式文章标签：组合模式设计模式开发语言 java

本文链接：https://blog.csdn.net/wzllby/article/details/136823651

版权

设计模式专栏收录该内容

6 篇文章 0 订阅

订阅专栏

📚 在计算机的文件系统中，我们有文件，也有文件夹，文件我们又有音乐类型的文件、视频类型的文件、图片类型的文件和文本类型文件等，文件夹中可以包含文件，也可以包含文件夹。如果A公司需要写一个服务器文件管理系统，系统需要列出服务器中的每个文件和文件夹信息展示，而且对文件进行管理。

🙋‍♂️ 类似于对于这种管理属性结构数据的需求，比如管理文件系统中的文件和文件夹、公司组织机构中的部门等，我们都可以使用组合模式来实现相关的功能。

1. 概述

组合模式将对象组合成树形结构以表示"部分-整体"的层次结构，使客户端对单个对象（具体内容）和组合对象（容器）的处理具有一致性，无需关系它们具体是哪种类型。

组合模式一般有两种对象：

叶子对象：叶子对象是组合中最基本的对象，不具备子对象；

组合对象：组合对象包含叶子对象和其他的组合对象。

2. 结构

组合模式结构包含以下几个角色：

抽象层组件（Component）：为叶子对象和组合对象声明接口，包含所有公共的行为的声明和实现，用于访问和管理组合对象的方法，如新增、删除、获取等，一般是抽象类或者接口；
叶子节点（Leaf）：实现抽象层组件，表示叶子对象，叶子节点没有子节点，是组合接口的最基本的单元，一般对于抽象层组件中的访问和管理组合对象的方法通过异常等方式进行处理；
组合对象（Composite）：实现抽象层组件，包含了子节点，可以是叶子节点也可以是组合对象，实现了抽象组件中定义的访问和管理对象的方法。

3. UML图

4. 实现

抽象层组件

 abstract class Component {
     public abstract void add(Component component); 
     public abstract void remove(Component component); 
     public abstract Component get(int index); 
     public abstract void oper(); 
 }

叶子节点

 class Leaf extends Component {
     public void add(Component component) {
         // 异常处理或者错误提示
     }
     public void remove(Component component) {
         // 异常处理或者错误提示
     } 
     public Component get(int index) {
         // 异常处理或者错误提示
     } 
     public void oper() {
         // 叶子节点业务方式实现
     }
 }

组合对象

 class Composite extends Component {
     private List<Component> children = new ArrayList<>();
     
    public void add(Component component) {
         children.add(component);
     }
     public void remove(Component component) {
         children.remove(component);
     } 
     public Component get(int index) {
         return children.get(index);
     } 
     public void oper() {
         // 组合对象的业务方式实现
         // 循环调用子节点的业务方法
         for(Component com : children) {
             com.oper();
         }
     }
 }

5. 案例分析

本案例实现文章开头介绍的文件管理系统。

0️⃣ 实现一个抽象组件类

 // 抽象组件类
 public abstract class EntityFile {
     private String name;
 
     private int size;
 
     public String getName() {
         return this.name;
     }
 
     public int getSize() {
         return this.size;
     }
     // 管理文件的方法
     public abstract void add(EntityFile file);
 
     public abstract void remove(EntityFile file);
 
     public abstract void showChild();
 
     public void setName(String name) {
         this.name = name;
     }
 
     public void setSize(int size) {
         this.size = size;
     }
 }

1️⃣ 实现视频文件和音频文件

 public class MusicFile extends EntityFile {
 
     public MusicFile(String name, int size) {
         setName(name);
         setSize(size);
     }
 
     @Override
     public void add(EntityFile file) {
         throw new RuntimeException("当前文件不支持添加子文件");
     }
 
     @Override
     public void remove(EntityFile file) {
         throw new RuntimeException("当前文件没有子文件");
     }
 
     @Override
     public void showChild() {
         throw new RuntimeException("当前文件没有子文件");
     }
 }
 
 public class VideoFile extends EntityFile {
 
     public VideoFile(String name, int size) {
         setName(name);
         setSize(size);
     }
 
     @Override
     public void add(EntityFile file) {
         throw new RuntimeException("当前文件不支持添加子文件");
     }
 
     @Override
     public void remove(EntityFile file) {
         throw new RuntimeException("当前文件没有子文件");
     }
 
     @Override
     public void showChild() {
         throw new RuntimeException("当前文件没有子文件");
     }
 }

2️⃣ 实现文件夹的功能

public class Directory extends EntityFile {
 
     private List<EntityFile> children;
 
     public Directory(String name) {
         setName(name);
         children = new ArrayList<>();
         setSize(0);
     }
 
     @Override
     public void add(EntityFile file) {
         children.add(file);
         setSize();
     }
 
     @Override
     public void remove(EntityFile file) {
         children.remove(file);
         setSize();
     }
 
     @Override
     public void showChild() {
         System.out.println("文件夹名称为" + getName() + ", 共有" + children.size() + "个文件,大小为" + getSize());
         for(EntityFile file : children) {
             System.out.println(file.getName() + "-" + file.getSize());
         }
     }
 
     public void setSize() {
         setSize(children.stream().mapToInt(EntityFile::getSize).reduce(0, Integer::sum));
     }
 }

3️⃣ 客户端代码实现

 public class Client {
     public static void main(String[] args) {
         // 创建一个名为dir的文件夹
         EntityFile directory = new Directory("dir");
         // 创建两个音乐文件
         EntityFile daYu = new MusicFile("大鱼", 3452);
         EntityFile guaiKa = new MusicFile("怪咖", 5621);
         // 将文件放进到文件夹
         directory.add(daYu);
         directory.add(guaiKa);
         directory.showChild();
         // 创建视频文件
         EntityFile video = new VideoFile("视频1", 52153);
         directory.add(video);
         // 删除大鱼音乐文件
         directory.remove(daYu);
         directory.showChild();
     }
 }