NoneHydrator.php

<?php

namespace AppBundle\Component\Doctrine\ORM\Hydrators;


use Doctrine\ORM\Internal\Hydration\AbstractHydrator;

class NoneHydrator extends AbstractHydrator
{
    const NAME = 'NoneHydrator';

    protected $fieldNamesCache = [];

    protected function hydrateAllData()
    {
        $result = [];

        foreach ($this->_stmt->fetchAll(\PDO::FETCH_ASSOC) as $row) {
            $this->hydrateRowData($row, $result);
        }

        unset($this->fieldNamesCache);

        return $result;
    }

    protected function hydrateRowData(array $sqlResult, array &$result)
    {
        $data = [];
        foreach ($sqlResult as $key => $value) {
            if (($fieldName = $this->getFieldName($key)) === null) {
                continue;
            }

            $data[$fieldName] = $value;
        }

        $result[] = $data;
    }

    /**
     * @param $key
     * @return null
     */
    protected function getFieldName($key)
    {
        if (isset($this->fieldNamesCache[$key])) {
            return $this->fieldNamesCache[$key];
        }

        switch (true) {
            case (isset($this->_rsm->scalarMappings[$key])):
                return $this->fieldNamesCache[$key] = $this->_rsm->scalarMappings[$key];
            case (isset($this->_rsm->fieldMappings[$key])):
                return $this->fieldNamesCache[$key] = $this->_rsm->fieldMappings[$key];
            case (isset($this->_rsm->newObjectMappings[$key])):
                return $this->fieldNamesCache[$key] = $this->_rsm->scalarMappings[$key];
            case (isset($this->_rsm->metaMappings[$key])):
                return $this->fieldNamesCache[$key] = $this->_rsm->metaMappings[$key];
        }

        return null;
    }

}