DataMapperResolver.php

Go to the documentation of this file.

<?php
namespace Magento\Elasticsearch\Model\Adapter\DataMapper;

use Magento\Framework\ObjectManagerInterface;
use Magento\Elasticsearch\Model\Adapter\DataMapperInterface;
use Magento\Elasticsearch\Model\Config;

class DataMapperResolver implements DataMapperInterface
{
    private $objectManager;

    private $dataMappers;

    private $dataMapperEntity;

    public function __construct(
        ObjectManagerInterface $objectManager,
        array $dataMappers = []
    ) {
        $this->objectManager = $objectManager;
        $this->dataMappers = $dataMappers;
    }

    public function map(
        $entityId,
        array $entityIndexData,
        $storeId,
        $context = []
    ) {
        $entityType = isset($context['entityType']) ? $context['entityType'] : Config::ELASTICSEARCH_TYPE_DEFAULT;
        return $this->getEntity($entityType)->map($entityId, $entityIndexData, $storeId, $context);
    }

    private function getEntity($entityType = '')
    {
        if (empty($this->dataMapperEntity)) {
            if (empty($entityType)) {
                throw new \Exception(
                    'No entity type given'
                );
            }
            if (!isset($this->dataMappers[$entityType])) {
                throw new \LogicException(
                    'There is no such data mapper: ' . $entityType
                );
            }
            $dataMapperClass = $this->dataMappers[$entityType];
            $this->dataMapperEntity = $this->objectManager->create($dataMapperClass);
            if (!($this->dataMapperEntity instanceof DataMapperInterface)) {
                throw new \InvalidArgumentException(
                    'Data mapper must implement \Magento\Elasticsearch\Model\Adapter\DataMapperInterface'
                );
            }
        }
        return $this->dataMapperEntity;
    }
}