MigrateDataBetweenShards.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\Framework\Setup\Declaration\Schema\Db\MySQL\DDL\Triggers;

use Magento\Framework\App\ResourceConnection;
use Magento\Framework\Setup\Declaration\Schema\DataSavior\SelectGenerator;
use Magento\Framework\Setup\Declaration\Schema\Db\DDLTriggerInterface;
use Magento\Framework\Setup\Declaration\Schema\Dto\Table;
use Magento\Framework\Setup\Declaration\Schema\ElementHistory;

class MigrateDataBetweenShards implements DDLTriggerInterface
{
    const SKIP_MIGRATION_DATA_FLAG = 'skip-migration';

    private $resourceConnection;

    private $selectGenerator;

    public function __construct(
        ResourceConnection $resourceConnection,
        SelectGenerator $selectGenerator
    ) {
        $this->resourceConnection = $resourceConnection;
        $this->selectGenerator = $selectGenerator;
    }

    public function isApplicable(string $statement) : bool
    {
        return $statement !== self::SKIP_MIGRATION_DATA_FLAG;
    }

    public function getCallback(ElementHistory $elementHistory) : callable
    {
        $newTable = $elementHistory->getNew();
        $oldTable = $elementHistory->getOld();
        $that = $this;

        return function () use ($newTable, $oldTable, $that) {
            $firstConnection = $that->resourceConnection->getConnection($oldTable->getResource());
            $secondConnection = $that->resourceConnection->getConnection($newTable->getResource());
            $select = $firstConnection->select()->from($oldTable->getName());

            foreach ($this->selectGenerator->generator($select, $oldTable->getResource()) as $data) {
                if (count($data)) {
                    $columns = array_keys($data[0]);
                    $secondConnection->insertArray($newTable->getName(), $columns, $data);
                }
            }
        };
    }
}