vendor/shopware/core/Framework/DataAbstractionLayer/EntityCollection.php line 33

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Framework\DataAbstractionLayer;
  5. use Shopware\Core\Framework\Struct\Collection;
  7. /**
  8.  * @template TElement of Entity
  9.  *
  10.  * @extends Collection<TElement>
  11.  */
  12. class EntityCollection extends Collection
  13. {
  14.     public function __construct(iterable $elements = [])
  15.     {
  16.         parent::__construct([]);
  18.         foreach ($elements as $element) {
  19.             $this->validateType($element);
  21.             $this->set($element->getUniqueIdentifier(), $element);
  22.         }
  23.     }
  25.     public function fill(array $entities): void
  26.     {
  27.         array_map([$this, 'add'], $entities);
  28.     }
  30.     /**
  31.      * @param TElement $entity
  32.      */
  33.     public function add($entity): void
  34.     {
  35.         $this->set($entity->getUniqueIdentifier(), $entity);
  36.     }
  38.     /**
  39.      * @return list<string>
  40.      */
  41.     public function getIds(): array
  42.     {
  43.         return $this->fmap(static function (Entity $entity) {
  44.             return $entity->getUniqueIdentifier();
  45.         });
  46.     }
  48.     public function filterByProperty(string $property, $value)
  49.     {
  50.         return $this->filter(
  51.             static function (Entity $struct) use ($property, $value) {
  52.                 return $struct->get($property) === $value;
  53.             }
  54.         );
  55.     }
  57.     public function filterAndReduceByProperty(string $property, $value)
  58.     {
  59.         $filtered = [];
  61.         foreach ($this->getIterator() as $key => $struct) {
  62.             if ($struct->get($property) !== $value) {
  63.                 continue;
  64.             }
  65.             $filtered[] = $struct;
  66.             $this->remove($key);
  67.         }
  69.         return $this->createNew($filtered);
  70.     }
  72.     /**
  73.      * @param EntityCollection<TElement> $collection
  74.      */
  75.     public function merge(self $collection): void
  76.     {
  77.         /** @var TElement $entity */
  78.         foreach ($collection as $entity) {
  79.             if ($this->has($entity->getUniqueIdentifier())) {
  80.                 continue;
  81.             }
  82.             $this->add($entity);
  83.         }
  84.     }
  86.     /**
  87.      * @param TElement $entity
  88.      */
  89.     public function insert(int $position, Entity $entity): void
  90.     {
  91.         $items = array_values($this->elements);
  93.         $this->elements = [];
  94.         foreach ($items as $index => $item) {
  95.             if ($index === $position) {
  96.                 $this->add($entity);
  97.             }
  98.             $this->add($item);
  99.         }
  100.     }
  102.     public function getList(array $ids)
  103.     {
  104.         return $this->createNew(array_intersect_key($this->elements, array_flip($ids)));
  105.     }
  107.     public function sortByIdArray(array $ids): void
  108.     {
  109.         $sorted = [];
  111.         foreach ($ids as $id) {
  112.             if (\is_array($id)) {
  113.                 $id = implode('-', array_unique($id));
  114.             }
  116.             if (\array_key_exists($id, $this->elements)) {
  117.                 $sorted[$id] = $this->elements[$id];
  118.             }
  119.         }
  120.         $this->elements = $sorted;
  121.     }
  123.     protected function getExpectedClass(): string
  124.     {
  125.         return Entity::class;
  126.     }
  127. }