JAVA设计模式之组合模式原理与用法详解

/**
 * 抽象类：用来管理子类
 * @description：
 * @date 2016-1-22 下午1:16:25
 */
public abstract class Firm {
  private String firmName;// 公司名称
  public Firm(String firmName) {
    this.firmName = firmName;
  }
  public String getFirmName() {
    return firmName;
  }
  public void setFirmName(String name) {
    this.firmName = name;
  }
  protected abstract void append(Firm firm);// 添加公司如：新增分公司
  protected abstract void romove(Firm firm);// 删除公司：如因故关闭分公司
  protected abstract void show(int level);//显示公司所在的层级:总公司，分公司，办事处等
}

/**
 * 分公司对象
 * @description：
 * @date 2016-1-22 下午1:25:43
 */
public class BranchFirm extends Firm {
  private List<Firm> cList;
  public BranchFirm(String name) {
    super(name);
    cList = new ArrayList<Firm>();
  }
  @Override
  protected void append(Firm firm) {
    cList.add(firm);
  }
  @Override
  protected void show(int depth) {
    StringBuilder sb = new StringBuilder("");
    for (int i = 0; i < depth; i++) {
      sb.append("*");
    }
    System.out.println(new String(sb) + this.getFirmName());
    for (Firm c : cList) {
      c.show(depth + 2);
    }
  }
  @Override
  protected void romove(Firm firm) {
    // TODO Auto-generated method stub
    cList.remove(firm);
  }
}

/**
 * Leaf结点类：公司市场部
 * @description：
 * @date 2016-1-22 下午1:46:02
 */
public class MarketDepartment extends Firm {
  public MarketDepartment(String name) {
    super(name);
  }
  @Override
  protected void append(Firm company) {
  }
  @Override
  protected void show(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.getFirmName());
  }
  @Override
  protected void romove(Firm company) {
  }
}

/**
 * Leaf结点类：公司技术部
 * @description：
 * @date 2016-1-22 下午1:21:41
 */
public class MinTechDepartment extends Firm {
  public MinTechDepartment(String name) {
    super(name);
  }
  @Override
  protected void append(Firm firm) {
  }
  @Override
  protected void show(int depth) {
    StringBuilder sb = new StringBuilder("");
    for (int i = 0; i < depth; i++) {
      sb.append("*");
    }
    System.out.println(new String(sb) + this.getFirmName());
  }
  @Override
  protected void romove(Firm company) {
  }
}

public class Test {
  public static void main(String[] args) {
    // TODO Auto-generated method stub
    Firm head = new BranchFirm("深圳XX集团有限公司");// 公司总部
    head.append(new MarketDepartment("深圳XX集团有限公司--市场部"));
    head.append(new MinTechDepartment("深圳XX集团有限公司--技术部"));
    Firm jsFirm = new BranchFirm("深圳XX集团有限公司-江苏分公司");
    jsFirm.append(new MarketDepartment("深圳XX集团有限公司-江苏分公司市场部"));
    jsFirm.append(new MinTechDepartment("深圳XX集团有限公司-江苏分公司技术部"));
    Firm szFirm = new BranchFirm("深圳XX集团有限公司-江苏分公司-苏州办事处");
    szFirm.append(new MinTechDepartment("深圳XX集团有限公司-江苏分公司-苏州办事处技术部"));
    szFirm.append(new MarketDepartment("深圳XX集团有限公司-江苏分公司-苏州办事处市场部"));
    jsFirm.append(szFirm);
    // 如果还需要新增分公司或办事处等，就继续按照上述操作处理
    head.append(jsFirm);
    head.show(0);
  }
}