[设计模式]之十:组合模式

// 声明组织具有的功能，所有子组织都要实现这些方法
abstract public class Component {
	protected String name;

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}
	
	public abstract void add(Component c);
	public abstract void remove(Component c);
	public abstract void display(int depth);
	
}
// Leaf节点，即实现部门功能。
public class Leaf extends Component {

	public Leaf(String name) {
		this.name = name;
	}
	
	@Override
	public void add(Component c) {
		System.out.println("Leaf cannot add anything");
	}

	@Override
	public void remove(Component c) {
		System.out.println("Leaf has nothing to remove");
	}

	@Override
	public void display(int depth) {
		String str = "";
		for (int i = 0; i < depth; i++) {
			str += "-";
		}
		System.out.println(str + name);
	}

}
// 组织结构，可以添加子组织
public class Composite extends Component {

	private List<Component> children = new ArrayList<>();
	
	public Composite(String str) {
		this.name = str;
	}
	
	@Override
	public void add(Component c) {
		children.add(c);
	}

	@Override
	public void remove(Component c) {
		children.remove(c);
	}

	@Override
	public void display(int depth) {
		String str = "";
		for (int i = 0; i < depth; i++) {
			str += "-";
		}
		System.out.println(str + name);
		
		for(Component c : children) {
			c.display(depth + 2);
		}
	}

}
//客户代码
public static void main(String[] args) {
	// TODO Auto-generated method stub
	Composite root = new Composite("总公司");
	root.add(new Leaf("财务部"));
	root.add(new Leaf("组织部"));
	
	Composite comp = new Composite("深圳分公司");
	comp.add(new Leaf("财务部"));
	comp.add(new Leaf("组织部"));
	
	root.add(comp);
	
	Composite comp2 = new Composite("中国香港分公司");
	comp2.add(new Leaf("财务部"));
	comp2.add(new Leaf("组织部"));
	
	root.add(comp2);
	root.display(1);
}
//输出
-总公司
---财务部
---组织部
---深圳分公司
-----财务部
-----组织部
---中国香港分公司
-----财务部
-----组织部