大话设计模式_组合模式

以总公司下设分公司用一套系统为例子.

package com.wzs.design;

import java.util.ArrayList;
import java.util.List;

/**
 * 大话设计模式--page194 组合模式
 * 
 * @author Administrator
 * 
 */
public class CompositeMode {
	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("结构图：");
		root.display(1);
		System.out.println("职责：");
		root.lineOfDuty();
	}
}

// 公司抽象类
abstract class Company {
	public String name;

	public Company(String name) {
		this.name = name;
	}

	public abstract void add(Company company);// 增加

	public abstract void remove(Company company);// 移除

	public abstract void display(int depth);// 显示

	public abstract void lineOfDuty();// 履行职责

}

// 具体公司类
class ConcreteCompany extends Company {

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

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

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

	@Override
	public void display(int depth) {
		System.out.println(MyString.createString("-", depth) + " " + this.name);
		for (Company company : children) {
			company.display(depth + 2);
		}
	}

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

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

}

// 人力资源部
class HRDepartment extends Company {

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

	@Override
	public void add(Company company) {

	}

	@Override
	public void display(int depth) {
		System.out.println(MyString.createString("-", depth) + " " + this.name);
	}

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

	@Override
	public void remove(Company company) {

	}

}

// 财务部
class FinanceDepartment extends Company {

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

	@Override
	public void add(Company company) {

	}

	@Override
	public void display(int depth) {
		System.out.println(MyString.createString("-", depth) + " " + this.name);
	}

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

	@Override
	public void remove(Company company) {

	}

}

/*
 * 创造字符前缀
 */
class MyString {
	public static String createString(String str, int depth) {
		String strTemp = "";
		for (int i = 0; i < depth; i++) {
			strTemp += str;
		}
		return strTemp;
	}
}