菜单列表返回树形结构

Author: windhoney

components/Helper.php

@@ -274,4 +274,35 @@ public static function get_shops()
 
         return $shop_id;
     }
+    
+    /**
+     * 树形格式化数据
+     *
+     * @param array  $list  数据
+     * @param string $pk    主
+     * @param string $pid   父
+     * @param string $child 子集变量名
+     * @param int    $root  根节点
+     *
+     * @return array
+     */
+    public static function makeTree($list, $pk = 'id', $pid = 'pid', $child = 'child', $root = 0)
+    {
+        $tree = array();
+        $packData = array();
+        foreach ($list as $data) {
+            $packData[$data[$pk]] = $data;
+        }
+        foreach ($packData as $key => $val) {
+            if ($val[$pid] == $root) {
+                //代表根节点
+                $tree[] =& $packData[$key];
+            } else {
+                //找到其父类
+                $packData[$val[$pid]][$child][] =& $packData[$key];
+            }
+        }
+        
+        return $tree;
+    }
 }

controllers/MenuController.php

@@ -57,7 +57,11 @@ public function actionUserMenu()
     public function actionIndex()
     {
         $res = $this->menu_model->getMenuSource();
-        
+        $is_tree = Yii::$app->request->get('is_tree');
+        if ($is_tree) {
+            return Helper::makeTree($res, 'name', 'parent_name', 'child', null);
+        }
+    
         return $res;
     }