src/Core/Framework/DataAbstractionLayer/Dbal/Common/RepositoryIterator.php line 91

  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(EntityRepository $repository, Context $context, ?Criteria $criteria = null)
  25.     {
  26.         if ($criteria === null) {
  27.             $criteria = new Criteria();
  28.             $criteria->setOffset(0);
  29.         }
  31.         if ($criteria->getLimit() === null || $criteria->getLimit() < 1) {
  32.             $criteria->setLimit(50);
  33.         }
  35.         if ($repository->getDefinition()->hasAutoIncrement()) {
  36.             $criteria->addSorting(new FieldSorting('autoIncrement', FieldSorting::ASCENDING));
  37.             $criteria->setFilter('increment', new RangeFilter('autoIncrement', [RangeFilter::GTE => 0]));
  38.             $this->autoIncrement = true;
  39.         }
  41.         $this->criteria = $criteria;
  42.         $this->repository = $repository;
  43.         $this->context = clone $context;
  44.     }
  46.     public function getTotal(): int
  47.     {
  48.         $criteria = clone $this->criteria;
  49.         $criteria->setOffset(0);
  50.         $criteria->setLimit(1);
  51.         $criteria->setTotalCountMode(Criteria::TOTAL_COUNT_MODE_EXACT);
  53.         $result = $this->repository->searchIds($criteria, $this->context);
  55.         return $result->getTotal();
  56.     }
  58.     public function fetchIds(): ?array
  59.     {
  60.         $this->criteria->setTotalCountMode(Criteria::TOTAL_COUNT_MODE_NONE);
  62.         $ids = $this->repository->searchIds($this->criteria, $this->context);
  64.         $values = $ids->getIds();
  66.         if (empty($values)) {
  67.             return null;
  68.         }
  70.         if (!$this->autoIncrement) {
  71.             $this->criteria->setOffset($this->criteria->getOffset() + $this->criteria->getLimit());
  73.             return $values;
  74.         }
  76.         $last = end($values);
  77.         if (!\is_string($last)) {
  78.             throw new \RuntimeException('Expected string as last element of ids array');
  79.         }
  81.         $increment = $ids->getDataFieldOfId($last, 'autoIncrement');
  82.         $this->criteria->setFilter('increment', new RangeFilter('autoIncrement', [RangeFilter::GT => $increment]));
  84.         return $values;
  85.     }
  87.     public function fetch(): ?EntitySearchResult
  88.     {
  89.         $this->criteria->setTotalCountMode(Criteria::TOTAL_COUNT_MODE_NONE);
  91.         $result = $this->repository->search(clone $this->criteria, $this->context);
  93.         // increase offset for next iteration
  94.         $this->criteria->setOffset($this->criteria->getOffset() + $this->criteria->getLimit());
  96.         if (empty($result->getIds())) {
  97.             return null;
  98.         }
  100.         return $result;
  101.     }
  102. }