大话设计模式-Java实现（19）-组合模式

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

举例：
一个企业做大后，会在全国各地开分公司，总公司有人力资源部和财务部，分公司也有自己的人力资源部和财务部，分公司下还可以有办事处，办事处也有自己的人力资源部和财务部，但是人力资源部和财务部并没有下属的机构。这种场景就可以用组合模式来实现公司组织架构的展示。

1.公司类

public abstract class Company {
	
    String name;
    
    public Company(String name) {
        this.name = name;
    }
    
    public abstract void add(Company company);
    public abstract void remove(Company company);
    public abstract void show(int depth); // 展示结构
    public abstract void duty(int depth); // 履行职责
    
}

2.树枝节点公司实体类

import java.util.ArrayList;
import java.util.Collections;
public class ConcreteCompany extends Company {

    private ArrayList<Company> children = new ArrayList<>();

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

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

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

    @Override
    public void show(int depth) {
        System.out.println(String.join("-", Collections.nCopies(depth, "-")) + name);
        for (Company company : children) {
            company.show(depth + 1);
        }
    }

    @Override
    public void duty(int depth) {
        System.out.println(String.join("-", Collections.nCopies(depth, "-")) + name + "公司");
        for (Company company : children) {
            company.duty(depth + 1);
        }
    }

}

3.树叶节点公司实体类

import java.util.Collections;
//财务部
public class FinanceDepartment extends Company {

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

	@Override
	public void add(Company company) {}

	@Override
	public void remove(Company company) {}

	@Override
	public void show(int depth) {
		System.out.println(String.join("-", Collections.nCopies(depth, "-")) + name);
	}

	@Override
	public void duty(int depth) {
		System.out.println(String.join("-", Collections.nCopies(depth, "-")) + name + ":公司财务收支管理");
	}

}

import java.util.Collections;
//人力资源部
public class HRDepartment extends Company {

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

	@Override
	public void add(Company company) {}

	@Override
	public void remove(Company company) {}

	@Override
	public void show(int depth) {
		System.out.println(String.join("-", Collections.nCopies(depth, "-")) + name);
	}

	@Override
	public void duty(int depth) {
		System.out.println(String.join("-", Collections.nCopies(depth, "-")) + name + ":员工招聘培训管理");
	}

}

4.主程序

public class Test {

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

		ConcreteCompany comp1 = new ConcreteCompany("华东分公司");
		comp1.add(new HRDepartment("华东分公司人力资源部"));
		comp1.add(new FinanceDepartment("华东分公司人力资源部"));
		root.add(comp1);

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

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

		root.show(1);
		System.out.println("========================");
		root.duty(1);
	}

}

运行结果如下：

使用场景：

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