代码功能更新

2025-05-18 07:42:27 +00:00
parent 7fc1d422a1
commit 4a90c41da2
2083 changed files with 695218 additions and 3 deletions
@@ -0,0 +1,221 @@
+<?php
+
+namespace laytp\library;
+
+/**
+ * 通用树型类 - laytp极速后台框架
+ * @Author:  JunStar
+ * @Version: 1.0.0
+ * @Date:    2022-1-7 21:48:38
+ * @LastModified by:   JunStar
+ * @LastModified time: 2022-1-8 21:45:22
+ */
+class Tree
+{
+    protected static $instance;
+
+    /**
+     * 生成树型结构所需要的2维数组
+     * @var array
+     */
+    public $map = [];
+    public $mapName = 'id';
+    public $pidName = 'pid';
+    public $childName = 'children';
+
+    /**
+     * 初始化
+     * @access public
+     * @return Tree
+     */
+    public static function instance()
+    {
+        if (is_null(self::$instance)) {
+            self::$instance = new static();
+        }
+
+        return self::$instance;
+    }
+
+    /**
+     * 初始化方法
+     *  注意，参数一定要是数组，数据库select()->toArray()；否则调用getRootTree会内存溢出
+     * @param array 2维数组，例如：
+     * array(
+     *      1 => array('id'=>'1','pid'=>0,'name'=>'一级栏目一'),
+     *      2 => array('id'=>'2','pid'=>0,'name'=>'一级栏目二'),
+     *      3 => array('id'=>'3','pid'=>1,'name'=>'二级栏目一'),
+     *      4 => array('id'=>'4','pid'=>1,'name'=>'二级栏目二'),
+     *      5 => array('id'=>'5','pid'=>2,'name'=>'二级栏目三'),
+     *      6 => array('id'=>'6','pid'=>3,'name'=>'三级栏目一'),
+     *      7 => array('id'=>'7','pid'=>3,'name'=>'三级栏目二')
+     *      )
+     * @return $this
+     */
+    public function init($arr = [])
+    {
+        $map = [];
+        //生成以id为key的map
+        foreach ($arr as &$it){
+            $map[$it['id']] = &$it;
+        }
+        $this->map = $map;
+        return $this;
+    }
+
+    /**
+     * 获取所有根树
+     * @return array
+     */
+    public function getRootTrees()
+    {
+        $map = $this->map;
+        $res = [];
+        $rootTree = [];
+        foreach ($map as $id => &$item) {
+            // 获取出每一条数据的父id
+            $pid = &$item[$this->pidName];
+            // 如果在map中没有设置过当前$item的pid索引, 说明$item是根节点
+            if(!isset($map[$pid])){
+                //将根节点的item的引用保存到$res中
+                $res[] = &$item;
+            }else{
+                // 如果在map中有设置过当前$item的pid索引, 则将当前item加入到他父亲的叶子节点中
+                // 此处关键需要理解内存地址引用，修改了$pItem，其实就是修改了$map[$pid]的值
+                $pItem = &$map[$pid];
+                $pItem[$this->childName][] = &$item;
+            }
+        }
+
+        // 循环res，将pid不为0的过滤掉，剩下的就是所有的根树
+        if($res){
+            foreach($res as $k=>$value){
+                if($value[$this->pidName] == 0){
+                    $rootTree[] = $value;
+                }
+            }
+        }
+
+        return $rootTree;
+    }
+
+    /**
+     * 获取所有树
+     */
+    public function getTrees(){
+        $map = $this->map;
+        $res = [];
+        foreach ($map as $id => &$item) {
+            // 获取出每一条数据的父id
+            $pid = &$item[$this->pidName];
+            // 如果在map中没有设置过当前$item的pid索引, 说明$item是根节点
+            if(!isset($map[$pid])){
+                //将根节点的item的引用保存到$res中
+                $res[] = &$item;
+            }else{
+                // 如果在map中有设置过当前$item的pid索引, 则将当前item加入到他父亲的叶子节点中
+                // 此处关键需要理解内存地址引用，修改了$pItem，其实就是修改了$map[$pid]的值
+                $pItem = &$map[$pid];
+                $pItem[$this->childName][] = &$item;
+            }
+        }
+
+        return $res;
+    }
+
+    /**
+     * 获取某棵子树
+     *  思路:
+     *      1.先获取当前节点的所有父级节点
+     *      2.组合一个根树的map，每个key都是id值，getRootTree是自增的
+     *      3.使用所有父级节点，直接在根树的map中索引得到子树
+     */
+
+    /**
+     * 获取某ID的所有子级ID
+     * @param $ids string|array 要查询子级的ID
+     * @param bool $withSelf 返回的结果是否包含$ids
+     * @return array
+     */
+    public function getChildIds($ids, $withSelf=true){
+        $map = $this->map;
+        $res = [];//最终结果
+        if(!is_array($ids)){
+            $ids = explode(',', $ids);
+            $sourceIds = $ids;
+        }else{
+            $sourceIds = $ids;
+        }
+
+        $state = true;
+        while($state){
+            $otherIds = [];
+            foreach ($ids as $id) {
+                foreach ($map as $key => $value) {
+                    if($value[$this->pidName] == $id){
+                        $res[] = $value[$this->mapName];//找到我的下级立即添加到最终结果中
+                        $otherIds[] = $value[$this->mapName];//将我的下级id保存起来用来下轮循环他的下级
+                    }
+                }
+            }
+
+            if(!$otherIds){
+                $state = false;
+            }else{
+                $ids = $otherIds;//foreach中找到的我的下级集合,用来下次循环
+            }
+        }
+
+        if($withSelf){
+            foreach($sourceIds as $sId){
+                $res[] = intval($sId);
+            }
+        }
+
+        return $res;
+    }
+
+    /**
+     * 获取某ID的所有父级ID
+     * @param $ids string|array 要查询父级的ID
+     * @param bool $withSelf 返回的结果是否包含$id
+     * @return array
+     */
+    public function getParentIds($ids, $withSelf=true){
+        $map = $this->map;
+        $res = [];//最终结果
+        if(!is_array($ids)){
+            $ids = explode(',', $ids);
+            $sourceIds = $ids;
+        }else{
+            $sourceIds = $ids;
+        }
+
+        $state = true;
+        while($state){
+            $otherIds = [];
+            foreach ($ids as $id) {
+                foreach ($map as $key => $value) {
+                    if($value[$this->mapName] == $map[$id][$this->pidName]){
+                        $res[] = $value[$this->mapName];//找到我的上级立即添加到最终结果中
+                        $otherIds[] = $value[$this->mapName];//将我的上级id保存起来用来下次轮循环他的下级
+                    }
+                }
+            }
+
+            if(!$otherIds){
+                $state = false;
+            }else{
+                $ids = $otherIds;//foreach中找到的我的上级集合,用来下次循环
+            }
+        }
+
+        if($withSelf){
+            foreach($sourceIds as $sId){
+                $res[] = intval($sId);
+            }
+        }
+
+        return $res;
+    }
+}