src/Repository/SpecificationTrait.php line 67

Open in your IDE?
  1. <?php
  2. /**
  3.  * Created by simpson <simpsonwork@gmail.com>
  4.  * Date: 2019-03-20
  5.  * Time: 13:11
  6.  */
  8. namespace App\Repository;
  10. use Doctrine\ORM\AbstractQuery;
  11. use Doctrine\ORM\Query;
  12. use Doctrine\ORM\QueryBuilder;
  13. use Gedmo\Translatable\Query\TreeWalker\TranslationWalker;
  14. use Happyr\DoctrineSpecification\Filter\Filter;
  15. use Happyr\DoctrineSpecification\Query\QueryModifier;
  16. use Happyr\DoctrineSpecification\Repository\EntitySpecificationRepositoryTrait;
  17. use Happyr\DoctrineSpecification\Result\ResultModifier;
  18. use Porpaginas\Doctrine\ORM\ORMQueryResult;
  20. trait SpecificationTrait
  21. {
  22.     use EntitySpecificationRepositoryTrait;
  24.     public function matchingSpec(Filter|QueryModifier $specification, ?ResultModifier $modifier = null, bool $fetchCollection = true): ORMQueryResult
  25.     {
  26.         $qb = $this->getQueryBuilder($specification);
  27.         if (method_exists($this,'modifyListingQueryBuilder')) {
  28.             $this->modifyListingQueryBuilder($qb, $this->getAlias());
  29.         }
  31.         $query = $qb->getQuery();
  32.         $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, TranslationWalker::class);
  34.         if (null !== $modifier) {
  35.             $modifier->modify($query);
  36.         }
  38.         return new ORMQueryResult($query, $fetchCollection);
  39.     }
  41.     public function listIdMatchingSpec(Filter|QueryModifier $specification, int $offset, ?int $limit): array
  42.     {
  43.         /** @var QueryBuilder $qb */
  44.         $qb = $this->createQueryBuilder($dqlAlias = 'p');
  45.         $qb->select('p.id');
  47.         $specification->modify($qb, $dqlAlias);
  48.         $qb->andWhere($specification->getFilter($qb, $dqlAlias));
  50.         if($offset !== -1) {
  51.             $qb->setFirstResult($offset);
  52.         }
  54.         if(null !== $limit) {
  55.             $qb->setMaxResults($limit);
  56.         }
  58.         $result = $qb->getQuery()->getResult();
  59.         return array_map(function(array $row) {
  60.             return $row['id'];
  61.         }, $result);
  62.     }
  64.     public function countMatchingSpec(Filter|QueryModifier $specification): int
  65.     {
  66.         /** @var QueryBuilder $qb */
  67.         $qb = $this->createQueryBuilder($dqlAlias = 'p');
  68.         $qb->select('count(p.id)');
  70.         $qb->andWhere($specification->getFilter($qb, $dqlAlias));
  71.         $specification->modify($qb, $dqlAlias);
  73.         return (int)$qb->getQuery()->getSingleScalarResult();
  74.     }
  76.     public function matchingSpecRaw(Filter|QueryModifier $specification, ?ResultModifier $modifier = null, bool $useListingModifier = true, callable $hydrator = null): array
  77.     {
  78.         $qb = $this->getQueryBuilder($specification);
  79.         if (true == $useListingModifier && method_exists($this,'modifyListingQueryBuilder')) {
  80.             $this->modifyListingQueryBuilder($qb, $this->getAlias());
  81.         }
  83.         $query = $qb->getQuery();
  84.         $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, TranslationWalker::class);
  86.         if (null !== $modifier) {
  87.             $modifier->modify($query);
  88.         }
  90.         if(null != $hydrator) {
  91.             $rows = $query->getArrayResult();
  92.             $result = array_map(function($row) use ($hydrator) {
  93.                 return $hydrator($row);
  94.             }, $rows);
  95.         } else {
  96.             $result = $query->getResult();
  97.         }
  99.         return $result;
  100.     }
  101. }