设计模式系列之十一组合模式

组合模式：允许将对象组合成树型结构来表现“整体/部分”层次结构。组合能让一致的方式处理个别对象以及对象组合。

  接上一篇设计模式系列之十迭代器模式的业务逻辑，现在集团公司业务扩张，新增了湖南分公司，那么采用迭代器模式可以很方便的打印出湖南分公司的所有部门。

一、新需求实现

//湖南分公司
public class HNBranch {
    ArrayList<Dept> depts;

    public HNBranch(){
        depts = new ArrayList<Dept>();
        //添加部门
        depts.add(new Dept("湖南-财务部"));
        depts.add(new Dept("湖南-开发部"));
    }

    //这里不再使用自定义实现的Iterator接口，而是使用Java原生集合对象的Iterator
    public Iterator createrIterator(){
        return new HNBranchIterator(depts);
    }
}

//湖南分公司的部门迭代器
public class HNBranchIterator implements Iterator{
    ArrayList<Dept> depts;
    int position    = 0;
    public HNBranchIterator(ArrayList<Dept> depts){
        this.depts = depts;
    }
    @Override
    public boolean hasNext() {
        if(position>=depts.size() || depts.get(position)==null){
            return false;
        }else{
            return true;
        }
    }

    @Override
    public Object next() {
        Dept dept = depts.get(position);
        position  = position + 1;
        return dept;
    }

}

public class TestNew {

    public static void main(String[] args) {
        Iterator BJ = new BJBranch().createrIterator();
        Iterator SH = new SHBranch().createrIterator();
        Iterator HN = new HNBranch().createrIterator();

        printDeptName(BJ);
        printDeptName(SH);
        //打印新设立的湖南分公司的所有部门
        printDeptName(HN);
    }

    private static void printDeptName(Iterator iterator){
        while(iterator.hasNext()){
            Dept dept = (Dept) iterator.next();
            System.out.println(dept.getName());
        }
    }
}

  依次类推，如果再设立其它省份的分公司一样的可以很快增加上去。但是，如果需要在湖南分公司下再设立下隶属于湖南分公司的长沙分公司呢？此时，整个结构变为一棵树型结构。

二、组合模式使用

  可使用组合模式来表现“整体/部分”层次结构。

//不管是分子机构，还是分支机构下的部门对象都需要继承组件对象
public abstract class Component {
    //不管是父节点还是叶子节点都有该方法
    public abstract String getName();
    public abstract void print();

    //叶子节点可以没有该方法
    public void add(Component component) {
        throw new UnsupportedOperationException();
    }
}

//分支机构，即父节点
public class Branch extends Component{
    ArrayList<Component> components;
    private String name;

    public Branch(String name){
        this.name = name;
        components = new ArrayList<Component>();
    }

    public void add(Component component){
        components.add(component);
    }

    public String getName(){
        return this.name;
    }

    public void print(){
        System.out.println(getName());
        System.out.println("----------------------");
        //打印子节点，这里既可以是叶子节点也可以是父节点。通过这里实现了递归
        for(Component com:components){
            com.print();
        }
    }
}

//机构中的部门对象，即叶子节点
public class Dept extends Component{
    private String name;

    public Dept(String name){
        this.name= name;
    }

    public String getName(){
        return this.name;
    }

    public void print(){
        System.out.println(getName());
        System.out.println("----------------------");
    }
}

//测试代码
public class Test {
    public static void main(String[] args) {
        //下面的为同级
        Component BJBranch = new Branch("北京分公司");
        BJBranch.add(new Dept("北京-财务部"));
        Component SHBranch = new Branch("上海分公司");
        SHBranch.add(new Dept("上海-财务部"));
        Component HNBranch = new Branch("湖南分公司");
        HNBranch.add(new Dept("湖南-财务部"));
        //下面的为子级
        Component CS = new Branch("长沙子公司");
        CS.add(new Dept("长沙-财务部"));
        HNBranch.add(CS);

        Component root = new Branch("集团公司");
        root.add(BJBranch);
        root.add(SHBranch);
        root.add(HNBranch);
        //这里会递归遍历下去，直至所有节点遍历完毕
        root.print();
    }
}

  从上面的代码中可以看出，核心在于父节点与叶子节点都属于Component，才能组合在一起，实现递归打印。

上一篇：设计模式系列之十迭代器模式