IndexIterator.php

Go to the documentation of this file.

<?php
namespace Magento\CatalogSearch\Model\Indexer\Fulltext\Action;

class IndexIterator implements \Iterator
{
    private $dataProvider;

    private $storeId;

    private $staticFields;

    private $productIds;

    private $dynamicFields;

    private $visibility;

    private $allowedVisibility;

    private $status;

    private $statusIds;

    private $lastProductId = 0;

    private $products = [];

    private $current = null;

    private $isValid = true;

    private $key = null;

    private $productAttributes = [];

    private $productRelations = [];

    public function __construct(
        DataProvider $dataProvider,
        $storeId,
        array $staticFields,
        $productIds,
        array $dynamicFields,
        \Magento\Eav\Model\Entity\Attribute $visibility,
        array $allowedVisibility,
        \Magento\Eav\Model\Entity\Attribute $status,
        array $statusIds
    ) {
        $this->dataProvider = $dataProvider;
        $this->storeId = $storeId;
        $this->staticFields = $staticFields;
        $this->productIds = $productIds;
        $this->dynamicFields = $dynamicFields;
        $this->visibility = $visibility;
        $this->allowedVisibility = $allowedVisibility;
        $this->status = $status;
        $this->statusIds = $statusIds;
    }

    public function current()
    {
        return $this->current;
    }

    public function next()
    {
        \next($this->products);
        if (\key($this->products) === null) {
            // check if storage has more items to process
            $this->products = $this->dataProvider->getSearchableProducts(
                $this->storeId,
                $this->staticFields,
                $this->productIds,
                $this->lastProductId
            );

            if (!count($this->products)) {
                $this->isValid = false;
                return;
            }

            $productAttributes = [];
            $this->productRelations = [];
            foreach ($this->products as $productData) {
                $this->lastProductId = $productData['entity_id'];
                $productAttributes[$productData['entity_id']] = $productData['entity_id'];
                $productChildren = $this->dataProvider->getProductChildIds(
                    $productData['entity_id'],
                    $productData['type_id']
                );
                $this->productRelations[$productData['entity_id']] = $productChildren;
                if ($productChildren) {
                    foreach ($productChildren as $productChildId) {
                        $productAttributes[$productChildId] = $productChildId;
                    }
                }
            }
            \reset($this->products);

            $this->productAttributes = $this->dataProvider->getProductAttributes(
                $this->storeId,
                $productAttributes,
                $this->dynamicFields
            );
        }

        $productData = \current($this->products);

        if (!isset($this->productAttributes[$productData['entity_id']])) {
            $this->next();
            return;
        }

        $productAttr = $this->productAttributes[$productData['entity_id']];
        if (!isset($productAttr[$this->visibility->getId()])
            || !in_array($productAttr[$this->visibility->getId()], $this->allowedVisibility)
        ) {
            $this->next();
            return;
        }
        if (!isset($productAttr[$this->status->getId()])
            || !in_array($productAttr[$this->status->getId()], $this->statusIds)
        ) {
            $this->next();
            return;
        }

        $productIndex = [$productData['entity_id'] => $productAttr];

        $hasChildren = false;
        $productChildren = $this->productRelations[$productData['entity_id']];
        if ($productChildren) {
            foreach ($productChildren as $productChildId) {
                if (isset($this->productAttributes[$productChildId])) {
                    $productChildAttr = $this->productAttributes[$productChildId];
                    if (!isset($productChildAttr[$this->status->getId()])
                        || !in_array($productChildAttr[$this->status->getId()], $this->statusIds)
                    ) {
                        continue;
                    }

                    $hasChildren = true;
                    $productIndex[$productChildId] = $productChildAttr;
                }
            }
        }
        if ($productChildren !== null && !$hasChildren) {
            $this->next();
            return;
        }

        $index = $this->dataProvider->prepareProductIndex($productIndex, $productData, $this->storeId);

        $this->current = $index;
        $this->key = $productData['entity_id'];
    }

    public function key()
    {
        return $this->key;
    }

    public function valid()
    {
        return $this->isValid;
    }

    public function rewind()
    {
        $this->lastProductId = 0;
        $this->key = null;
        $this->current = null;
        unset($this->products);
        $this->products = [];
        $this->next();
    }
}