AbstractHydrator.php
9 months ago
ArrayHydrator.php
9 months ago
HydrationException.php
9 months ago
IterableResult.php
9 months ago
ObjectHydrator.php
9 months ago
ScalarColumnHydrator.php
9 months ago
ScalarHydrator.php
9 months ago
SimpleObjectHydrator.php
9 months ago
SingleScalarHydrator.php
9 months ago
index.php
9 months ago
ArrayHydrator.php
187 lines
| 1 | <?php |
| 2 | declare (strict_types=1); |
| 3 | namespace MailPoetVendor\Doctrine\ORM\Internal\Hydration; |
| 4 | if (!defined('ABSPATH')) exit; |
| 5 | use MailPoetVendor\Doctrine\ORM\Mapping\ClassMetadata; |
| 6 | use function count; |
| 7 | use function end; |
| 8 | use function is_array; |
| 9 | use function key; |
| 10 | use function reset; |
| 11 | class ArrayHydrator extends AbstractHydrator |
| 12 | { |
| 13 | private $rootAliases = []; |
| 14 | private $isSimpleQuery = \false; |
| 15 | private $identifierMap = []; |
| 16 | private $resultPointers = []; |
| 17 | private $idTemplate = []; |
| 18 | private $resultCounter = 0; |
| 19 | protected function prepare() |
| 20 | { |
| 21 | $this->isSimpleQuery = count($this->resultSetMapping()->aliasMap) <= 1; |
| 22 | foreach ($this->resultSetMapping()->aliasMap as $dqlAlias => $className) { |
| 23 | $this->identifierMap[$dqlAlias] = []; |
| 24 | $this->resultPointers[$dqlAlias] = []; |
| 25 | $this->idTemplate[$dqlAlias] = ''; |
| 26 | } |
| 27 | } |
| 28 | protected function hydrateAllData() |
| 29 | { |
| 30 | $result = []; |
| 31 | while ($data = $this->statement()->fetchAssociative()) { |
| 32 | $this->hydrateRowData($data, $result); |
| 33 | } |
| 34 | return $result; |
| 35 | } |
| 36 | protected function hydrateRowData(array $row, array &$result) |
| 37 | { |
| 38 | // 1) Initialize |
| 39 | $id = $this->idTemplate; |
| 40 | // initialize the id-memory |
| 41 | $nonemptyComponents = []; |
| 42 | $rowData = $this->gatherRowData($row, $id, $nonemptyComponents); |
| 43 | // 2) Now hydrate the data found in the current row. |
| 44 | foreach ($rowData['data'] as $dqlAlias => $data) { |
| 45 | $index = \false; |
| 46 | if (isset($this->resultSetMapping()->parentAliasMap[$dqlAlias])) { |
| 47 | // It's a joined result |
| 48 | $parent = $this->resultSetMapping()->parentAliasMap[$dqlAlias]; |
| 49 | $path = $parent . '.' . $dqlAlias; |
| 50 | // missing parent data, skipping as RIGHT JOIN hydration is not supported. |
| 51 | if (!isset($nonemptyComponents[$parent])) { |
| 52 | continue; |
| 53 | } |
| 54 | // Get a reference to the right element in the result tree. |
| 55 | // This element will get the associated element attached. |
| 56 | if ($this->resultSetMapping()->isMixed && isset($this->rootAliases[$parent])) { |
| 57 | $first = reset($this->resultPointers); |
| 58 | // TODO: Exception if $key === null ? |
| 59 | $baseElement =& $this->resultPointers[$parent][key($first)]; |
| 60 | } elseif (isset($this->resultPointers[$parent])) { |
| 61 | $baseElement =& $this->resultPointers[$parent]; |
| 62 | } else { |
| 63 | unset($this->resultPointers[$dqlAlias]); |
| 64 | // Ticket #1228 |
| 65 | continue; |
| 66 | } |
| 67 | $relationAlias = $this->resultSetMapping()->relationMap[$dqlAlias]; |
| 68 | $parentClass = $this->_metadataCache[$this->resultSetMapping()->aliasMap[$parent]]; |
| 69 | $relation = $parentClass->associationMappings[$relationAlias]; |
| 70 | // Check the type of the relation (many or single-valued) |
| 71 | if (!($relation['type'] & ClassMetadata::TO_ONE)) { |
| 72 | $oneToOne = \false; |
| 73 | if (!isset($baseElement[$relationAlias])) { |
| 74 | $baseElement[$relationAlias] = []; |
| 75 | } |
| 76 | if (isset($nonemptyComponents[$dqlAlias])) { |
| 77 | $indexExists = isset($this->identifierMap[$path][$id[$parent]][$id[$dqlAlias]]); |
| 78 | $index = $indexExists ? $this->identifierMap[$path][$id[$parent]][$id[$dqlAlias]] : \false; |
| 79 | $indexIsValid = $index !== \false ? isset($baseElement[$relationAlias][$index]) : \false; |
| 80 | if (!$indexExists || !$indexIsValid) { |
| 81 | $element = $data; |
| 82 | if (isset($this->resultSetMapping()->indexByMap[$dqlAlias])) { |
| 83 | $baseElement[$relationAlias][$row[$this->resultSetMapping()->indexByMap[$dqlAlias]]] = $element; |
| 84 | } else { |
| 85 | $baseElement[$relationAlias][] = $element; |
| 86 | } |
| 87 | end($baseElement[$relationAlias]); |
| 88 | $this->identifierMap[$path][$id[$parent]][$id[$dqlAlias]] = key($baseElement[$relationAlias]); |
| 89 | } |
| 90 | } |
| 91 | } else { |
| 92 | $oneToOne = \true; |
| 93 | if (!isset($nonemptyComponents[$dqlAlias]) && !isset($baseElement[$relationAlias])) { |
| 94 | $baseElement[$relationAlias] = null; |
| 95 | } elseif (!isset($baseElement[$relationAlias])) { |
| 96 | $baseElement[$relationAlias] = $data; |
| 97 | } |
| 98 | } |
| 99 | $coll =& $baseElement[$relationAlias]; |
| 100 | if (is_array($coll)) { |
| 101 | $this->updateResultPointer($coll, $index, $dqlAlias, $oneToOne); |
| 102 | } |
| 103 | } else { |
| 104 | // It's a root result element |
| 105 | $this->rootAliases[$dqlAlias] = \true; |
| 106 | // Mark as root |
| 107 | $entityKey = $this->resultSetMapping()->entityMappings[$dqlAlias] ?: 0; |
| 108 | // if this row has a NULL value for the root result id then make it a null result. |
| 109 | if (!isset($nonemptyComponents[$dqlAlias])) { |
| 110 | $result[] = $this->resultSetMapping()->isMixed ? [$entityKey => null] : null; |
| 111 | $resultKey = $this->resultCounter; |
| 112 | ++$this->resultCounter; |
| 113 | continue; |
| 114 | } |
| 115 | // Check for an existing element |
| 116 | if ($this->isSimpleQuery || !isset($this->identifierMap[$dqlAlias][$id[$dqlAlias]])) { |
| 117 | $element = $this->resultSetMapping()->isMixed ? [$entityKey => $data] : $data; |
| 118 | if (isset($this->resultSetMapping()->indexByMap[$dqlAlias])) { |
| 119 | $resultKey = $row[$this->resultSetMapping()->indexByMap[$dqlAlias]]; |
| 120 | $result[$resultKey] = $element; |
| 121 | } else { |
| 122 | $resultKey = $this->resultCounter; |
| 123 | $result[] = $element; |
| 124 | ++$this->resultCounter; |
| 125 | } |
| 126 | $this->identifierMap[$dqlAlias][$id[$dqlAlias]] = $resultKey; |
| 127 | } else { |
| 128 | $index = $this->identifierMap[$dqlAlias][$id[$dqlAlias]]; |
| 129 | $resultKey = $index; |
| 130 | } |
| 131 | $this->updateResultPointer($result, $index, $dqlAlias, \false); |
| 132 | } |
| 133 | } |
| 134 | if (!isset($resultKey)) { |
| 135 | $this->resultCounter++; |
| 136 | } |
| 137 | // Append scalar values to mixed result sets |
| 138 | if (isset($rowData['scalars'])) { |
| 139 | if (!isset($resultKey)) { |
| 140 | // this only ever happens when no object is fetched (scalar result only) |
| 141 | $resultKey = isset($this->resultSetMapping()->indexByMap['scalars']) ? $row[$this->resultSetMapping()->indexByMap['scalars']] : $this->resultCounter - 1; |
| 142 | } |
| 143 | foreach ($rowData['scalars'] as $name => $value) { |
| 144 | $result[$resultKey][$name] = $value; |
| 145 | } |
| 146 | } |
| 147 | // Append new object to mixed result sets |
| 148 | if (isset($rowData['newObjects'])) { |
| 149 | if (!isset($resultKey)) { |
| 150 | $resultKey = $this->resultCounter - 1; |
| 151 | } |
| 152 | $scalarCount = isset($rowData['scalars']) ? count($rowData['scalars']) : 0; |
| 153 | foreach ($rowData['newObjects'] as $objIndex => $newObject) { |
| 154 | $class = $newObject['class']; |
| 155 | $args = $newObject['args']; |
| 156 | $obj = $class->newInstanceArgs($args); |
| 157 | if (count($args) === $scalarCount || $scalarCount === 0 && count($rowData['newObjects']) === 1) { |
| 158 | $result[$resultKey] = $obj; |
| 159 | continue; |
| 160 | } |
| 161 | $result[$resultKey][$objIndex] = $obj; |
| 162 | } |
| 163 | } |
| 164 | } |
| 165 | private function updateResultPointer(?array &$coll, $index, string $dqlAlias, bool $oneToOne) : void |
| 166 | { |
| 167 | if ($coll === null) { |
| 168 | unset($this->resultPointers[$dqlAlias]); |
| 169 | // Ticket #1228 |
| 170 | return; |
| 171 | } |
| 172 | if ($oneToOne) { |
| 173 | $this->resultPointers[$dqlAlias] =& $coll; |
| 174 | return; |
| 175 | } |
| 176 | if ($index !== \false) { |
| 177 | $this->resultPointers[$dqlAlias] =& $coll[$index]; |
| 178 | return; |
| 179 | } |
| 180 | if (!$coll) { |
| 181 | return; |
| 182 | } |
| 183 | end($coll); |
| 184 | $this->resultPointers[$dqlAlias] =& $coll[key($coll)]; |
| 185 | } |
| 186 | } |
| 187 |