《大话设计模式》组合模式 Java代码

定义

组合模式Composite：将对象组合成树形结构以表示‘部分-整体’的层次结构。组合模式使得用户对单个对象和组合对象的使用具有一致性。

结构图

代码

//客户端代码，能通过Component接口操作组合部件的对象
public class Main {
    public static void main(String[] args) {
        Composite root=new Composite("root");
        root.Add(new Leaf("LeafA"));
        root.Add(new Leaf("LeafB"));

        Composite comp=new Composite("Composite X");
        comp.Add(new Leaf("LeafXA"));
        comp.Add(new Leaf("LeafXB"));

        root.Add(comp);

        Composite comp2=new Composite("Composite XY");
        comp2.Add(new Leaf("LeafXYA"));
        comp2.Add(new Leaf("LeafXYB"));

        comp.Add(comp2);

        root.Add(new Leaf("Leaf C"));//又长出C

        Leaf leaf=new Leaf("Leaf D");//又长出D,D又走了
        root.Add(leaf);
        root.Remove(leaf);

        root.Display(1);

    }
}
//Component 为组合中的对象声明接口，在适当情况下，实现所有类共有接口的默认行为。声明一个接口用于访问和管理Component的子部件。
abstract class Component{
    protected String name;
    public Component(String name){
        this.name=name;
    }
    public abstract void Add(Component c);
    public abstract void Remove(Component c);
    public abstract void Display(int depth);
}
//Leaf在组合中表示叶节点对象，叶节点没有子节点
class Leaf extends Component{
    public Leaf(String name){
        super(name);
    }
    @Override
    public void Add(Component c) {
        System.out.println("Cannot add to a leaf");
    }

    @Override
    public void Remove(Component c) {
        System.out.println("Cannot remove from a leaf");
    }

    @Override
    public void Display(int depth) {
        String s="";
        for(int i=0;i<depth;i++){
            s=s+"-";
        }
        System.out.println(s+name);
    }
}
//Composite定义有枝节点行为，用来存储子部件，在Component接口中实现与子部件有关的操作，比如增加Add和删除Remove
class Composite extends Component{
    private ArrayList<Component> children=new ArrayList<>();
    public Composite(String name){
        super(name);
    }
    @Override
    public void Add(Component c) {
        children.add(c);
    }

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

    @Override
    public void Display(int depth) {
        String s="";
        for(int i=0;i<depth;i++){
            s=s+"-";
        }
        System.out.println(s+name);

        //对下级进行遍历
        for(Component c:children){
            c.Display(depth+2);
        }
    }
}

何时使用组合模式

当需求中是体现部分与整体层次的结构时，
以及希望用户可以忽略组合对象与单个对象的不同，统一地使用组合结构中的所有对象时，
就应该考虑用组合模式了。

组合模式好处

总结：组合模式让客户可以一致地使用组合结构和单个对象。
基本对象可以被组合成更复杂的组合对象，而这个组合对象又可以被组合，这样不断地递归下去，客户代码中，任何用到基本对象的地方都可以使用组合对象了。
用户不用关心到底是处理一个叶节点还是处理一个组合组件，也就用不着为定义组合而写一些选择判断语句了。

透明方式与安全方式

透明方式：在Component中声明所有用来管理子对象的方法。这样做的好处时叶节点枝节点对于外界没有区别，它们具备完全一致的行为接口。但问题也很明显，因为Leaf类本身不具备部分方法的功能，所以实现它是没有意义的。
安全方式：在Component接口中不去声明部分方法，那么Leaf也就不需要去实现它，而是在Composite声明所有用来管理子类对象的方法。不过由于不够透明，所以树枝和树叶不具备相同的接口，客户端的调用就要做响应的判断，带来了不便。