设计模式学习笔记(十三)组合模式及其在树形结构中的应用

组合(Composite)模式,又叫做树形模式,主要用来处理树形结构数据。是将一组对象组织成树形结构,以表示一种“部分-整体”的层次结构。让客户端可以统一单个对象和组合对象的处理逻辑。

设计模式学习笔记(十三)组合模式及其在树形结构中的应用_第1张图片

一、组合模式介绍

组合模式通过以树形结构来表示“部分-整体”,使得用户对叶对象和组合对象的使用具有一致性。也就是说在组合模式中,整个树形结构的对象都属于同一种类型,用户可以对叶对象和组合对象统一处理。

1.1 组合模式分类

组合模式主要有透明式和安全式两种分类,下面来分别说明

1.1.1 透明式组合模式

在该方式中,抽象构件声明了所有子类中的全部方法,这样实现抽象构件接口的所有子类都具备了全部方法,这样的好处是叶节点和枝节点对于外界没有任何区别,它们具备了完全一致的接口。但是对于叶节点有些本身不具备的方法,就可能会有安全隐患(空指针异常等)。其结构类图如下所示:

设计模式学习笔记(十三)组合模式及其在树形结构中的应用_第2张图片

  • Component:抽象构件,为叶节点和树枝节点声明公共接口,以及访问和管理子类的接口
  • Composite:树枝构件,组合中的分支节点对象,作用是存储和管理子部件
  • Leaf:树叶构件,组合中的叶节点对象,用于继承和实现抽象构件
  • Client:客户端

1.1.2 安全式组合模式

前面提到透明式组合模式中,因为抽象构件声明所有子类方法,有可能会造成安全问题。所以在安全式中,将管理叶节点的方法转移到树枝构件中,抽象构件和树叶构件没有对子对象的管理方法,这样就避免了透明式组合模式中的安全问题。但是由于树叶和树枝构件有不同的接口,因此在使用时,就不能将两种构件一概而论,对于客户端调用方而言,就失去了透明性。其结构类图如下所示:

设计模式学习笔记(十三)组合模式及其在树形结构中的应用_第3张图片

  • Component:抽象构件,为叶节点和树枝节点声明公共接口,没有访问和管理子类的接口
  • Composite:树枝构件,组合中的分支节点对象,作用是存储和管理子部件
  • Leaf:树叶构件,组合中的叶节点对象,没有对子类的管理方法
  • Client:客户端

1.2 组合模式实现

根据上面的类图,可以实现如下代码:

1.2.1 透明式组合模式实现

/**
 * @description: 透明式抽象构件
 * @author: wjw
 * @date: 2022/4/3
 */
public interface Component {

    /**公共操作方法**/
   void operation();

    /**
     * 添加构件
     * @param c 组合模式中的构件
     */
    void add(Component c);

    /**
     * 移除构件
     * @param c 组合模式中的构件
     */
    void remove(Component c);

    /**
     * 获得子对象
     * @param t 子对象序号
     * @return  子对象
     */
    Component getChild(int t);

}

/**
 * @description: 树枝节点
 * @author: wjw
 * @date: 2022/4/3
 */
public class Composite implements Component{

    private ArrayList children = new ArrayList<>();

    @Override
    public void operation() {
        for (Component child : children) {
            child.operation();
        }
    }

    @Override
    public void add(Component c) {
        children.add(c);
    }

    @Override
    public void remove(Component c) {
        children.remove(c);
    }

    @Override
    public Component getChild(int t) {
        return children.get(t);
    }
}

/**
 * @description: 树叶节点
 * @author: wjw
 * @date: 2022/4/3
 */
public class Leaf implements Component{
    private String name;

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

    @Override
    public void operation() {
        System.out.println("我是树叶节点:" + name);
    }

    @Override
    public void add(Component c) {

    }

    @Override
    public void remove(Component c) {

    }

    @Override
    public Component getChild(int t) {
        return null;
    }
}
/**
 * @description: 客户端类
 * @author: wjw
 * @date: 2022/4/3
 */
public class Client {
    public static void main(String[] args) {
        Component component = new Composite();
        Component leaf1 = new Leaf("1");
        Component leaf2 = new Leaf("2");
        component.add(leaf1);
        component.add(leaf2);
        component.operation();
        component.getChild(1).operation();
        //这里树叶构件能调用add方法就会造成安全隐患
        leaf1.add(leaf1);
    }
}

客户端运行结果:

我是树叶节点:1
我是树叶节点:2
我是树叶节点:2

1.2.2 安全式组合模式实现

/**
 * @description: 安全式抽象构件
 * @author: wjw
 * @date: 2022/4/3
 */
public interface Component {

    /**公共操作方法**/
   void operation();
}

/**
 * @description: 树枝节点
 * @author: wjw
 * @date: 2022/4/3
 */
public class Composite implements Component{

    private ArrayList children = new ArrayList<>();

    @Override
    public void operation() {
        for (Component child : children) {
            child.operation();
        }
    }


    public void add(Component c) {
        children.add(c);
    }


    public void remove(Component c) {
        children.remove(c);
    }


    public Component getChild(int t) {
        return children.get(t);
    }
}

/**
 * @description: 树叶节点
 * @author: wjw
 * @date: 2022/4/3
 */
public class Leaf implements Component{
    private String name;

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

    @Override
    public void operation() {
        System.out.println("我是树叶节点:" + name);
    }

}

/**
 * @description: 客户端类
 * @author: wjw
 * @date: 2022/4/3
 */
public class Client {
    public static void main(String[] args) {
        Composite composite = new Composite();
        Leaf leaf1 = new Leaf("1");
        Leaf leaf2 = new Leaf("2");
        composite.add(leaf1);
        composite.add(leaf2);
        composite.operation();
    }
}

客户端测试结果:

我是树叶节点:1
我是树叶节点:2

二、组合模式应用场景

组合模式常见的应用场景主要是出现树形结构的地方,比如文件目录,公司人员架构图等等

2.1 公司人员架构

比如按照部门和员工组织成树形结构,可以统一处理薪资:

设计模式学习笔记(十三)组合模式及其在树形结构中的应用_第4张图片

/**
 * @description: 人力资源抽象构件
 * @author: wjw
 * @date: 2022/4/3
 */
public abstract class HumanResource {
    protected long id;
    protected double salary;

    public HumanResource(long id) {
        this.id = id;
    }

    public long getId() {
        return id;
    }

    /**
     * 计算工资
     * @return 工资结果
     */
    public abstract double calculateSalary();
}

/**
 * @description: 部门树枝构件
 * @author: wjw
 * @date: 2022/4/3
 */
public class Department extends HumanResource{

    private List humanResources = new ArrayList<>();

    public Department(long id) {
        super(id);
    }

    @Override
    public double calculateSalary() {
        double totalSalary = 0;
        for (HumanResource humanResource : humanResources) {
            totalSalary += humanResource.calculateSalary();
        }
        this.salary = totalSalary;
        return totalSalary;
    }

    public void addHumanResource(HumanResource humanResource) {
        humanResources.add(humanResource);
    }
}
/**
 * @description: 员工树叶构件
 * @author: wjw
 * @date: 2022/4/3
 */
public class Employee extends HumanResource{

    public Employee(long id, double salary) {
        super(id);
        this.salary = salary;
    }

    @Override
    public double calculateSalary() {
        return salary;
    }
}

参考资料

《设计模式之美》

http://c.biancheng.net/view/1373.html

《Java 设计模式》

《设计模式:可复用面向对象软件的基础》

你可能感兴趣的:(设计模式学习笔记(十三)组合模式及其在树形结构中的应用)