Nimble\ElasticBundle\Doctrine\ORM\Populator\DoctrineORMPopulationFetcher

Complexity

Total Complexity

6

Size/Duplication

Total Lines	76
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	8	1
A	createQueryBuilder()	0	4	1
A	getEntityCount()	0	8	1
A	fetchEntities()	0	20	3

<?php

namespace Nimble\ElasticBundle\Doctrine\ORM\Populator;

use Doctrine\ORM\EntityManager;
use Doctrine\ORM\EntityRepository;
use Doctrine\ORM\Mapping\ClassMetadata;
use Doctrine\ORM\QueryBuilder;
use Nimble\ElasticBundle\Populator\PopulationFetcherInterface;

class DoctrineORMPopulationFetcher implements PopulationFetcherInterface
{
    const ROOT_ALIAS = 'e';

    /**
     * @var EntityRepository
     */
    protected $entityRepository;

    /**
     * @var ClassMetadata
     */
    protected $entityClassMetadata;

    /**
     * @var EntityManager
     */
    protected $entityManager;

    /**
     * @param EntityManager $entityManager
     * @param string $entityName
     */
    public function __construct(EntityManager $entityManager, $entityName)

You have injected the EntityManager via parameter $entityManager. This is generally not recommended as it might get closed and become unusable. Instead, it is recommended to inject the ManagerRegistry and retrieve the EntityManager via getManager() each time you need it.

    {
        $this->entityManager = $entityManager;
        $this->entityRepository = $entityManager->getRepository($entityName);
        $this->entityClassMetadata = $entityManager->getClassMetadata(
            $this->entityRepository->getClassName()
        );
    }

    /**
     * @return QueryBuilder
     */
    protected function createQueryBuilder()
    {
        return $this->entityRepository->createQueryBuilder(self::ROOT_ALIAS);
    }

    /**
     * {@inheritdoc}
     */
    public function getEntityCount()
    {
        $qb = $this->createQueryBuilder();

        $qb->select($qb->expr()->count(self::ROOT_ALIAS));

        return $qb->getQuery()->getSingleScalarResult();
    }

    /**
     * {@inheritdoc}
     */
    public function fetchEntities($offset = 0, $limit = null)
    {
        $this->entityManager->clear(); /* TODO: Check if this gives us any value. */

        $qb = $this->createQueryBuilder();

        $qb->select(self::ROOT_ALIAS);

        foreach ($this->entityClassMetadata->getIdentifierFieldNames() as $fieldName) {
            $qb->addOrderBy(sprintf('%s.%s', self::ROOT_ALIAS, $fieldName));
        }

        $qb->setFirstResult($offset);

        if (null !== $limit) {
            $qb->setMaxResults($limit);
        }

        return $qb->getQuery()->getResult();
    }
}