ColumnSavior.php

Go to the documentation of this file.

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

use Magento\Framework\App\ResourceConnection;
use Magento\Framework\DB\SelectFactory;
use Magento\Framework\Setup\Declaration\Schema\Dto\Column;
use Magento\Framework\Setup\Declaration\Schema\Dto\ElementInterface;
use Magento\Framework\Setup\Declaration\Schema\Dto\Table;

class ColumnSavior implements DataSaviorInterface
{
    private $selectGeneratorFactory;

    private $resourceConnection;

    private $uniqueConstraintsResolver;

    private $dumpAccessor;

    private $selectFactory;

    public function __construct(
        ResourceConnection $resourceConnection,
        SelectGeneratorFactory $selectGeneratorFactory,
        DumpAccessorInterface $dumpAccessor,
        UniqueConstraintsResolver $uniqueConstraintsResolver,
        SelectFactory $selectFactory
    ) {
        $this->selectGeneratorFactory = $selectGeneratorFactory;
        $this->resourceConnection = $resourceConnection;
        $this->uniqueConstraintsResolver = $uniqueConstraintsResolver;
        $this->dumpAccessor = $dumpAccessor;
        $this->selectFactory = $selectFactory;
    }

    private function prepareColumnSelect(Column $column, array $fieldsToDump)
    {
        $adapter = $this->resourceConnection->getConnection($column->getTable()->getResource());
        $select = $this->selectFactory->create($adapter);
        $select->from($column->getTable()->getName(), $fieldsToDump);
        return $select;
    }

    public function dump(ElementInterface $column)
    {
        $columns = $this->uniqueConstraintsResolver->resolve($column->getTable());

        if ($columns) {
            $connectionName = $column->getTable()->getResource();
            $columns[] = $column->getName();
            $select = $this->prepareColumnSelect($column, $columns);
            $selectGenerator = $this->selectGeneratorFactory->create();
            $resourceSignature = $this->generateDumpFileSignature($column);

            foreach ($selectGenerator->generator($select, $connectionName) as $data) {
                $this->dumpAccessor->save($resourceSignature, $data);
            }
        }
    }

    private function applyDumpChunk(Table $table, $data)
    {
        $columns = [];
        $adapter = $this->resourceConnection->getConnection($table->getResource());
        $firstRow = reset($data);

        foreach ($table->getColumns() as $column) {
            $columns[$column->getName()] = $column->getName();
        }

        $adapter->insertOnDuplicate($table->getName(), $data, array_keys($firstRow));
    }

    private function generateDumpFileSignature(Column $column)
    {
        $dimensions = [
            $column->getTable()->getName(),
            $column->getElementType(),
            $column->getName()
        ];

        return implode("_", $dimensions);
    }

    public function restore(ElementInterface $column)
    {
        $file = $this->generateDumpFileSignature($column);
        $generator = $this->dumpAccessor->read($file);

        while ($generator->valid()) {
            $data = $generator->current();
            $this->applyDumpChunk(
                $column->getTable(),
                $data
            );
            $generator->next();
        }

        $this->dumpAccessor->destruct($file);
    }

    public function isAcceptable(ElementInterface $element)
    {
        return $element instanceof Column;
    }
}