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

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\DBAL\Types\ConversionException;
  16. use Doctrine\DBAL\Types\Type;
  17. use Doctrine\ORM\QueryBuilder;
  18. use Symfony\Component\Form\Exception\TransformationFailedException;
  20. /**
  21.  * Loads entities using a {@link QueryBuilder} instance.
  22.  *
  23.  * @author Benjamin Eberlei <kontakt@beberlei.de>
  24.  * @author Bernhard Schussek <bschussek@gmail.com>
  25.  */
  26. class ORMQueryBuilderLoader implements EntityLoaderInterface
  27. {
  28.     /**
  29.      * Contains the query builder that builds the query for fetching the
  30.      * entities.
  31.      *
  32.      * This property should only be accessed through queryBuilder.
  33.      *
  34.      * @var QueryBuilder
  35.      */
  36.     private $queryBuilder;
  38.     public function __construct(QueryBuilder $queryBuilder)
  39.     {
  40.         $this->queryBuilder = $queryBuilder;
  41.     }
  43.     /**
  44.      * {@inheritdoc}
  45.      */
  46.     public function getEntities()
  47.     {
  48.         return $this->queryBuilder->getQuery()->execute();
  49.     }
  51.     /**
  52.      * {@inheritdoc}
  53.      */
  54.     public function getEntitiesByIds(string $identifier, array $values)
  55.     {
  56.         if (null !== $this->queryBuilder->getMaxResults() || 0 < (int) $this->queryBuilder->getFirstResult()) {
  57.             // an offset or a limit would apply on results including the where clause with submitted id values
  58.             // that could make invalid choices valid
  59.             $choices = [];
  60.             $metadata = $this->queryBuilder->getEntityManager()->getClassMetadata(current($this->queryBuilder->getRootEntities()));
  62.             foreach ($this->getEntities() as $entity) {
  63.                 if (\in_array((string) current($metadata->getIdentifierValues($entity)), $values, true)) {
  64.                     $choices[] = $entity;
  65.                 }
  66.             }
  68.             return $choices;
  69.         }
  71.         $qb = clone $this->queryBuilder;
  72.         $alias = current($qb->getRootAliases());
  73.         $parameter = 'ORMQueryBuilderLoader_getEntitiesByIds_'.$identifier;
  74.         $parameter = str_replace('.', '_', $parameter);
  75.         $where = $qb->expr()->in($alias.'.'.$identifier, ':'.$parameter);
  77.         // Guess type
  78.         $entity = current($qb->getRootEntities());
  79.         $metadata = $qb->getEntityManager()->getClassMetadata($entity);
  80.         if (\in_array($type = $metadata->getTypeOfField($identifier), ['integer', 'bigint', 'smallint'])) {
  81.             $parameterType = Connection::PARAM_INT_ARRAY;
  83.             // Filter out non-integer values (e.g. ""). If we don't, some
  84.             // databases such as PostgreSQL fail.
  85.             $values = array_values(array_filter($values, function ($v) {
  86.                 return (string) $v === (string) (int) $v || ctype_digit($v);
  87.             }));
  88.         } elseif (\in_array($type, ['ulid', 'uuid', 'guid'])) {
  89.             $parameterType = Connection::PARAM_STR_ARRAY;
  91.             // Like above, but we just filter out empty strings.
  92.             $values = array_values(array_filter($values, function ($v) {
  93.                 return '' !== (string) $v;
  94.             }));
  96.             // Convert values into right type
  97.             if (Type::hasType($type)) {
  98.                 $doctrineType = Type::getType($type);
  99.                 $platform = $qb->getEntityManager()->getConnection()->getDatabasePlatform();
  100.                 foreach ($values as &$value) {
  101.                     try {
  102.                         $value = $doctrineType->convertToDatabaseValue($value, $platform);
  103.                     } catch (ConversionException $e) {
  104.                         throw new TransformationFailedException(sprintf('Failed to transform "%s" into "%s".', $value, $type), 0, $e);
  105.                     }
  106.                 }
  107.                 unset($value);
  108.             }
  109.         } else {
  110.             $parameterType = Connection::PARAM_STR_ARRAY;
  111.         }
  112.         if (!$values) {
  113.             return [];
  114.         }
  116.         return $qb->andWhere($where)
  117.                   ->getQuery()
  118.                   ->setParameter($parameter, $values, $parameterType)
  119.                   ->getResult();
  120.     }
  121. }