ModeSwitcher.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\Catalog\Model\Indexer\Product\Price;

use Magento\Framework\Search\Request\Dimension;
use Magento\Store\Model\Indexer\WebsiteDimensionProvider;
use Magento\Customer\Model\Indexer\CustomerGroupDimensionProvider;
use Magento\Indexer\Model\DimensionModes;
use Magento\Indexer\Model\DimensionMode;

class ModeSwitcher implements \Magento\Indexer\Model\ModeSwitcherInterface
{
    private $tableMaintainer;

    private $dimensionCollectionFactory;

    private $dimensionsArray;

    private $dimensionModeConfiguration;

    private $modeSwitcherConfiguration;

    public function __construct(
        TableMaintainer $tableMaintainer,
        DimensionCollectionFactory $dimensionCollectionFactory,
        DimensionModeConfiguration $dimensionModeConfiguration,
        ModeSwitcherConfiguration $modeSwitcherConfiguration
    ) {
        $this->tableMaintainer = $tableMaintainer;
        $this->dimensionCollectionFactory = $dimensionCollectionFactory;
        $this->dimensionModeConfiguration = $dimensionModeConfiguration;
        $this->modeSwitcherConfiguration = $modeSwitcherConfiguration;
    }

    public function getDimensionModes(): DimensionModes
    {
        $dimensionsList = [];
        foreach ($this->dimensionModeConfiguration->getDimensionModes() as $dimension => $modes) {
            $dimensionsList[] = new DimensionMode($dimension, $modes);
        }

        return new DimensionModes($dimensionsList);
    }

    public function switchMode(string $currentMode, string $previousMode)
    {
        //Create new tables and move data
        $this->createTables($currentMode);
        $this->moveData($currentMode, $previousMode);

        //Change config options
        $this->modeSwitcherConfiguration->saveMode($currentMode);

        //Delete old tables
        $this->dropTables($previousMode);
    }

    public function createTables(string $currentMode)
    {
        foreach ($this->getDimensionsArray($currentMode) as $dimensions) {
            if (!empty($dimensions)) {
                $this->tableMaintainer->createTablesForDimensions($dimensions);
            }
        }
    }

    public function moveData(string $currentMode, string $previousMode)
    {
        $dimensionsArrayForCurrentMode = $this->getDimensionsArray($currentMode);
        $dimensionsArrayForPreviousMode = $this->getDimensionsArray($previousMode);

        foreach ($dimensionsArrayForCurrentMode as $dimensionsForCurrentMode) {
            $newTable = $this->tableMaintainer->getMainTable($dimensionsForCurrentMode);
            if (empty($dimensionsForCurrentMode)) {
                // new mode is 'none'
                foreach ($dimensionsArrayForPreviousMode as $dimensionsForPreviousMode) {
                    $oldTable = $this->tableMaintainer->getMainTable($dimensionsForPreviousMode);
                    $this->insertFromOldTablesToNew($newTable, $oldTable);
                }
            } else {
                // new mode is not 'none'
                foreach ($dimensionsArrayForPreviousMode as $dimensionsForPreviousMode) {
                    $oldTable = $this->tableMaintainer->getMainTable($dimensionsForPreviousMode);
                    $this->insertFromOldTablesToNew($newTable, $oldTable, $dimensionsForCurrentMode);
                }
            }
        }
    }

    public function dropTables(string $previousMode)
    {
        foreach ($this->getDimensionsArray($previousMode) as $dimensions) {
            if (empty($dimensions)) {
                $this->tableMaintainer->truncateTablesForDimensions($dimensions);
            } else {
                $this->tableMaintainer->dropTablesForDimensions($dimensions);
            }
        }
    }

    private function getDimensionsArray(string $mode): \Magento\Framework\Indexer\MultiDimensionProvider
    {
        if (isset($this->dimensionsArray[$mode])) {
            return $this->dimensionsArray[$mode];
        }

        $this->dimensionsArray[$mode] = $this->dimensionCollectionFactory->create($mode);

        return $this->dimensionsArray[$mode];
    }

    private function insertFromOldTablesToNew(string $newTable, string $oldTable, array $dimensions = [])
    {
        $select = $this->tableMaintainer->getConnection()->select()->from($oldTable);

        foreach ($dimensions as $dimension) {
            if ($dimension->getName() === WebsiteDimensionProvider::DIMENSION_NAME) {
                $select->where('website_id = ?', $dimension->getValue());
            }
            if ($dimension->getName() === CustomerGroupDimensionProvider::DIMENSION_NAME) {
                $select->where('customer_group_id = ?', $dimension->getValue());
            }
        }
        $this->tableMaintainer->getConnection()->query(
            $this->tableMaintainer->getConnection()->insertFromSelect(
                $select,
                $newTable,
                [],
                \Magento\Framework\DB\Adapter\AdapterInterface::INSERT_ON_DUPLICATE
            )
        );
    }
}