vendor/shopware/core/Framework/DataAbstractionLayer/Dbal/Common/RepositoryIterator.php line 94

  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Framework\DataAbstractionLayer\Dbal\Common;
  5. use Shopware\Core\Framework\Context;
  6. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepository;
  7. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  8. use Shopware\Core\Framework\DataAbstractionLayer\Search\EntitySearchResult;
  9. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\RangeFilter;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Search\Sorting\FieldSorting;
  11. use Shopware\Core\Framework\Log\Package;
  13. #[Package('core')]
  14. class RepositoryIterator
  15. {
  16.     private readonly Criteria $criteria;
  18.     private readonly EntityRepository $repository;
  20.     private readonly Context $context;
  22.     private bool $autoIncrement = false;
  24.     public function __construct(
  25.         EntityRepository $repository,
  26.         Context $context,
  27.         ?Criteria $criteria = null
  28.     ) {
  29.         if ($criteria === null) {
  30.             $criteria = new Criteria();
  31.             $criteria->setOffset(0);
  32.         }
  34.         if ($criteria->getLimit() === null || $criteria->getLimit() < 1) {
  35.             $criteria->setLimit(50);
  36.         }
  38.         if ($repository->getDefinition()->hasAutoIncrement()) {
  39.             $criteria->addSorting(new FieldSorting('autoIncrement', FieldSorting::ASCENDING));
  40.             $criteria->setFilter('increment', new RangeFilter('autoIncrement', [RangeFilter::GTE => 0]));
  41.             $this->autoIncrement = true;
  42.         }
  44.         $this->criteria = $criteria;
  45.         $this->repository = $repository;
  46.         $this->context = clone $context;
  47.     }
  49.     public function getTotal(): int
  50.     {
  51.         $criteria = clone $this->criteria;
  52.         $criteria->setOffset(0);
  53.         $criteria->setLimit(1);
  54.         $criteria->setTotalCountMode(Criteria::TOTAL_COUNT_MODE_EXACT);
  56.         $result = $this->repository->searchIds($criteria, $this->context);
  58.         return $result->getTotal();
  59.     }
  61.     public function fetchIds(): ?array
  62.     {
  63.         $this->criteria->setTotalCountMode(Criteria::TOTAL_COUNT_MODE_NONE);
  65.         $ids = $this->repository->searchIds($this->criteria, $this->context);
  67.         $values = $ids->getIds();
  69.         if (empty($values)) {
  70.             return null;
  71.         }
  73.         if (!$this->autoIncrement) {
  74.             $this->criteria->setOffset($this->criteria->getOffset() + $this->criteria->getLimit());
  76.             return $values;
  77.         }
  79.         $last = end($values);
  80.         if (!\is_string($last)) {
  81.             throw new \RuntimeException('Expected string as last element of ids array');
  82.         }
  84.         $increment = $ids->getDataFieldOfId($last, 'autoIncrement');
  85.         $this->criteria->setFilter('increment', new RangeFilter('autoIncrement', [RangeFilter::GT => $increment]));
  87.         return $values;
  88.     }
  90.     public function fetch(): ?EntitySearchResult
  91.     {
  92.         $this->criteria->setTotalCountMode(Criteria::TOTAL_COUNT_MODE_NONE);
  94.         $result = $this->repository->search(clone $this->criteria, $this->context);
  96.         // increase offset for next iteration
  97.         $this->criteria->setOffset($this->criteria->getOffset() + $this->criteria->getLimit());
  99.         if (empty($result->getIds())) {
  100.             return null;
  101.         }
  103.         return $result;
  104.     }
  105. }