設計模式--組合模式Composite（結構型）

public abstract class Corp {
	// 公司每個人都有名稱
	private String name = "";
	// 公司每個人都職位
	private String position = "";
	// 公司每個人都有薪水
	private int salary = 0;
	// 父節點是誰
	private Corp parent = null;

	/*
	 * 通過介面的方式傳遞，我們改變一下習慣，傳遞進來的引數名以下劃線開始 這個在一些開源專案中非常常見，一般建構函式都是定義的
	 */
	public Corp(String _name, String _position, int _salary) {
		this.name = _name;
		this.position = _position;
		this.salary = _salary;
	}

	// 獲得員工資訊
	public String getInfo() {
		String info = "";
		info = "姓名：" + this.name;
		info = info + "\t職位：" + this.position;
		info = info + "\t薪水：" + this.salary;
		return info;
	}

	// 設定父節點
	protected void setParent(Corp _parent) {
		this.parent = _parent;
	}

	// 等到父節點
	public Corp getParent() {
		return this.parent;
	}
}

public class Branch extends Corp {
	// 領導下邊有那些下級領導和小兵
	ArrayList<Corp> subordinateList = new ArrayList<Corp>();

	// 建構函式是必須的了
	public Branch(String _name, String _position, int _salary) {
		super(_name, _position, _salary);
	}

	// 增加一個下屬，可能是小頭目，也可能是個小兵
	public void addSubordinate(Corp corp) {
		corp.setParent(this); // 設定父節點
		this.subordinateList.add(corp);
	}

	// 我有哪些下屬
	public ArrayList<Corp> getSubordinate() {
		return this.subordinateList;
	}
}

public class Leaf extends Corp {
	// 就寫一個建構函式，這個是必須的
	public Leaf(String _name, String _position, int _salary) {
		super(_name, _position, _salary);
	}
}

public class Client {
	static Leaf f;

	public static void main(String[] args) {
		// 首先是組裝一個組織結構出來
		Branch ceo = compositeCorpTree();
		// 首先把CEO的資訊列印出來：
		System.out.println(ceo.getInfo());
		System.out.println();
		System.out.println("開發人員 f 的上司是：" + f.getParent().getInfo());
		System.out.println();
		// 然後是所有員工資訊
		System.out.println(getTreeInfo(ceo));
	}

	// 把整個樹組裝出來
	public static Branch compositeCorpTree() {
		// 首先產生總經理CEO
		Branch root = new Branch("王大麻子", "總經理", 100000);
		// 把三個部門經理產生出來
		Branch developDep = new Branch("劉大瘸子", "研發部門經理", 10000);
		Branch salesDep = new Branch("馬二柺子", "銷售部門經理", 20000);
		Branch financeDep = new Branch("趙三駝子", "財務部經理", 30000);
		// 再把三個小組長產生出來
		Branch firstDevGroup = new Branch("楊三乜斜", "開發一組組長", 5000);
		Branch secondDevGroup = new Branch("吳大棒槌", "開發二組組長", 6000);
		// 把所有的小兵都產生出來
		Leaf a = new Leaf("a", "開發人員", 2000);
		Leaf b = new Leaf("b", "開發人員", 2000);
		Leaf c = new Leaf("c", "開發人員", 2000);
		Leaf d = new Leaf("d", "開發人員", 2000);
		Leaf e = new Leaf("e", "開發人員", 2000);
		f = new Leaf("f", "開發人員", 2000);
		Leaf g = new Leaf("g", "開發人員", 2000);
		Leaf h = new Leaf("h", "銷售人員", 5000);
		Leaf i = new Leaf("i", "銷售人員", 4000);
		Leaf j = new Leaf("j", "財務人員", 5000);
		Leaf k = new Leaf("k", "CEO祕書", 8000);
		Leaf zhengLaoLiu = new Leaf("鄭老六", "研發部副經理", 20000);
		// 開始組裝
		// CEO下有三個部門經理和一個祕書
		root.addSubordinate(k);
		root.addSubordinate(developDep);
		root.addSubordinate(salesDep);
		root.addSubordinate(financeDep);
		// 研發部經理
		developDep.addSubordinate(zhengLaoLiu);
		developDep.addSubordinate(firstDevGroup);
		developDep.addSubordinate(secondDevGroup);
		// 看看開發兩個開發小組下有什麼
		firstDevGroup.addSubordinate(a);
		firstDevGroup.addSubordinate(b);
		firstDevGroup.addSubordinate(c);
		secondDevGroup.addSubordinate(d);
		secondDevGroup.addSubordinate(e);
		secondDevGroup.addSubordinate(f);
		// 再看銷售部下的人員情況
		salesDep.addSubordinate(h);
		salesDep.addSubordinate(i);
		// 最後一個財務
		financeDep.addSubordinate(j);
		return root;
	}

	// 遍歷整棵樹,只要給我根節點，我就能遍歷出所有的節點
	public static String getTreeInfo(Branch root) {
		ArrayList<Corp> subordinateList = root.getSubordinate();
		String info = "";
		for (Corp s : subordinateList) {
			if (s instanceof Leaf) { // 是員工就直接獲得資訊
				info = info + s.getInfo() + "\n";
			} else { // 是個小頭目
				info = info + s.getInfo() + "\n" + getTreeInfo((Branch) s);
			}
		}
		return info;
	}
}