ServiceSource.php

Go to the documentation of this file.

<?php
namespace Magento\Indexer\Model\Source;

use Magento\Framework\Api\SearchCriteriaBuilder;
use Magento\Framework\Api\SearchResults;
use Magento\Framework\Exception\NotFoundException;
use Magento\Framework\ObjectManagerInterface;

class ServiceSource implements DataInterface
{
    private $objectManager;

    private $service;

    private $searchCriteriaBuilder;

    public function __construct(
        ObjectManagerInterface $objectManager,
        SearchCriteriaBuilder $searchCriteriaBuilder,
        $service
    ) {
        $this->objectManager = $objectManager;
        $this->searchCriteriaBuilder = $searchCriteriaBuilder;
        $this->service = $service;
    }

    public function getData(array $fieldsData)
    {
        $service = $this->getService();

        $searchCriteria = $this->searchCriteriaBuilder->create();

        $list = $service->getList($searchCriteria);

        return $this->getRequestedFields($list, $fieldsData);
    }

    private function getRequestedFields(SearchResults $list, array $fields)
    {
        $requestedData = [];
        foreach ($list->getItems() as $key => $item) {
            foreach (array_keys($fields) as $fieldName) {
                if (!isset($item[$fieldName])) {
                    throw new NotFoundException(__("Field '%1' not found", $fieldName));
                }

                $requestedData[$key][$fieldName] = $item[$fieldName];
            }
        }
        return $requestedData;
    }

    private function getService()
    {
        return $this->objectManager->get($this->service);
    }
}