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

Open in your IDE?
  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\EntityRepositoryInterface;
  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\Feature;
  12. use Shopware\Core\Framework\Log\Package;
  14. #[Package('core')]
  15. class RepositoryIterator
  16. {
  17.     /**
  18.      * @var Criteria
  19.      */
  20.     private $criteria;
  22.     /**
  23.      * @var EntityRepositoryInterface
  24.      */
  25.     private $repository;
  27.     /**
  28.      * @var Context
  29.      */
  30.     private $context;
  32.     private bool $autoIncrement = false;
  34.     public function __construct(EntityRepositoryInterface $repository, Context $context, ?Criteria $criteria = null)
  35.     {
  36.         if ($criteria === null) {
  37.             $criteria = new Criteria();
  38.             $criteria->setOffset(0);
  39.         }
  41.         if ($criteria->getLimit() === null || $criteria->getLimit() < 1) {
  42.             $criteria->setLimit(50);
  43.         }
  45.         if ($repository->getDefinition()->hasAutoIncrement()) {
  46.             $criteria->addSorting(new FieldSorting('autoIncrement', FieldSorting::ASCENDING));
  47.             $criteria->setFilter('increment', new RangeFilter('autoIncrement', [RangeFilter::GTE => 0]));
  48.             $this->autoIncrement = true;
  49.         }
  51.         $this->criteria = $criteria;
  52.         $this->repository = $repository;
  53.         $this->context = clone $context;
  54.     }
  56.     public function getTotal(): int
  57.     {
  58.         $criteria = clone $this->criteria;
  59.         $criteria->setOffset(0);
  60.         $criteria->setLimit(1);
  61.         $criteria->setTotalCountMode(Criteria::TOTAL_COUNT_MODE_EXACT);
  63.         $result = $this->repository->searchIds($criteria, $this->context);
  65.         return $result->getTotal();
  66.     }
  68.     public function fetchIds(): ?array
  69.     {
  70.         $this->criteria->setTotalCountMode(Criteria::TOTAL_COUNT_MODE_NONE);
  72.         $ids = $this->repository->searchIds($this->criteria, $this->context);
  74.         $values = $ids->getIds();
  76.         if (empty($values)) {
  77.             return null;
  78.         }
  80.         if (!$this->autoIncrement) {
  81.             $this->criteria->setOffset($this->criteria->getOffset() + $this->criteria->getLimit());
  83.             return $values;
  84.         }
  86.         $last = end($values);
  87.         if (!\is_string($last)) {
  88.             throw new \RuntimeException('Expected string as last element of ids array');
  89.         }
  91.         $increment = $ids->getDataFieldOfId($last, Feature::isActive('v6.5.0.0') ? 'autoIncrement' : 'auto_increment');
  92.         $this->criteria->setFilter('increment', new RangeFilter('autoIncrement', [RangeFilter::GT => $increment]));
  94.         return $values;
  95.     }
  97.     public function fetch(): ?EntitySearchResult
  98.     {
  99.         $this->criteria->setTotalCountMode(Criteria::TOTAL_COUNT_MODE_NONE);
  101.         $result = $this->repository->search(clone $this->criteria, $this->context);
  103.         // increase offset for next iteration
  104.         $this->criteria->setOffset($this->criteria->getOffset() + $this->criteria->getLimit());
  106.         if (empty($result->getIds())) {
  107.             return null;
  108.         }
  110.         return $result;
  111.     }
  112. }