AttributeProvider.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\Model\EntitySnapshot;

use Magento\Framework\EntityManager\MetadataPool;
use Magento\Framework\ObjectManagerInterface as ObjectManager;

class AttributeProvider implements AttributeProviderInterface
{
    protected $providers;

    protected $metadataPool;

    protected $objectManager;

    protected $registry;

    public function __construct(
        MetadataPool $metadataPool,
        ObjectManager $objectManager,
        $providers = []
    ) {
        $this->metadataPool = $metadataPool;
        $this->objectManager = $objectManager;
        $this->providers = $providers;
    }

    public function getAttributes($entityType)
    {
        if (!isset($this->registry[$entityType])) {
            $metadata = $this->metadataPool->getMetadata($entityType);
            $this->registry[$entityType] = $metadata->getEntityConnection()->describeTable($metadata->getEntityTable());
            if ($metadata->getLinkField() != $metadata->getIdentifierField()) {
                unset($this->registry[$entityType][$metadata->getLinkField()]);
            }
            $providers = [];
            if (isset($this->providers[$entityType])) {
                $providers = $this->providers[$entityType];
            } elseif (isset($this->providers['default'])) {
                $providers = $this->providers['default'];
            }
            foreach ($providers as $providerClass) {
                $provider = $this->objectManager->get($providerClass);
                $this->registry[$entityType] = array_merge(
                    $this->registry[$entityType],
                    $provider->getAttributes($entityType)
                );
            }
        }
        return $this->registry[$entityType];
    }
}