TableSavior.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\ElementInterface;
use Magento\Framework\Setup\Declaration\Schema\Dto\Table;

class TableSavior implements DataSaviorInterface
{
    private $selectGeneratorFactory;

    private $resourceConnection;

    private $dumpAccessor;

    private $selectFactory;

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

    private function prepareTableSelect(Table $table)
    {
        $adapter = $this->resourceConnection->getConnection($table->getResource());
        $select = $this->selectFactory->create($adapter);
        $select->from($table->getName());
        return $select;
    }

    public function dump(ElementInterface $table)
    {
        $connectionName = $table->getResource();
        $select = $this->prepareTableSelect($table);
        $selectGenerator = $this->selectGeneratorFactory->create();
        $resourceSignature = $this->generateDumpFileSignature($table);

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

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

        return $columns;
    }

    private function applyDumpChunk(Table $table, $data)
    {
        $columns = $this->getTableColumnNames($table);
        $adapter = $this->resourceConnection->getConnection($table->getResource());
        $adapter->insertArray($table->getName(), $columns, $data);
    }

    private function generateDumpFileSignature(Table $table)
    {
        return $table->getName();
    }

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

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

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

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