entitiesMap = []; foreach ($entities as $entity) { $this->entitiesMap[$entity->getId()] = $entity; } } /** * Get Entity * * @param int $entityId * @return \Magento\Framework\DataObject|null */ public function get($entityId) { return $this->entitiesMap[$entityId] ?? null; } /** * List Entities * * @return \Magento\Framework\DataObject[] */ public function getList() { return $this->entitiesMap; } }