DataSourcePool.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\View;

use Magento\Framework\View\Element\BlockFactory;

class DataSourcePool
{
    protected $blockFactory;

    protected $dataSources = [];

    protected $assignments = [];

    public function __construct(BlockFactory $blockFactory)
    {
        $this->blockFactory = $blockFactory;
    }

    public function add($name, $class)
    {
        if (!isset($this->dataSources[$name])) {
            if (!class_exists($class)) {
                throw new \InvalidArgumentException(
                    (string)new \Magento\Framework\Phrase('Invalid Data Source class name: %1', [$class])
                );
            }

            $data = $this->blockFactory->createBlock($class);

            $this->dataSources[$name] = $data;
        }

        return $this->dataSources[$name];
    }

    public function get($name = null)
    {
        if (!isset($name)) {
            return $this->dataSources;
        }

        return $this->dataSources[$name] ?? null;
    }

    public function assign($dataName, $namespace, $alias)
    {
        $alias = $alias ?: $dataName;
        $data = $this->get($dataName);

        $this->assignments[$namespace][$alias] = $data;
    }

    public function getNamespaceData($namespace)
    {
        return $this->assignments[$namespace] ?? [];
    }
}