组合（Composite）

意图

将对象组合成树形结构来表示“整体/部分”层次关系，允许用户以相同的方式处理单独对象和组合对象。

类图

组件（Component）类是组合类（Composite）和叶子类（Leaf）的父类，可以把组合类看成是树的中间节点。

组合对象拥有一个或者多个组件对象，因此组合对象的操作可以委托给组件对象去处理，而组件对象可以是另一个组合对象或者叶子对象。

实现

public abstract class Component {
    protected String name;

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

    public void print() {
        print(0);
    }

    abstract void print(int level);

    abstract public void add(Component component);

    abstract public void remove(Component component);
}

public class Composite extends Component {

    private List<Component> child;

    public Composite(String name) {
        super(name);
        child = new ArrayList<>();
    }

    @Override
    void print(int level) {
        for (int i = 0; i < level; i++) {
            System.out.print("--");
        }
        System.out.println("Composite:" + name);
        for (Component component : child) {
            component.print(level + 1);
        }
    }

    @Override
    public void add(Component component) {
        child.add(component);
    }

    @Override
    public void remove(Component component) {
        child.remove(component);
    }
}

public class Leaf extends Component {
    public Leaf(String name) {
        super(name);
    }

    @Override
    void print(int level) {
        for (int i = 0; i < level; i++) {
            System.out.print("--");
        }
        System.out.println("Leaf:" + name);
    }

    @Override
    public void add(Component component) {
        throw new UnsupportedOperationException(); // 牺牲单一职责原则换取透明性，这样就不用考虑是叶子节点还是组合节点
    }

    @Override
    public void remove(Component component) {
        throw new UnsupportedOperationException();
    }
}

public class Client {
    public static void main(String[] args) {
        Composite root = new Composite("root");
        Component node1 = new Leaf("1");
        Component node2 = new Composite("2");
        Component node3 = new Leaf("3");
        root.add(node1);
        root.add(node2);
        root.add(node3);
        Component node21 = new Leaf("21");
        Component node22 = new Composite("22");
        node2.add(node21);
        node2.add(node22);
        Component node221 = new Leaf("221");
        node22.add(node221);
        root.print();
    }
}

Composite:root
--Leaf:1
--Composite:2
----Leaf:21
----Composite:22
------Leaf:221
--Leaf:3

root.print() 只调用了一次，Composite 把调用转给每个子节点，遇到子组合再继续向下，因此客户端不用区分叶子和组合。这里把 add()、remove() 声明在 Component 中，叶子调用时只能在运行时抛出异常；另一种做法是只在 Composite 中声明这两个方法，更安全，但客户端添加子节点前要先判断类型。

JDK 中的应用

• javax.swing.JComponent#add(Component)
• java.awt.Container#add(Component)：Container 本身也是 Component，窗口里可以放面板，面板里再放按钮，绘制和布局时统一按 Component 递归处理。

参考资料

• 弗里曼. Head First 设计模式 [M]. 中国电力出版社, 2007.
• Gamma E. 设计模式：可复用面向对象软件的基础 [M]. 机械工业出版社, 2007.