Исправлены ошибки, улучшена производительность

Smoren · Smoren · commit fe933583cf68 · 2020-09-09T18:04:33.000+03:00
diff --git a/src/Base/QueryRelationManagerBase.php b/src/Base/QueryRelationManagerBase.php
@@ -168,36 +168,35 @@ public function all($db = null): array
 
         foreach($rows as $row) {
             $this->tableCollection->each(function(Table $table) use (&$map, &$row, &$bufMap) {
-                try {
-                    [$item, $pkValue, $alias, $aliasTo, $fkValue, $containerFieldAlias, $type]
-                        = $table->getDataFromRow($row, $this->joinConditionCollection);
+                if(!$table->issetDataInRow($row)) {
+                    return;
+                }
 
-                    if(!isset($map[$alias][$pkValue])) {
-                        $map[$alias][$pkValue] = &$item;
-                    }
+                [$item, $pkValue, $alias, $aliasTo, $fkValue, $containerFieldAlias, $type]
+                    = $table->getDataFromRow($row, $this->joinConditionCollection);
 
-                    if($aliasTo !== null) {
-                        $bufMapKey = implode('-', [$aliasTo, $fkValue, $containerFieldAlias, $pkValue]);
-                        switch($type) {
-                            case JoinCondition::TYPE_SINGLE:
-                                if(!isset($bufMap[$bufMapKey])) {
-                                    $map[$aliasTo][$fkValue][$containerFieldAlias] = &$item;
-                                    $bufMap[$bufMapKey] = 1;
-                                }
-                                break;
-                            case JoinCondition::TYPE_MULTIPLE:
-                                if(!isset($bufMap[$bufMapKey])) {
-                                    $map[$aliasTo][$fkValue][$containerFieldAlias][] = &$item;
-                                    $bufMap[$bufMapKey] = 1;
-                                }
-                                break;
-                            default:
-                                throw new QueryRelationManagerException("unknown condition type '{$type}'");
-                        }
-                    }
+                if(!isset($map[$alias][$pkValue])) {
+                    $map[$alias][$pkValue] = &$item;
+                }
 
-                } catch(QueryRelationManagerException $e) {
-                    // пропускаем
+                if($aliasTo !== null) {
+                    $bufMapKey = implode('-', [$aliasTo, $fkValue, $containerFieldAlias, $pkValue]);
+                    switch($type) {
+                        case JoinCondition::TYPE_SINGLE:
+                            if(!isset($bufMap[$bufMapKey])) {
+                                $map[$aliasTo][$fkValue][$containerFieldAlias] = &$item;
+                                $bufMap[$bufMapKey] = 1;
+                            }
+                            break;
+                        case JoinCondition::TYPE_MULTIPLE:
+                            if(!isset($bufMap[$bufMapKey])) {
+                                $map[$aliasTo][$fkValue][$containerFieldAlias][] = &$item;
+                                $bufMap[$bufMapKey] = 1;
+                            }
+                            break;
+                        default:
+                            throw new QueryRelationManagerException("unknown condition type '{$type}'");
+                    }
                 }
             });
         }
@@ -219,6 +218,10 @@ public function all($db = null): array
      */
     public function prepare(): QueryWrapperInterface
     {
+        $this->tableCollection->each(function(Table $table) {
+            $table->setPkFieldChain($this->tableCollection->getPkFieldChain($table->alias, $this->joinConditionCollection));
+        });
+
         $this->query = $this->createQuery();
 
         $arSelect = [];
diff --git a/src/Base/Structs/JoinConditionCollection.php b/src/Base/Structs/JoinConditionCollection.php
@@ -48,17 +48,27 @@ public function add(JoinCondition $condition): self
         return $this;
     }
 
+    /**
+     * Проверка наличия объекта условия присоединения таблицы по ее псевдониму в запросе
+     * @param string $joinAs псевдоним присоединяемой таблицы
+     * @return bool
+     */
+    public function issetByJoinAs(string $joinAs): bool
+    {
+        if(!isset($this->mapByJoinAs[$joinAs])) {
+            return false;
+        }
+
+        return true;
+    }
+
     /**
      * Получение объекта условия присоединения таблицы по ее псевдониму в запросе
      * @param string $joinAs псевдоним присоединяемой таблицы
      * @return JoinCondition
-     * @throws QueryRelationManagerException
      */
     public function byJoinAs(string $joinAs): JoinCondition
     {
-        if(!isset($this->mapByJoinAs[$joinAs])) {
-            throw new QueryRelationManagerException("no condition found by table alias '{$joinAs}'");
-        }
         return $this->mapByJoinAs[$joinAs];
     }
 
diff --git a/src/Base/Structs/Table.php b/src/Base/Structs/Table.php
@@ -55,6 +55,11 @@ class Table
      */
     protected $pkFieldMapReverse = [];
 
+    /**
+     * @var array цепочка полей первичных ключей подключаемых таблиц до данной
+     */
+    protected $pkFieldChain = [];
+
     /**
      * Table constructor.
      * @param string $className ORM-класс, представляющий таблицу
@@ -118,6 +123,21 @@ public function stringifyPrimaryKey(): string
         return implode('-', $this->primaryKey);
     }
 
+    /**
+     * Проверка наличия данных из таблицы в кортеже, представляющем из себя строку из результата запроса к БД
+     * @param array $row строка из результата запроса SELECT
+     * @return bool
+     */
+    public function issetDataInRow(array &$row): bool
+    {
+        foreach($this->pkFieldMapReverse as $prefixedKey => $key) {
+            if($row[$prefixedKey] !== null) {
+                return true;
+            }
+        }
+        return false;
+    }
+
     /**
      * Получение данных из кортежа, представляющего из себя строку из результата запроса к БД
      * @param array $row строка из результата запроса SELECT
@@ -137,10 +157,8 @@ public function getDataFromRow(array $row, JoinConditionCollection $conditionCol
     {
         $item = [];
 
-        foreach($row as $key => $val) {
-            if(isset($this->fieldMapReverse[$key])) {
-                $item[$this->fieldMapReverse[$key]] = $val;
-            }
+        foreach($this->fieldMapReverse as $fieldPrefixed => $field) {
+            $item[$field] = $row[$fieldPrefixed];
         }
 
         /** @var JoinCondition $cond */
@@ -159,13 +177,13 @@ public function getDataFromRow(array $row, JoinConditionCollection $conditionCol
 
         $primaryKeyValue = $this->stringifyPrimaryKeyValue($row);
 
-        try {
+        if($conditionCollection->issetByJoinAs($this->alias)) {
             $cond = $conditionCollection->byJoinAs($this->alias);
             $joinTo = $cond->joinTo;
             $aliasTo = $joinTo->alias;
             $foreignKeyValue = $joinTo->stringifyPrimaryKeyValue($row);
             $type = $cond->type;
-        } catch(QueryRelationManagerException $e) {
+        } else {
             $aliasTo = null;
             $foreignKeyValue = null;
             $containerFieldAlias = null;
@@ -189,21 +207,28 @@ public function getPrimaryKeyForSelect(): array
         return $result;
     }
 
+    /**
+     * Установка цепочки полей первичных ключей присоединяемых таблиц до данной
+     * @param array $pkFieldChain
+     * @return $this
+     */
+    public function setPkFieldChain(array $pkFieldChain): self
+    {
+        $this->pkFieldChain = $pkFieldChain;
+        return $this;
+    }
+
     /**
      * Получение значений полей первичного ключа таблицы в виде строки через дефис
      * @param array $row строка из результата запроса SELECT
      * @return string
-     * @throws QueryRelationManagerException
      */
     protected function stringifyPrimaryKeyValue(array $row): string
     {
         $primaryKeyValues = [];
 
-        foreach($this->pkFieldMapReverse as $fieldPrefixed => $field) {
-            if(!isset($row[$fieldPrefixed])) {
-                throw new QueryRelationManagerException("no primary key field '{$field}' found in row");
-            }
-            $primaryKeyValues[] = $row[$fieldPrefixed];
+        foreach($this->pkFieldChain as $field) {
+            $primaryKeyValues[] = $row[$field];
         }
 
         return implode('-', $primaryKeyValues);
diff --git a/src/Base/Structs/TableCollection.php b/src/Base/Structs/TableCollection.php
@@ -79,6 +79,52 @@ public function each(callable $callback): self
         return $this;
     }
 
+    /**
+     * Получение цепочки первичных ключей присоединяемых таблиц до данной
+     * @param string $tableAlias
+     * @param JoinConditionCollection $joinConditions
+     * @return array
+     * @throws QueryRelationManagerException
+     */
+    public function getPkFieldChain(string $tableAlias, JoinConditionCollection $joinConditions): array
+    {
+        $tableAliasChain = $this->getTableAliasChain($tableAlias, $joinConditions);
+        $result = [];
+
+        foreach($tableAliasChain as $alias) {
+            $table = $this->byAlias($alias);
+
+            foreach($table->primaryKey as $field) {
+                $result[] = "{$alias}_{$field}";
+            }
+        }
+
+        return $result;
+    }
+
+    /**
+     * Получение цепочки псеводнимов присоединяемых таблиц до данной
+     * @param string $tableAlias
+     * @param JoinConditionCollection $joinConditions
+     * @return array
+     */
+    public function getTableAliasChain(string $tableAlias, JoinConditionCollection $joinConditions): array
+    {
+        $result = [];
+
+        while(true) {
+            $result[] = $tableAlias;
+
+            if(!$joinConditions->issetByJoinAs($tableAlias)) {
+                break;
+            }
+
+            $tableAlias = $joinConditions->byJoinAs($tableAlias)->joinTo->alias;
+        }
+
+        return array_reverse($result);
+    }
+
     /**
      * Добавление объекта таблицы в карту
      * @param string $mapName имя члена класса-карты
