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

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