ServiceClassLocator.php

Go to the documentation of this file.

<?php
namespace Magento\Eav\Model\TypeLocator;

use Magento\Eav\Api\AttributeRepositoryInterface;
use Magento\Framework\Exception\NoSuchEntityException;
use Magento\Framework\Reflection\TypeProcessor;
use Magento\Framework\Webapi\CustomAttributeTypeLocatorInterface;
use Magento\Framework\Webapi\ServiceTypeToEntityTypeMap;

class ServiceClassLocator implements CustomAttributeTypeLocatorInterface
{
    private $serviceTypeMap;

    private $attributeRepository;

    private $complexTypeLocator;

    private $simpleTypeLocator;

    public function __construct(
        ServiceTypeToEntityTypeMap $serviceTypeMap,
        AttributeRepositoryInterface $attributeRepository,
        ComplexType $complexTypeLocator,
        SimpleType $simpleTypeLocator
    ) {
        $this->serviceTypeMap = $serviceTypeMap;
        $this->attributeRepository = $attributeRepository;
        $this->complexTypeLocator = $complexTypeLocator;
        $this->simpleTypeLocator = $simpleTypeLocator;
    }

    public function getType($attributeCode, $entityType)
    {
        $entityCode = $this->serviceTypeMap->getEntityType($entityType);
        if (!$entityCode) {
            return TypeProcessor::NORMALIZED_ANY_TYPE;
        }

        $type = $this->complexTypeLocator->getType($attributeCode, $entityCode);
        if ($type === TypeProcessor::NORMALIZED_ANY_TYPE) {
            $type = $this->simpleTypeLocator->getType($attributeCode, $entityCode);
        }

        return $type;
    }

    public function getAllServiceDataInterfaces()
    {
        return $this->complexTypeLocator->getDataTypes();
    }
}