DataMapperResolver.php

Go to the documentation of this file.

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

use Magento\Framework\Exception\NoSuchEntityException;
use Magento\Framework\Exception\ConfigurationMismatchException;
use Magento\Elasticsearch\Model\Adapter\BatchDataMapperInterface;
use Magento\Elasticsearch\Model\Config;

class DataMapperResolver implements BatchDataMapperInterface
{
    private $dataMapperEntity;

    private $dataMapperFactory;

    public function __construct(DataMapperFactory $dataMapperFactory)
    {
        $this->dataMapperFactory = $dataMapperFactory;
    }

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

    private function getDataMapper($entityType)
    {
        if (!isset($this->dataMapperEntity[$entityType])) {
            $this->dataMapperEntity[$entityType] = $this->dataMapperFactory->create($entityType);
        }

        return $this->dataMapperEntity[$entityType];
    }
}