设计模式（十三）：组合模式

转载请注明作者和出处：https://www.jianshu.com/p/aa6f7b3edb00

运行平台： Windows

php版本： php7.0

作者简介： 一个本该成为游戏职业选手却被编程耽误的程序员

---

相信很多人都接触过下面这个例子

<?php
$array = array(
    array('id' => 1, 'pid' => 0, 'n' => '河北省'),
    array('id' => 2, 'pid' => 0, 'n' => '北京市'),
    array('id' => 3, 'pid' => 1, 'n' => '邯郸市'),
    array('id' => 4, 'pid' => 2, 'n' => '朝阳区'),
    array('id' => 5, 'pid' => 2, 'n' => '通州区'),
    array('id' => 6, 'pid' => 4, 'n' => '望京'),
    array('id' => 7, 'pid' => 4, 'n' => '酒仙桥'),
    array('id' => 8, 'pid' => 3, 'n' => '永年区'),
    array('id' => 9, 'pid' => 1, 'n' => '武安市'),
    array('id' => 10, 'pid' => 8, 'n' => '永年区镇'),
    array('id' => 11, 'pid' => 0, 'n' => '上海市')
);

/** 所有的分类
 * @parem $array 数组
 * @parem $pid ，最高级别,默认为0，输出从pid 级别的数据
 * @parem $level 层级，默认0
 * */
function getTree($array, $pid =0, $level = 0){

    $f_name=__FUNCTION__; // 定义当前函数名
    //声明静态数组,避免递归调用时,多次声明导致数组覆盖
    static $list = [];
    foreach ($array as $key => $value){
        //第一次遍历,找到父节点为根节点的节点 也就是pid=0的节点
        if ($value['pid'] == $pid){
            //父节点为根节点的节点,级别为0，也就是第一级
            $flg = str_repeat('|--',$level);
            // 更新 名称值
            $value['n'] = $flg.$value['n'];
            // 输出 名称
            echo $value['n']."<br/>";
            //把数组放到list中
            $list[] = $value;
            //把这个节点从数组中移除,减少后续递归消耗
            unset($array[$key]);
            //开始递归,查找父ID为该节点ID的节点,级别则为原级别+1
            $f_name($array, $value['id'], $level+1);
        }else{
            // 删除数组，减少递归
            unset($array[$key]);}
    }
    return $list;
}
// 调用
$list=getTree($array);

这是一个无限极分类的例子，可是这是一个函数的做法，我们把它换成对象的形式，也就是组合模式。

组合模式 （Composite Pattern）

将对象组合成树形结构以表示“部分整体”的层次结构。组合模式使得用户对单个对象和组合对象的使用具有一致性。组合模式也叫合成模式，有时候又叫做部分-整体模式。

abstract class MenuComponent
{
    public $name;
    public abstract function getName();
    public abstract function add(MenuComponent $menu);

    public abstract function remove(MenuComponent $menu);

    public abstract function getChild($i);

    public abstract function show();
}

class MenuItem extends MenuComponent
{
    public function __construct($name)
    {
        $this->name = $name;
    }
    public function getName(){
        return $this->name;
    }
    public function add(MenuComponent $menu){
        return false;
    }
    public function remove(MenuComponent $menu){
        return false;
    }

    public function getChild($i){
        return null;
    }
    public function show(){
        echo " |-".$this->getName()."\n";
    }
}

class Menu extends MenuComponent
{
    public $menuComponents = array();
    public function __construct($name)
    {
        $this->name = $name;
    }

    public function getName()
    {
        return $this->name;
    }

    public function add(MenuComponent $menu)
    {
        $this->menuComponents[] = $menu;
    }

    public function remove(MenuComponent $menu)
    {
        $key = array_search($menu, $this->menuComponents);
        if($key !== false) unset($this->menuComponents[$key]);
    }

    public function getChild($i)
    {
        if(isset($this->menuComponents[$i])) return $this->menuComponents[$i];
        return null;
    }

    public function show()
    {
        echo ":" . $this->getName() . "\n";
        foreach($this->menuComponents as $v){
            $v->show();
        }
    }
}


$menu1 = new Menu('北京市');
$menuitem1 = new MenuItem('邯郸市');
$menuitem2 = new MenuItem('朝阳区');
$menuitem3 = new MenuItem('通州区');
$menuitem4 = new MenuItem('望京');
$menu1->add($menuitem1);
$menu1->add($menuitem2);
$menu1->add($menuitem3);
$menu1->add($menuitem4);
$menu1->show();

执行结果

:北京市
  |-邯郸市
  |-朝阳区
  |-通州区
  |-望京