ReCreateTable.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\Setup\Declaration\Schema\Operations;

use Magento\Framework\Setup\Declaration\Schema\Db\MySQL\DDL\Triggers\MigrateDataBetweenShards;
use Magento\Framework\Setup\Declaration\Schema\Db\Statement;
use Magento\Framework\Setup\Declaration\Schema\Dto\ElementFactory;
use Magento\Framework\Setup\Declaration\Schema\Dto\Table;
use Magento\Framework\Setup\Declaration\Schema\ElementHistory;
use Magento\Framework\Setup\Declaration\Schema\ElementHistoryFactory;
use Magento\Framework\Setup\Declaration\Schema\OperationInterface;

class ReCreateTable implements OperationInterface
{
    const OPERATION_NAME = 'recreate_table';

    private $createTable;

    private $dropTable;

    private $migrateDataBetweenShards;

    private $elementHistoryFactory;

    private $elementFactory;

    public function __construct(
        CreateTable $createTable,
        DropTable $dropTable,
        MigrateDataBetweenShards $migrateDataBetweenShards,
        ElementHistoryFactory $elementHistoryFactory,
        ElementFactory $elementFactory
    ) {
        $this->createTable = $createTable;
        $this->dropTable = $dropTable;
        $this->migrateDataBetweenShards = $migrateDataBetweenShards;
        $this->elementHistoryFactory = $elementHistoryFactory;
        $this->elementFactory = $elementFactory;
    }

    public function isOperationDestructive()
    {
        return true;
    }

    public function getOperationName()
    {
        return self::OPERATION_NAME;
    }

    private function getRecreatedTable(ElementHistory $elementHistory) : Table
    {
        $newTable = $elementHistory->getNew();
        $oldTable = $elementHistory->getOld();
        $recreationTable = $this->elementFactory->create(
            'table',
            [
                'name' => $newTable->getName(),
                'type' => 'table',
                'nameWithoutPrefix' => $newTable->getNameWithoutPrefix(),
                'resource' => $newTable->getResource(),
                'engine' => $newTable->getEngine(),
                'charset' => $newTable->getCharset(),
                'collation' => $newTable->getCollation(),
                'onCreate' => $newTable->getOnCreate(),
                'comment' => $newTable->getOnCreate(),
                'columns' => $oldTable->getColumns(),
                'indexes' => $oldTable->getIndexes(),
                'constraints' => array_merge($oldTable->getInternalConstraints(), $newTable->getReferenceConstraints())
            ]
        );

        return $recreationTable;
    }

    public function doOperation(ElementHistory $elementHistory)
    {
        $recreatedTable = $this->getRecreatedTable($elementHistory);
        $recreatedElementHistory = $this->elementHistoryFactory->create(
            [
                'old' => $elementHistory->getOld(),
                'new' => $recreatedTable
            ]
        );
        $statements = $this->createTable->doOperation($recreatedElementHistory);
        foreach ($statements as $statement) {
            if ($this->migrateDataBetweenShards->isApplicable((string) $recreatedTable->getOnCreate())) {
                $statement->addTrigger($this->migrateDataBetweenShards->getCallback($recreatedElementHistory));
            }
        }

        return array_merge($statements, $this->dropTable->doOperation($recreatedElementHistory));
    }
}