大话设计模式——15组合模式

一、概念

将对象组成树形结构，以表示部分整体的层次结构。整体和部分可以被一致对待。当需求中体现整体和部分层次的结构，以及希望用户可以忽略组合对象与单个对象的不同，统一地使用组合结构中所有的对象时，就可以考虑该模式。

二、代码示例

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 duty();
}

public class ParentCompany extends Company {
    private List<Company> companyList = new ArrayList<Company>();
    public ParentCompany(String name) {
        super(name);
    }
    @Override
    public void add(Company c) {
        companyList.add(c);
    }
    @Override
    public void remove(Company c) {
        companyList.remove(c);
    }
    @Override
    public void display(int depth) {
        System.out.println(depth + "-" + name);
        for (Company company : companyList) {
            company.display(depth + 1);
        }
    }
    @Override
    public void duty() {
        for (Company company : companyList) {
            company.duty();
        }
    }
}

public class HRDepartment extends Company {
    public HRDepartment(String name) {
        super(name);
    }
    @Override
    public void add(Company c) {
        // do nothing
    }
    @Override
    public void remove(Company c) {
        // do nothing
    }
    @Override
    public void display(int depth) {
        System.out.println(depth + "-" + name);
    }
    @Override
    public void duty() {
        System.out.println("公司员工招聘管理" + name);
    }
}

public class FinanceDepartment extends Company {
    public FinanceDepartment(String name) {
        super(name);
    }
    @Override
    public void add(Company c) {
        // do nothing
    }
    @Override
    public void remove(Company c) {
        // do nothing
    }
    @Override
    public void display(int depth) {
        System.out.println(depth + "-" + name);
    }
    @Override
    public void duty() {
        System.out.println("公司财务收支管理" + name);
    }
}

public class RunMain {
    public static void main(String[] args) {
        ParentCompany parentCompany1 = new ParentCompany("总公司");
        parentCompany1.add(new FinanceDepartment("总公司财务部"));
        parentCompany1.add(new HRDepartment("总公司人资部"));
        ParentCompany parentCompany2 = new ParentCompany("分公司");
        parentCompany2.add(new FinanceDepartment("分公司财务部"));
        parentCompany2.add(new HRDepartment("分公司人资部"));
        parentCompany1.add(parentCompany2);
        System.out.println("架构");
        parentCompany1.display(1);
        System.out.println("职责");
        parentCompany1.duty();
    }
}

输出结果如下： 

架构
1-总公司
2-总公司财务部
2-总公司人资部
2-分公司
3-分公司财务部
3-分公司人资部
职责
公司财务收支管理总公司财务部
公司员工招聘管理总公司人资部
公司财务收支管理分公司财务部
公司员工招聘管理分公司人资部