（@WhiteTaken）设计模式学习——组合模式

下面来学习一下组合模式。

组合模式概念是什么呢。从别的地方抄来了一些理论。

理论：将对象组合成树形结构以表示“部分-整体”的层次结构。Composite模式使得用户对单个对象和组合对象的使用具有一致性。

拙见：组合模式就是为处理树形结构优化出来的一套理论方法。-_-!!!用递归可以统一处理树形结构中的对象。

下面直接上代码。

总公司  --分公司

　　　　　　    --分公司下属分公司

　　　　　　    --分公司下属办事处

　　     --分公司办事处

接口类ICompany。

 import java.util.List;

 public interface ICompany {
     public void display();
     public boolean add(ICompany company);
     public boolean remove(ICompany company);
     public List<ICompany> getChild();
 }

公司类Company。

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

 public class Company implements ICompany {
     private String name;
     private List<ICompany> children;

     public Company(String name) {
         this.name = name;
         this.children = new ArrayList<ICompany>();
     }
     @Override
     public void display() {
         System.out.println(name);
     }
     @Override
     public boolean add(ICompany company) {
         return children.add(company);
     }
     @Override
     public boolean remove(ICompany company) {
         return children.remove(company);
     }
     @Override
     public List<ICompany> getChild() {
         return children;
     }
 }

办事处类ConcreteCompany。

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

 public class ConcreteCompany implements ICompany {
     private String name;

     public ConcreteCompany(String name) {
         this.name = name;
     }
     @Override
     public void display() {
         System.out.println(name);
     }
     @Override
     public boolean add(ICompany company) {
         return false;
     }
     @Override
     public boolean remove(ICompany company) {
         return false;
     }
     @Override
     public List<ICompany> getChild() {
         return null;
     }
 }

测试类MainClass。

 import java.util.List;

 /*
  * 组合模式测试
  */
 public class MainClass {
     public static void main(String[] args) {
         ICompany company = new Company("总公司");
         ICompany company1 = new Company("分公司1");
         ICompany company2 = new Company("分公司2");
         ICompany company3 = new ConcreteCompany("分公司办事处");
         company.add(company1);
         company.add(company2);
         company.add(company3);

         ICompany company11 = new Company("分公司1下属公司1");
         ICompany company12 = new Company("分公司1下属公司2");
         ICompany company13 = new ConcreteCompany("分公司1下属办事处");
         company1.add(company11);
         company1.add(company12);
         company1.add(company13);

         ICompany company21 = new Company("分公司2下属公司");
         ICompany company22 = new ConcreteCompany("分公司2下属办事处");
         company2.add(company21);
         company2.add(company22);

         displayAllCompany(company, );
     }

     private static void displayAllCompany(ICompany rootCompany, int deep) {
         for(int i = ; i < deep; i++)
         {
             System.out.print("-");
         }
         rootCompany.display();
         List<ICompany> children = rootCompany.getChild();
         for(ICompany company:children)
         {
             if(company instanceof ConcreteCompany)
             {
                 for(int i = ; i <= deep; i++)
                 {
                     System.out.print("-");
                 }
                 company.display();
             }
             else
             {
                 displayAllCompany(company, deep + );
             }
         }

     }
 }

以上代码主要就是用递归方法遍历了公司的结构。

测试结果。

这个模式的精华都在代码里，大家自己体会吧。。。共同进步！！！