Mdiyakov\DoctrineSolrBundle\Query\Hydrator\SelectQueryHydrator::hydrate()

Complexity

Conditions	7
Paths	8

Size

Total Lines

74

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

<?php
    public function hydrate($documentsArray)
    {
        if (!is_array($documentsArray)) {

The condition is_array($documentsArray) is always true.

            throw new HydratorException('Result dataset must be an array');
        }

        if (empty($documentsArray)) {
            throw new HydratorException('Result dataset is empty');
        }

        $scoreStatements = [];
        $discriminatorField = $this->schema->getDiscriminatorConfigField();
        $entityPrimaryKeyField = $this->schema->getEntityPrimaryKeyField();
        $primaryKeyValues = [];
        $orderValue = count($documentsArray);
        foreach($documentsArray as $documentRow) {
            if (!array_key_exists($discriminatorField->getDocumentFieldName(), $documentRow)) {
                throw new DiscriminatorFieldException(
                    sprintf('Discriminator field "%s" must be presented in dataset', $discriminatorField->getDocumentFieldName())
                );
            }

            if ($discriminatorField->getValue($this->entityConfig) !== $documentRow[$discriminatorField->getDocumentFieldName()]) {
                throw new DiscriminatorFieldException(
                    sprintf(
                        'Discriminator field value "%s" must be "%s" but "%s" is provided',
                        $discriminatorField->getConfigFieldName(),
                        $discriminatorField->getValue($this->entityConfig),
                        $documentRow[$discriminatorField->getDocumentFieldName()]
                    )
                );
            }

            $primaryKeyValues[] = $documentRow[$entityPrimaryKeyField->getDocumentFieldName()];
            $scoreStatements[] = sprintf(
                'WHEN entity.%s = %s THEN %s',
                $entityPrimaryKeyField->getEntityFieldName(),
                $documentRow[$entityPrimaryKeyField->getDocumentFieldName()],
                $orderValue
            );
            $orderValue = --$orderValue;
        }

        $entities = $this->entityRepository
            ->createQueryBuilder('entity')
            ->addSelect(sprintf('(CASE %s ELSE 0 END) AS HIDDEN score', join(' ' ,$scoreStatements)))
            ->where(
                sprintf('entity.%s in (:ids)', $entityPrimaryKeyField->getEntityFieldName())
            )
            ->setParameter(':ids', $primaryKeyValues)
            ->addOrderBy('score', 'desc')
            ->getQuery()->getResult()
        ;

        if (count($entities) != count($documentsArray)) {
            $entityIds = array_map(
                function($entity) use ($entityPrimaryKeyField)
                {
                    return  $entityPrimaryKeyField->getEntityFieldValue($entity);
                },
                $entities
            );

            $notFoundEntityIds = array_diff($primaryKeyValues, $entityIds);
            throw new HydratorException(
                sprintf(
                    'Entities of "%s" with "%s" primary keys are not found in database',
                    $this->entityConfig['class'],
                    join(', ', $notFoundEntityIds)
                )
            );
        }

        return $entities;
    }
}