vendor/sylius/resource-bundle/src/Bundle/Doctrine/ORM/ResourceRepositoryTrait.php line 35

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\EntityManagerInterface;
  17. use Doctrine\ORM\Mapping\ClassMetadata;
  18. use Doctrine\ORM\QueryBuilder;
  19. use Pagerfanta\Adapter\ArrayAdapter;
  20. use Pagerfanta\Doctrine\ORM\QueryAdapter;
  21. use Pagerfanta\Pagerfanta;
  22. use Sylius\Component\Resource\Model\ResourceInterface;
  24. /**
  25.  * @property EntityManagerInterface $_em
  26.  * @property ClassMetadata          $_class
  27.  *
  28.  * @method QueryBuilder createQueryBuilder(string $alias, string $indexBy = null)
  29.  * @method ?object      find($id, $lockMode = null, $lockVersion = null)
  30.  */
  31. trait ResourceRepositoryTrait
  32. {
  33.     public function add(ResourceInterface $resource): void
  34.     {
  35.         $this->_em->persist($resource);
  36.         $this->_em->flush();
  37.     }
  39.     public function remove(ResourceInterface $resource): void
  40.     {
  41.         if (null !== $this->find($resource->getId())) {
  42.             $this->_em->remove($resource);
  43.             $this->_em->flush();
  44.         }
  45.     }
  47.     /**
  48.      * @return iterable<int, ResourceInterface>
  49.      */
  50.     public function createPaginator(array $criteria = [], array $sorting = []): iterable
  51.     {
  52.         $queryBuilder = $this->createQueryBuilder('o');
  54.         $this->applyCriteria($queryBuilder, $criteria);
  55.         $this->applySorting($queryBuilder, $sorting);
  57.         return $this->getPaginator($queryBuilder);
  58.     }
  60.     protected function getPaginator(QueryBuilder $queryBuilder): Pagerfanta
  61.     {
  62.         if (!class_exists(QueryAdapter::class)) {
  63.             throw new \LogicException('You can not use the "paginator" if Pargefanta Doctrine ORM Adapter is not available. Try running "composer require pagerfanta/doctrine-orm-adapter".');
  64.         }
  66.         // Use output walkers option in the query adapter should be false as it affects performance greatly (see sylius/sylius#3775)
  67.         return new Pagerfanta(new QueryAdapter($queryBuilder, false, false));
  68.     }
  70.     /**
  71.      * @param array $objects
  72.      */
  73.     protected function getArrayPaginator($objects): Pagerfanta
  74.     {
  75.         return new Pagerfanta(new ArrayAdapter($objects));
  76.     }
  78.     protected function applyCriteria(QueryBuilder $queryBuilder, array $criteria = []): void
  79.     {
  80.         foreach ($criteria as $property => $value) {
  81.             if (!in_array($property, array_merge($this->_class->getAssociationNames(), $this->_class->getFieldNames()), true)) {
  82.                 continue;
  83.             }
  85.             $name = $this->getPropertyName($property);
  87.             if (null === $value) {
  88.                 $queryBuilder->andWhere($queryBuilder->expr()->isNull($name));
  89.             } elseif (is_array($value)) {
  90.                 $queryBuilder->andWhere($queryBuilder->expr()->in($name, $value));
  91.             } elseif ('' !== $value) {
  92.                 $parameter = str_replace('.', '_', $property);
  93.                 $queryBuilder
  94.                     ->andWhere($queryBuilder->expr()->eq($name, ':' . $parameter))
  95.                     ->setParameter($parameter, $value)
  96.                 ;
  97.             }
  98.         }
  99.     }
  101.     protected function applySorting(QueryBuilder $queryBuilder, array $sorting = []): void
  102.     {
  103.         foreach ($sorting as $property => $order) {
  104.             if (!in_array($property, array_merge($this->_class->getAssociationNames(), $this->_class->getFieldNames()), true)) {
  105.                 continue;
  106.             }
  108.             if (!empty($order)) {
  109.                 $queryBuilder->addOrderBy($this->getPropertyName($property), $order);
  110.             }
  111.         }
  112.     }
  114.     protected function getPropertyName(string $name): string
  115.     {
  116.         if (false === strpos($name, '.')) {
  117.             return 'o' . '.' . $name;
  118.         }
  120.         return $name;
  121.     }
  122. }