vendor/sylius/resource-bundle/src/Bundle/Doctrine/ORM/EntityRepository.php line 27

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of the Sylius package.
  5.  *
  6.  * (c) Paweł Jędrzejewski
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. declare(strict_types=1);
  14. namespace Sylius\Bundle\ResourceBundle\Doctrine\ORM;
  16. use Doctrine\ORM\EntityRepository as BaseEntityRepository;
  17. use Doctrine\ORM\QueryBuilder;
  18. use Pagerfanta\Adapter\ArrayAdapter;
  19. use Pagerfanta\Doctrine\ORM\QueryAdapter;
  20. use Pagerfanta\Pagerfanta;
  21. use Sylius\Component\Resource\Model\ResourceInterface;
  22. use Sylius\Component\Resource\Repository\RepositoryInterface;
  24. /** @psalm-suppress DeprecatedInterface */
  25. class EntityRepository extends BaseEntityRepository implements RepositoryInterface
  26. {
  27.     public function add(ResourceInterface $resource): void
  28.     {
  29.         $this->_em->persist($resource);
  30.         $this->_em->flush();
  31.     }
  33.     public function remove(ResourceInterface $resource): void
  34.     {
  35.         if (null !== $this->find($resource->getId())) {
  36.             $this->_em->remove($resource);
  37.             $this->_em->flush();
  38.         }
  39.     }
  41.     public function createPaginator(array $criteria = [], array $sorting = []): iterable
  42.     {
  43.         $queryBuilder = $this->createQueryBuilder('o');
  45.         $this->applyCriteria($queryBuilder, $criteria);
  46.         $this->applySorting($queryBuilder, $sorting);
  48.         return $this->getPaginator($queryBuilder);
  49.     }
  51.     protected function getPaginator(QueryBuilder $queryBuilder): Pagerfanta
  52.     {
  53.         // Use output walkers option in the query adapter should be false as it affects performance greatly (see sylius/sylius#3775)
  54.         return new Pagerfanta(new QueryAdapter($queryBuilder, false, false));
  55.     }
  57.     /**
  58.      * @param array $objects
  59.      */
  60.     protected function getArrayPaginator($objects): Pagerfanta
  61.     {
  62.         return new Pagerfanta(new ArrayAdapter($objects));
  63.     }
  65.     protected function applyCriteria(QueryBuilder $queryBuilder, array $criteria = []): void
  66.     {
  67.         foreach ($criteria as $property => $value) {
  68.             if (!in_array($property, array_merge($this->_class->getAssociationNames(), $this->_class->getFieldNames()), true)) {
  69.                 continue;
  70.             }
  72.             $name = $this->getPropertyName($property);
  74.             if (null === $value) {
  75.                 $queryBuilder->andWhere($queryBuilder->expr()->isNull($name));
  76.             } elseif (is_array($value)) {
  77.                 $queryBuilder->andWhere($queryBuilder->expr()->in($name, $value));
  78.             } elseif ('' !== $value) {
  79.                 $parameter = str_replace('.', '_', $property);
  80.                 $queryBuilder
  81.                     ->andWhere($queryBuilder->expr()->eq($name, ':' . $parameter))
  82.                     ->setParameter($parameter, $value)
  83.                 ;
  84.             }
  85.         }
  86.     }
  88.     protected function applySorting(QueryBuilder $queryBuilder, array $sorting = []): void
  89.     {
  90.         foreach ($sorting as $property => $order) {
  91.             if (!in_array($property, array_merge($this->_class->getAssociationNames(), $this->_class->getFieldNames()), true)) {
  92.                 continue;
  93.             }
  95.             if (!empty($order)) {
  96.                 $queryBuilder->addOrderBy($this->getPropertyName($property), $order);
  97.             }
  98.         }
  99.     }
  101.     protected function getPropertyName(string $name): string
  102.     {
  103.         if (false === strpos($name, '.')) {
  104.             return 'o' . '.' . $name;
  105.         }
  107.         return $name;
  108.     }
  109. }