vendor/symfony/doctrine-bridge/Form/ChoiceList/ORMQueryBuilderLoader.php line 45

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of the Symfony package.
  5.  *
  6.  * (c) Fabien Potencier <fabien@symfony.com>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. namespace Symfony\Bridge\Doctrine\Form\ChoiceList;
  14. use Doctrine\DBAL\Connection;
  15. use Doctrine\ORM\QueryBuilder;
  17. /**
  18.  * Loads entities using a {@link QueryBuilder} instance.
  19.  *
  20.  * @author Benjamin Eberlei <kontakt@beberlei.de>
  21.  * @author Bernhard Schussek <bschussek@gmail.com>
  22.  */
  23. class ORMQueryBuilderLoader implements EntityLoaderInterface
  24. {
  25.     /**
  26.      * Contains the query builder that builds the query for fetching the
  27.      * entities.
  28.      *
  29.      * This property should only be accessed through queryBuilder.
  30.      *
  31.      * @var QueryBuilder
  32.      */
  33.     private $queryBuilder;
  35.     public function __construct(QueryBuilder $queryBuilder)
  36.     {
  37.         $this->queryBuilder = $queryBuilder;
  38.     }
  40.     /**
  41.      * {@inheritdoc}
  42.      */
  43.     public function getEntities()
  44.     {
  45.         return $this->queryBuilder->getQuery()->execute();
  46.     }
  48.     /**
  49.      * {@inheritdoc}
  50.      */
  51.     public function getEntitiesByIds(string $identifier, array $values)
  52.     {
  53.         if (null !== $this->queryBuilder->getMaxResults() || null !== $this->queryBuilder->getFirstResult()) {
  54.             // an offset or a limit would apply on results including the where clause with submitted id values
  55.             // that could make invalid choices valid
  56.             $choices = [];
  57.             $metadata = $this->queryBuilder->getEntityManager()->getClassMetadata(current($this->queryBuilder->getRootEntities()));
  59.             foreach ($this->getEntities() as $entity) {
  60.                 if (\in_array((string) current($metadata->getIdentifierValues($entity)), $values, true)) {
  61.                     $choices[] = $entity;
  62.                 }
  63.             }
  65.             return $choices;
  66.         }
  68.         $qb = clone $this->queryBuilder;
  69.         $alias = current($qb->getRootAliases());
  70.         $parameter = 'ORMQueryBuilderLoader_getEntitiesByIds_'.$identifier;
  71.         $parameter = str_replace('.', '_', $parameter);
  72.         $where = $qb->expr()->in($alias.'.'.$identifier, ':'.$parameter);
  74.         // Guess type
  75.         $entity = current($qb->getRootEntities());
  76.         $metadata = $qb->getEntityManager()->getClassMetadata($entity);
  77.         if (\in_array($metadata->getTypeOfField($identifier), ['integer', 'bigint', 'smallint'])) {
  78.             $parameterType = Connection::PARAM_INT_ARRAY;
  80.             // Filter out non-integer values (e.g. ""). If we don't, some
  81.             // databases such as PostgreSQL fail.
  82.             $values = array_values(array_filter($values, function ($v) {
  83.                 return (string) $v === (string) (int) $v || ctype_digit($v);
  84.             }));
  85.         } elseif (\in_array($metadata->getTypeOfField($identifier), ['uuid', 'guid'])) {
  86.             $parameterType = Connection::PARAM_STR_ARRAY;
  88.             // Like above, but we just filter out empty strings.
  89.             $values = array_values(array_filter($values, function ($v) {
  90.                 return '' !== (string) $v;
  91.             }));
  92.         } else {
  93.             $parameterType = Connection::PARAM_STR_ARRAY;
  94.         }
  95.         if (!$values) {
  96.             return [];
  97.         }
  99.         return $qb->andWhere($where)
  100.                   ->getQuery()
  101.                   ->setParameter($parameter, $values, $parameterType)
  102.                   ->getResult();
  103.     }
  104. }