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

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.  * @package core
  9.  * @template TElement of Entity
  10.  *
  11.  * @extends Collection<TElement>
  12.  */
  13. class EntityCollection extends Collection
  14. {
  15.     public function __construct(iterable $elements = [])
  16.     {
  17.         parent::__construct([]);
  19.         foreach ($elements as $element) {
  20.             $this->validateType($element);
  22.             $this->set($element->getUniqueIdentifier(), $element);
  23.         }
  24.     }
  26.     public function fill(array $entities): void
  27.     {
  28.         array_map([$this, 'add'], $entities);
  29.     }
  31.     /**
  32.      * @param TElement $entity
  33.      */
  34.     public function add($entity): void
  35.     {
  36.         $this->set($entity->getUniqueIdentifier(), $entity);
  37.     }
  39.     /**
  40.      * @return list<string>
  41.      */
  42.     public function getIds(): array
  43.     {
  44.         return $this->fmap(static function (Entity $entity) {
  45.             return $entity->getUniqueIdentifier();
  46.         });
  47.     }
  49.     public function filterByProperty(string $property, $value)
  50.     {
  51.         return $this->filter(
  52.             static function (Entity $struct) use ($property, $value) {
  53.                 return $struct->get($property) === $value;
  54.             }
  55.         );
  56.     }
  58.     public function filterAndReduceByProperty(string $property, $value)
  59.     {
  60.         $filtered = [];
  62.         foreach ($this->getIterator() as $key => $struct) {
  63.             if ($struct->get($property) !== $value) {
  64.                 continue;
  65.             }
  66.             $filtered[] = $struct;
  67.             $this->remove($key);
  68.         }
  70.         return $this->createNew($filtered);
  71.     }
  73.     /**
  74.      * @param EntityCollection<TElement> $collection
  75.      */
  76.     public function merge(self $collection): void
  77.     {
  78.         /** @var TElement $entity */
  79.         foreach ($collection as $entity) {
  80.             if ($this->has($entity->getUniqueIdentifier())) {
  81.                 continue;
  82.             }
  83.             $this->add($entity);
  84.         }
  85.     }
  87.     /**
  88.      * @param TElement $entity
  89.      */
  90.     public function insert(int $position, Entity $entity): void
  91.     {
  92.         $items = array_values($this->elements);
  94.         $this->elements = [];
  95.         foreach ($items as $index => $item) {
  96.             if ($index === $position) {
  97.                 $this->add($entity);
  98.             }
  99.             $this->add($item);
  100.         }
  101.     }
  103.     public function getList(array $ids)
  104.     {
  105.         return $this->createNew(array_intersect_key($this->elements, array_flip($ids)));
  106.     }
  108.     public function sortByIdArray(array $ids): void
  109.     {
  110.         $sorted = [];
  112.         foreach ($ids as $id) {
  113.             if (\is_array($id)) {
  114.                 $id = implode('-', array_unique($id));
  115.             }
  117.             if (\array_key_exists($id, $this->elements)) {
  118.                 $sorted[$id] = $this->elements[$id];
  119.             }
  120.         }
  121.         $this->elements = $sorted;
  122.     }
  124.     protected function getExpectedClass(): string
  125.     {
  126.         return Entity::class;
  127.     }
  128. }