大話設計模式-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);
	}

}
           

運作結果如下：

使用場景：

需求中是展現部分與整體層次的結構時，希望使用者可以忽略組合對象與單個對象的不同，統一地使用組合結構中的所有對象時，就應該考慮使用組合結構。