（十六）设计模式-组合模式

1、定义

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

2、使用场景

北京总公司总部下属三个机构：人力资源部、上海华东分公司、财务部门；上海华东分公司下也有几个机构：南京办事处、人力资源部、财务部、杭州办事处；南京办事处也有两个下属机构：人力资源部、财务部；杭州办事处也有两个下属机构：人力资源部、财务部。会有一组同类的机构。

3、代码结构UML图

公司：组合中的对象声明接口，在适当的情况下，实现所有类共有接口的默认行为。声明一个接口用于访问和管理（Component）的子部件；
财务部：在组合中表示叶节点对象，叶节点没有子节点；
人力资源部：在组合中表示叶节点对象，叶节点没有子节点；
具体公司：定义有枝节点行为，用来存储子部件，在Comonet接口中实现与子部件有关的操作，比如增加Add和删除Remove。

4、类的实现

（1）、Company（公司类 抽象类或接口）

public abstract class Company {
    protected String name;
    public Company(String name){
        this.name=name;
    }
    public abstract void add(Company c);//增加
    public abstract void remove(Company c);//移除
    public abstract void display(int depth);//显示
    public abstract void lineOfDuty();//履职,不同的部门需履行不同的职责
}

（2）、ConcreteCompany （具体公司类 实现接口 树枝节点）

public class ConcreteCompany extends Company {
    private List<Company> children = new ArrayList<>();

    public ConcreteCompany(String name) {
        super(name);
    }

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

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

    @Override
    public void display(int depth) {
        String s = "";
        for (int i = 0; i < depth; i++) {
            s += "-";
        }
        System.out.println(s + name);
        for (Company component : children) {
            component.display(depth + 2);
        }
    }

    @Override
    public void lineOfDuty() {
        for (Company component : children) {
            component.lineOfDuty();
        }
    }
}

（3）、HRDepartment （人力资源部类 树叶节点）

public class HRDepartment extends Company {
    public HRDepartment(String name) {
        super(name);
    }

    @Override
    public void add(Company c) {
    }

    @Override
    public void remove(Company c) {

    }

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

    @Override
    public void lineOfDuty() {
        System.out.println(name + " 员工培训管理.");
    }
}

（4）、FinanceDepartment （财务部类 树叶节点）

public class FinanceDepartment extends Company {
    public FinanceDepartment(String name) {
        super(name);
    }

    @Override
    public void add(Company c) {

    }

    @Override
    public void remove(Company c) {

    }

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

    @Override
    public void lineOfDuty() {
        System.out.println(name + " 公司财务收支管理.");
    }
}

5、客户端调用

public static void main(String[] args) {
        ConcreteCompany root=new ConcreteCompany("北京总公司");
        root.add(new HRDepartment("总公司人力资源部"));
        root.add(new FinanceDepartment("总公司财务部"));

        ConcreteCompany comp=new ConcreteCompany("上海华东分公司");
        comp.add(new HRDepartment("华东分公司人力资源部"));
        comp.add(new FinanceDepartment("华东分公司财务部"));

        root.add(comp);

        ConcreteCompany comp1=new ConcreteCompany("南京办事处");
        comp1.add(new HRDepartment("南京办事处人力资源部"));
        comp1.add(new FinanceDepartment("南京办事处财务部"));

        comp.add(comp1);

        ConcreteCompany comp2=new ConcreteCompany("杭州办事处");
        comp2.add(new HRDepartment("杭州办事处人力资源部"));
        comp2.add(new FinanceDepartment("杭州办事处财务部"));

        comp.add(comp2);

        System.out.println("\n结构图:");
        root.display(1);

        System.out.println("\n职责:");
        root.lineOfDuty();
    }

输出：

6、总结

组合模式这样就定义了包含人力资源部和财务部这些基本对象和分公司、办事处等组合对象的类层次结构。基本对象可以被组合成更复杂的组合对象，而这个组合对象又可以被组合，这样不断地递归下去，客户代码中，任何用到基本对象的地方都可以使用组合对象了。
用户时不用关心到底是处理一个叶节点还是处理一个组合组件，也就用不着为定义组合而写一些选择判断语句了 。
组合模式是让客户可以一致的使用组合结构和单个对象。

---

参考：《大话设计模式》