（@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, 0);
    }

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

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

测试结果。

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