以代码处理代替循环查询数据库

公司的产品的主体类型架构是一个树结构，其实说白了就是以一个父级ID来组织而已。

其中有个需求是要把所有主体类型树拿出来，这个简单呀，其实就是一个递归查询就解决了。

表结构如图：

Paste_Image.png

我马上在类里面写了下面5行代码：

public static function getAll1($pid)
{
        $children = self::find()->select('id, pid, name')->where(['pid' => $pid])->asArray()->all();
        foreach ($children as $key => $value) {
            $children[$key]['children'] = self::getAll1($value['id']);
        }
        return $children;
 }

所有树就都拉出来啦。

但是发现当数据库有10几棵树，400条数据的时候，每次查询要耗费2s左右的时间，刚开始为了快我就直接加个redis缓存。对于有强迫症的我不能忍，于是今天有空了就想做一下优化。

优化的方向是用php处理来代替在递归里面循环查询。

于是我又写了下面一种方法：

/**
* 只查一次，再对数据处理
**/
 public static function getAll2()
    {
        $organization = self::find()->select('id, pid, name')->asArray()->all();
        return self::getChildrenByRecursion(0, $organization);
    }
/**
* 根据父ID查询子集
*/
  public static function getChildren($organizations, $pid)
    {
        $res = [];
        foreach ($organizations as $key => $value) {
            if($value['pid'] == $pid) {
                $res[] = $value;
            }
        }
        return $res;
    }
/**
* 递归获取全部
*/
   public static function getChildrenByRecursion($pid, $organization)
   {
        $children = self::getChildren($organization, $pid);
        foreach ($children as $key => $value) {
            $children[$key]['children'] = self::getChildrenByRecursion($value['id'], $organization);
        }
        return $children;
   }

然后就写了个测试

    /**
     * 测试
     */
    public function actionTest()
    {
        $start1 = microtime(true);
        $all1 = EntityType::getAll1(0);
        print_r(['first way used time' => microtime(true)-$start1]);
        $start2 = microtime(true);
        $all2 = EntityType::getAll2();
        print_r(['second way used time' => microtime(true)-$start2]);
        print_r(['res1 equal res2 ?' => $all1 == $all2 ? 'yes' : 'no']);
    }

直接贴效果：

Paste_Image.png Paste_Image.png

哈哈，多次测试，速度基本都是提高了几十倍，第二种方法只用了几十毫秒，连缓存都不需要啦~~

2017-08-09更新
getChildren方法的算法复杂度可以降为O(1)，无需遍历。$organizations参数可以先处理：

 /**
     * 处理获取到的全部机构，将pid作为$key
     * @param $organizations
     */
    public static function dealOrganizations($organizations)
    {
        $newOrganizations = [];
        foreach ($organizations as $organization) {
            $newOrganizations[$organization['pid']][] = $organization;
        }
        return $newOrganizations;
    }

取子集可以修改为：

 /**
     * 获取该父ID下的机构
     * @param array $organization 公众号下的所有机构
     * @param int $pid 父ID
     * @return array
     */
    public static function getChildren($organization, $pid)
    {
        if(!empty($organization[$pid])) {
            ArrayHelper::multisort($organization[$pid], 'sort');
            return $organization[$pid];
        }
        return [];
    }

速度将大大提升！