设计模式——组合模式

抽象接口类：

package com.zyh.designpattern.composite;

public abstract class Company {

private String name;

public Company(String name) {

super();

this.name = name;

}

public Company(){}

public String getName() {

return name;

}

public void setName(String name) {

this.name = name;

}

protected abstract void add(Company company);

protected abstract void romove(Company company);

protected abstract void display(int depth);

}

枝结点类：

package com.zyh.designpattern.composite;

import java.util.ArrayList;

import java.util.List;

public class ConcreteCompany extends Company {

private List cList;　public ConcreteCompany() {　cList = new ArrayList();　}　public ConcreteCompany(String name) {　super(name);　cList = new ArrayList();　}　@Override

protected void add(Company company) {

// TODO Auto-generated method stub

cList.add(company);

}

@Override

protected void display(int depth) {

// TODO Auto-generated method stub

StringBuilder sb = new StringBuilder("");

for (int i = 0; i < depth; i++) {

sb.append("-");

}

System.out.println(new String(sb) + this.getName());

for (Company c : cList) {

c.display(depth + 2);

}

}

@Override

protected void romove(Company company) {

// TODO Auto-generated method stub

cList.remove(company);

}

}

两个叶结点类：

-------------------------1---------------------------.

package com.zyh.designpattern.composite;

public class HRDepartment extends Company {

public HRDepartment(String name) {

super(name);

}

@Override

protected void add(Company company) {

}

@Override

protected void display(int depth) {

// TODO Auto-generated method stub

StringBuilder sb = new StringBuilder("");

for (int i = 0; i < depth; i++) {

sb.append("-");

}

System.out.println(new String(sb) + this.getName());

}

@Override

protected void romove(Company company) {

}

}

----------------2-------------------

package com.zyh.designpattern.composite;

public class FinanceDepartment extends Company {

public FinanceDepartment(String name) {

super(name);

}

@Override

protected void add(Company company) {

}

@Override

protected void display(int depth) {

// TODO Auto-generated method stub

StringBuilder sb = new StringBuilder("");

for (int i = 0; i < depth; i++) {

sb.append("-");

}

System.out.println(new String(sb) + this.getName());

}

@Override

protected void romove(Company company) {

}

}

客户端：

package com.zyh.designpattern.composite;

public class Client {

public static void main(String[] args) {

// TODO Auto-generated method stub

Company root = new ConcreteCompany();

root.setName("北京总公司");

root.add(new HRDepartment("总公司人力资源部"));

root.add(new FinanceDepartment("总公司财务部"));

Company shandongCom = new ConcreteCompany("山东分公司");

shandongCom.add(new HRDepartment("山东分公司人力资源部"));

shandongCom.add(new FinanceDepartment("山东分公司账务部"));

Company zaozhuangCom = new ConcreteCompany("枣庄办事处");

zaozhuangCom.add(new FinanceDepartment("枣庄办事处财务部"));

zaozhuangCom.add(new HRDepartment("枣庄办事处人力资源部"));

Company jinanCom = new ConcreteCompany("济南办事处");

jinanCom.add(new FinanceDepartment("济南办事处财务部"));

jinanCom.add(new HRDepartment("济南办事处人力资源部"));

shandongCom.add(jinanCom);

shandongCom.add(zaozhuangCom);

Company huadongCom = new ConcreteCompany("上海华东分公司");

huadongCom.add(new HRDepartment("上海华东分公司人力资源部"));

huadongCom.add(new FinanceDepartment("上海华东分公司账务部"));

Company hangzhouCom = new ConcreteCompany("杭州办事处");

hangzhouCom.add(new FinanceDepartment("杭州办事处财务部"));

hangzhouCom.add(new HRDepartment("杭州办事处人力资源部"));

Company nanjingCom = new ConcreteCompany("南京办事处");

nanjingCom.add(new FinanceDepartment("南京办事处财务部"));

nanjingCom.add(new HRDepartment("南京办事处人力资源部"));

huadongCom.add(hangzhouCom);

huadongCom.add(nanjingCom);

root.add(shandongCom);

root.add(huadongCom);

root.display(0);

}

}