vendor/friendsofsymfony/elastica-bundle/Doctrine/ORM/Provider.php line 9

Open in your IDE?
  1. <?php
  3. namespace FOS\ElasticaBundle\Doctrine\ORM;
  5. use Doctrine\ORM\QueryBuilder;
  6. use FOS\ElasticaBundle\Doctrine\AbstractProvider;
  7. use FOS\ElasticaBundle\Exception\InvalidArgumentTypeException;
  9. @trigger_error(sprintf('The %s class is deprecated since version 4.1 and will be removed in 5.0.', Provider::class), E_USER_DEPRECATED);
  11. /**
  12.  * @deprecated since 4.1 will be removed in 5.x. Use PagerProvider instead
  13.  */
  14. class Provider extends AbstractProvider
  15. {
  16.     const ENTITY_ALIAS = 'a';
  18.     /**
  19.      * Disables logging and returns the logger that was previously set.
  20.      *
  21.      * @return mixed
  22.      */
  23.     protected function disableLogging()
  24.     {
  25.         $configuration = $this->managerRegistry
  26.             ->getManagerForClass($this->objectClass)
  27.             ->getConnection()
  28.             ->getConfiguration();
  30.         $logger = $configuration->getSQLLogger();
  31.         $configuration->setSQLLogger(null);
  33.         return $logger;
  34.     }
  36.     /**
  37.      * Reenables the logger with the previously returned logger from disableLogging();.
  38.      *
  39.      * @param mixed $logger
  40.      *
  41.      * @return mixed
  42.      */
  43.     protected function enableLogging($logger)
  44.     {
  45.         $configuration = $this->managerRegistry
  46.             ->getManagerForClass($this->objectClass)
  47.             ->getConnection()
  48.             ->getConfiguration();
  50.         $configuration->setSQLLogger($logger);
  51.     }
  53.     /**
  54.      * {@inheritDoc}
  55.      */
  56.     protected function countObjects($queryBuilder)
  57.     {
  58.         if (!$queryBuilder instanceof QueryBuilder) {
  59.             throw new InvalidArgumentTypeException($queryBuilder, 'Doctrine\ORM\QueryBuilder');
  60.         }
  62.         /* Clone the query builder before altering its field selection and DQL,
  63.          * lest we leave the query builder in a bad state for fetchSlice().
  64.          */
  65.         $qb = clone $queryBuilder;
  66.         $rootAliases = $queryBuilder->getRootAliases();
  68.         return $qb
  69.             ->select($qb->expr()->count($rootAliases[0]))
  70.             // Remove ordering for efficiency; it doesn't affect the count
  71.             ->resetDQLPart('orderBy')
  72.             ->getQuery()
  73.             ->getSingleScalarResult();
  74.     }
  76.     /**
  77.      * This method should remain in sync with SliceFetcher::fetch until it is deprecated and removed.
  78.      *
  79.      * {@inheritDoc}
  80.      */
  81.     protected function fetchSlice($queryBuilder, $limit, $offset)
  82.     {
  83.         if (!$queryBuilder instanceof QueryBuilder) {
  84.             throw new InvalidArgumentTypeException($queryBuilder, 'Doctrine\ORM\QueryBuilder');
  85.         }
  87.         /*
  88.          * An orderBy DQL  part is required to avoid fetching the same row twice.
  89.          * @see http://stackoverflow.com/questions/6314879/does-limit-offset-length-require-order-by-for-pagination
  90.          * @see http://www.postgresql.org/docs/current/static/queries-limit.html
  91.          * @see http://www.sqlite.org/lang_select.html#orderby
  92.          */
  93.         $orderBy = $queryBuilder->getDQLPart('orderBy');
  94.         if (empty($orderBy)) {
  95.             $rootAliases = $queryBuilder->getRootAliases();
  96.             $identifierFieldNames = $this->managerRegistry
  97.                 ->getManagerForClass($this->objectClass)
  98.                 ->getClassMetadata($this->objectClass)
  99.                 ->getIdentifierFieldNames();
  100.             foreach ($identifierFieldNames as $fieldName) {
  101.                 $queryBuilder->addOrderBy($rootAliases[0].'.'.$fieldName);
  102.             }
  103.         }
  105.         return $queryBuilder
  106.             ->setFirstResult($offset)
  107.             ->setMaxResults($limit)
  108.             ->getQuery()
  109.             ->getResult();
  110.     }
  112.     /**
  113.      * {@inheritDoc}
  114.      */
  115.     protected function createQueryBuilder($method, array $arguments = array())
  116.     {
  117.         $repository = $this->managerRegistry
  118.             ->getManagerForClass($this->objectClass)
  119.             ->getRepository($this->objectClass);
  120.         // ORM query builders require an alias argument
  121.         $arguments = [static::ENTITY_ALIAS] + $arguments;
  123.         return call_user_func_array([$repository, $method], $arguments);
  124.     }
  125. }